TB 43817: Add tests for Tor Browser (9ffa3054) · Commits · The Tor Project / Applications / Tor Browser

testing/moz.build

+2 −0

Original line number	Diff line number	Diff line
		@@ -17,3 +17,5 @@ DIRS += ["mozbase", "mozharness"]
		PERFTESTS_MANIFESTS += [
		"performance/perftest.toml",
		]

		MARIONETTE_MANIFESTS += ["tor/marionette.toml"]

testing/tor/marionette.toml

0 → 100644

+4 −0

Original line number	Diff line number	Diff line
		[DEFAULT]

		["test_circuit_isolation.py"]
		["test_network_check.py"]

testing/tor/test_circuit_isolation.py

0 → 100644

+98 −0

Original line number	Diff line number	Diff line
		from ipaddress import ip_address

		from marionette_driver import By
		from marionette_driver.errors import NoSuchElementException
		from marionette_harness import MarionetteTestCase

		TOR_BOOTSTRAP_TIMEOUT = 30000 # 30s


		class TestCircuitIsolation(MarionetteTestCase):

		def bootstrap(self):
		with self.marionette.using_context("chrome"):
		self.marionette.execute_async_script(
		"""
		const { TorConnect, TorConnectTopics } = ChromeUtils.importESModule(
		"resource://gre/modules/TorConnect.sys.mjs"
		);
		const [resolve] = arguments;

		function waitForBootstrap() {
		const topic = TorConnectTopics.BootstrapComplete;
		Services.obs.addObserver(function observer() {
		Services.obs.removeObserver(observer, topic);
		resolve();
		}, topic);
		TorConnect.beginBootstrapping();
		}

		const stageTopic = TorConnectTopics.StageChange;
		function stageObserver() {
		if (TorConnect.canBeginNormalBootstrap) {
		Services.obs.removeObserver(stageObserver, stageTopic);
		waitForBootstrap();
		}
		}
		Services.obs.addObserver(stageObserver, stageTopic);
		stageObserver();
		""",
		script_timeout=TOR_BOOTSTRAP_TIMEOUT,
		)

		def extract_from_check_tpo(self):
		# Fetch the IP from check.torproject.org.
		# In addition to that, since we are loading this page, we
		# perform some additional sanity checks.
		self.marionette.navigate("https://check.torproject.org/")
		# When check.tpo's check succeed (i.e., it thinks we're
		# connecting through tor), we should be able to find a h1.on,
		# with some message...
		on = self.marionette.find_element(By.CLASS_NAME, "on")
		self.assertIsNotNone(
		on,
		"h1.on not found, you might not be connected through tor",
		)
		# ... but if it fails, the message is inside a h1.off. We want
		# to make sure we do not find that either (even though there is
		# no reason for both of the h1 to be outputted at the moment).
		self.assertRaises(
		NoSuchElementException,
		self.marionette.find_element,
		By.CLASS_NAME,
		"off",
		)
		ip = self.marionette.find_element(By.TAG_NAME, "strong")
		return ip_address(ip.text.strip())

		def extract_generic(self, url):
		# Fetch the IP address from any generic page that only contains
		# the address.
		self.marionette.navigate(url)
		return ip_address(
		self.marionette.execute_script(
		"return document.documentElement.textContent"
		).strip()
		)

		def test_circuit_isolation(self):
		self.bootstrap()
		ips = [
		self.extract_from_check_tpo(),
		self.extract_generic("https://am.i.mullvad.net/ip"),
		self.extract_generic("https://test1.ifconfig.me/ip"),
		]
		self.logger.info(f"Found the following IP addresses: {ips}")
		unique_ips = set(ips)
		self.logger.info(f"Found the following unique IP addresses: {unique_ips}")
		self.assertEqual(
		len(ips),
		len(unique_ips),
		"Some of the IP addresses we got are not unique.",
		)
		duplicate = self.extract_generic("https://test2.ifconfig.me/ip")
		self.assertEqual(
		ips[-1],
		duplicate,
		"Two IPs that were expected to be equal are different, we might be over isolating!",
		)

testing/tor/test_network_check.py

0 → 100644

+68 −0

Original line number	Diff line number	Diff line
		from marionette_driver import By, Wait, errors
		from marionette_driver.localization import L10n
		from marionette_harness import MarionetteTestCase

		NETWORK_CHECK_URL = "https://check.torproject.org/"
		TOR_BOOTSTRAP_TIMEOUT = 30 # 30s

		STRINGS_LOCATION = "chrome://torbutton/locale/torConnect.properties"


		class TestNetworkCheck(MarionetteTestCase):
		def setUp(self):
		MarionetteTestCase.setUp(self)

		self.l10n = L10n(self.marionette)

		def attemptConnection(self, tries=1):
		if tries > 3:
		self.assertTrue(False, "Failed to connect to Tor after 3 attempts")

		connectBtn = self.marionette.find_element(By.ID, "connectButton")
		Wait(self.marionette, timeout=10).until(
		lambda _: connectBtn.is_displayed(),
		message="Timed out waiting for tor connect button to show up.",
		)
		connectBtn.click()

		try:

		def check(m):
		if not m.get_url().startswith("about:torconnect"):
		# We have finished connecting and have been redirected.
		return True

		try:
		heading = self.marionette.find_element(By.ID, "tor-connect-heading")
		except errors.NoSuchElementException:
		# Page is probably redirecting.
		return False

		if heading.text not in [
		self.l10n.localize_property(
		[STRINGS_LOCATION], "torConnect.torConnecting"
		),
		self.l10n.localize_property(
		[STRINGS_LOCATION], "torConnect.torConnected"
		),
		]:
		raise ValueError("Tor connect page is not connecting or connected")

		return False

		Wait(self.marionette, timeout=TOR_BOOTSTRAP_TIMEOUT).until(check)
		except (errors.TimeoutException, ValueError):
		cancelBtn = self.marionette.find_element(By.ID, "cancelButton")
		if cancelBtn.is_displayed():
		cancelBtn.click()

		self.attemptConnection(tries + 1)

		def test_network_check(self):
		self.attemptConnection()
		self.marionette.navigate(NETWORK_CHECK_URL)
		self.assertRegex(
		self.marionette.title,
		r"^Congratulations\.",
		f"{NETWORK_CHECK_URL} should have the expected title.",
		)