summaryrefslogtreecommitdiffstats
path: root/testing/condprofile/condprof/tests/test_client.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 17:32:43 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 17:32:43 +0000
commit6bf0a5cb5034a7e684dcc3500e841785237ce2dd (patch)
treea68f146d7fa01f0134297619fbe7e33db084e0aa /testing/condprofile/condprof/tests/test_client.py
parentInitial commit. (diff)
downloadthunderbird-upstream.tar.xz
thunderbird-upstream.zip
Adding upstream version 1:115.7.0.upstream/1%115.7.0upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rw-r--r--testing/condprofile/condprof/tests/test_client.py126
1 files changed, 126 insertions, 0 deletions
diff --git a/testing/condprofile/condprof/tests/test_client.py b/testing/condprofile/condprof/tests/test_client.py
new file mode 100644
index 0000000000..8410133e16
--- /dev/null
+++ b/testing/condprofile/condprof/tests/test_client.py
@@ -0,0 +1,126 @@
+import json
+import os
+import re
+import shutil
+import tarfile
+import tempfile
+import unittest
+
+import responses
+from mozprofile.prefs import Preferences
+
+from condprof.client import ROOT_URL, TC_SERVICE, get_profile
+from condprof.util import _DEFAULT_SERVER
+
+PROFILE = re.compile(ROOT_URL + "/.*/.*tgz")
+PROFILE_FOR_TESTS = os.path.join(os.path.dirname(__file__), "profile")
+SECRETS = re.compile(_DEFAULT_SERVER + "/.*")
+SECRETS_PROXY = re.compile("http://taskcluster/secrets/.*")
+
+
+class TestClient(unittest.TestCase):
+ def setUp(self):
+ self.profile_dir = tempfile.mkdtemp()
+
+ # creating profile.tgz on the fly for serving it
+ profile_tgz = os.path.join(self.profile_dir, "profile.tgz")
+ with tarfile.open(profile_tgz, "w:gz") as tar:
+ tar.add(PROFILE_FOR_TESTS, arcname=".")
+
+ # self.profile_data is the tarball we're sending back via HTTP
+ with open(profile_tgz, "rb") as f:
+ self.profile_data = f.read()
+
+ self.target = tempfile.mkdtemp()
+ self.download_dir = os.path.expanduser("~/.condprof-cache")
+ if os.path.exists(self.download_dir):
+ shutil.rmtree(self.download_dir)
+
+ responses.add(
+ responses.GET,
+ PROFILE,
+ body=self.profile_data,
+ headers={"content-length": str(len(self.profile_data)), "ETag": "'12345'"},
+ status=200,
+ )
+
+ responses.add(
+ responses.HEAD,
+ PROFILE,
+ body="",
+ headers={"content-length": str(len(self.profile_data)), "ETag": "'12345'"},
+ status=200,
+ )
+
+ responses.add(responses.HEAD, TC_SERVICE, body="", status=200)
+
+ secret = {"secret": {"username": "user", "password": "pass"}}
+ secret = json.dumps(secret)
+ for pattern in (SECRETS, SECRETS_PROXY):
+ responses.add(
+ responses.GET,
+ pattern,
+ body=secret,
+ headers={"content-length": str(len(secret))},
+ status=200,
+ )
+
+ def tearDown(self):
+ shutil.rmtree(self.target)
+ shutil.rmtree(self.download_dir)
+ shutil.rmtree(self.profile_dir)
+
+ @responses.activate
+ def test_cache(self):
+ download_dir = os.path.expanduser("~/.condprof-cache")
+ if os.path.exists(download_dir):
+ num_elmts = len(os.listdir(download_dir))
+ else:
+ num_elmts = 0
+
+ get_profile(self.target, "win64", "settled", "default")
+
+ # grabbing a profile should generate two files
+ self.assertEqual(len(os.listdir(download_dir)), num_elmts + 2)
+
+ # we do at least two network calls when getting a file,
+ # a HEAD and a GET and possibly a TC secret
+ self.assertTrue(len(responses.calls) >= 2)
+
+ # reseting the response counters
+ responses.calls.reset()
+
+ # and we should reuse them without downloading the file again
+ get_profile(self.target, "win64", "settled", "default")
+
+ # grabbing a profile should not download new stuff
+ self.assertEqual(len(os.listdir(download_dir)), num_elmts + 2)
+
+ # and do a single extra HEAD call, everything else is cached,
+ # even the TC secret
+ self.assertEqual(len(responses.calls), 2)
+
+ prefs_js = os.path.join(self.target, "prefs.js")
+ prefs = Preferences.read_prefs(prefs_js)
+
+ # check that the gfx.blacklist prefs where cleaned out
+ for name, value in prefs:
+ self.assertFalse(name.startswith("gfx.blacklist"))
+
+ # check that we have the startupScanScopes option forced
+ prefs = dict(prefs)
+ self.assertEqual(prefs["extensions.startupScanScopes"], 1)
+
+ # make sure we don't have any marionette option set
+ user_js = os.path.join(self.target, "user.js")
+ for name, value in Preferences.read_prefs(user_js):
+ self.assertFalse(name.startswith("marionette."))
+
+
+if __name__ == "__main__":
+ try:
+ import mozunit
+ except ImportError:
+ pass
+ else:
+ mozunit.main(runwith="unittest")