diff --git a/etc/taskcluster/mock.py b/etc/taskcluster/mock.py index 107a8380f1a..14b52720db7 100755 --- a/etc/taskcluster/mock.py +++ b/etc/taskcluster/mock.py @@ -38,6 +38,7 @@ class Index: stringDate = str slugId = b"".lower sys.exit = Queue = fromNow = MagicMock() +sys.modules["taskcluster"] = sys.modules[__name__] sys.dont_write_bytecode = True os.environ.update(**{k: k for k in "TASK_ID TASK_OWNER TASK_SOURCE GIT_URL GIT_SHA".split()}) os.environ["GIT_REF"] = "refs/heads/auto"