summaryrefslogtreecommitdiffstats
path: root/testing/condprofile/condprof/tests/test_runner.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_runner.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_runner.py123
1 files changed, 123 insertions, 0 deletions
diff --git a/testing/condprofile/condprof/tests/test_runner.py b/testing/condprofile/condprof/tests/test_runner.py
new file mode 100644
index 0000000000..e200a537e1
--- /dev/null
+++ b/testing/condprofile/condprof/tests/test_runner.py
@@ -0,0 +1,123 @@
+import asyncio
+import os
+import re
+import shutil
+import tarfile
+import tempfile
+import unittest
+
+import responses
+
+from condprof import client
+from condprof.client import ROOT_URL, TC_SERVICE
+from condprof.main import main
+
+client.RETRIES = 1
+client.RETRY_PAUSE = 0
+GECKODRIVER = os.path.join(os.path.dirname(__file__), "fakegeckodriver.py")
+FIREFOX = os.path.join(os.path.dirname(__file__), "fakefirefox.py")
+CHANGELOG = re.compile(ROOT_URL + "/.*/changelog.json")
+FTP = "https://ftp.mozilla.org/pub/firefox/nightly/latest-mozilla-central/"
+PROFILE = re.compile(ROOT_URL + "/.*/.*tgz")
+
+
+with open(os.path.join(os.path.dirname(__file__), "ftp_mozilla.html")) as f:
+ FTP_PAGE = f.read()
+
+FTP_ARCHIVE = re.compile(
+ "https://ftp.mozilla.org/pub/firefox/nightly/" "latest-mozilla-central/firefox.*"
+)
+
+ADDON = re.compile("https://addons.mozilla.org/.*/.*xpi")
+
+
+async def fakesleep(duration):
+ if duration > 1:
+ duration = 1
+ await asyncio.realsleep(duration)
+
+
+asyncio.realsleep = asyncio.sleep
+asyncio.sleep = fakesleep
+
+
+class TestRunner(unittest.TestCase):
+ def setUp(self):
+ self.archive_dir = tempfile.mkdtemp()
+ responses.add(responses.GET, CHANGELOG, json={"error": "not found"}, status=404)
+ responses.add(
+ responses.GET, FTP, content_type="text/html", body=FTP_PAGE, status=200
+ )
+
+ profile_tgz = os.path.join(os.path.dirname(__file__), "profile.tgz")
+ profile = os.path.join(os.path.dirname(__file__), "profile")
+
+ with tarfile.open(profile_tgz, "w:gz") as tar:
+ tar.add(profile, arcname=".")
+
+ with open(profile_tgz, "rb") as f:
+ PROFILE_DATA = f.read()
+
+ os.remove(profile_tgz)
+
+ responses.add(
+ responses.GET,
+ FTP_ARCHIVE,
+ body="1",
+ headers={"content-length": "1"},
+ status=200,
+ )
+
+ responses.add(
+ responses.GET,
+ PROFILE,
+ body=PROFILE_DATA,
+ headers={"content-length": str(len(PROFILE_DATA))},
+ status=200,
+ )
+
+ responses.add(
+ responses.HEAD,
+ PROFILE,
+ body="",
+ headers={"content-length": str(len(PROFILE_DATA))},
+ status=200,
+ )
+
+ responses.add(responses.HEAD, FTP_ARCHIVE, body="", status=200)
+
+ responses.add(
+ responses.GET, ADDON, body="1", headers={"content-length": "1"}, status=200
+ )
+
+ responses.add(
+ responses.HEAD, ADDON, body="", headers={"content-length": "1"}, status=200
+ )
+
+ responses.add(responses.HEAD, TC_SERVICE, body="", status=200)
+
+ def tearDown(self):
+ shutil.rmtree(self.archive_dir)
+
+ @responses.activate
+ def test_runner(self):
+ if "FXA_USERNAME" not in os.environ:
+ os.environ["FXA_USERNAME"] = "me"
+ if "FXA_PASSWORD" not in os.environ:
+ os.environ["FXA_PASSWORD"] = "password"
+ try:
+ args = [
+ "--geckodriver",
+ GECKODRIVER,
+ "--firefox",
+ FIREFOX,
+ self.archive_dir,
+ ]
+ main(args)
+ # XXX we want a bunch of assertions here to check
+ # that the archives dir gets filled correctly
+ finally:
+ if os.environ["FXA_USERNAME"] == "me":
+ del os.environ["FXA_USERNAME"]
+ if os.environ["FXA_PASSWORD"] == "password":
+ del os.environ["FXA_PASSWORD"]