Commit e2c30270 authored by Beatriz Rizental's avatar Beatriz Rizental Committed by Pier Angelo Vendrame
Browse files

TB 43817: Add tests for Tor Browser 

This is a catch all commits for adding any tests or
testing infrastructure that doesn't obviously fit
any other commit.
parent f4993bbd

python/mozbuild/mozbuild/action/test_archive.py

+6 −0
Original line number Diff line number Diff line
@@ -230,6 +230,12 @@ ARCHIVE_FILES = { 
            "pattern": "**",

            "dest": "certs",

        },

        {

            "source": buildconfig.topsrcdir,

            "base": "",

            "pattern": "testing/tor",

            "dest": "tor",

        },

    ],

    "cppunittest": [

        {"source": STAGE, "base": "", "pattern": "cppunittest/**"},

testing/marionette/harness/marionette_harness/tests/integration-tests.toml

+4 −0
Original line number Diff line number Diff line
@@ -58,6 +58,10 @@ 


["include:../../../../../netwerk/test/marionette/manifest.toml"]



# tor tests



["include:../../../../../testing/tor/manifest.toml"]



# toolkit tests



["include:../../../../../toolkit/components/antitracking/bouncetrackingprotection/test/marionette/manifest.toml"]

testing/moz.build

+2 −0
Original line number Diff line number Diff line
@@ -28,3 +28,5 @@ DIRS += ["manifest", "mozbase", "mozharness"] 
PERFTESTS_MANIFESTS += [

    "performance/perftest.toml",

]



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

testing/tor/manifest.toml

0 → 100644
+6 −0
Original line number Diff line number Diff line 
[DEFAULT]

tags = "tor"



["test_circuit_isolation.py"]



["test_network_check.py"]

testing/tor/test_circuit_isolation.py

0 → 100644
+101 −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 tearDown(self):

        self.marionette.restart(in_app=False, clean=True)

        super().tearDown()



    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!",

        )