Adding upstream version 1:115.7.0.upstream/1%115.7.0upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 176 insertions, 0 deletions

diff --git a/taskcluster/gecko_taskgraph/test/test_decision.py b/taskcluster/gecko_taskgraph/test/test_decision.py
new file mode 100644
index 0000000000..81be7e0c44
--- /dev/null
+++ b/taskcluster/gecko_taskgraph/test/test_decision.py
@@ -0,0 +1,176 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+
+import json
+import os
+import shutil
+import tempfile
+import unittest
+from unittest.mock import patch
+
+import pytest
+from mozunit import MockedOpen, main
+from taskgraph.util.yaml import load_yaml
+
+from gecko_taskgraph import decision
+from gecko_taskgraph.parameters import register_parameters
+
+FAKE_GRAPH_CONFIG = {"product-dir": "browser", "taskgraph": {}}
+
+
+@pytest.fixture(scope="module", autouse=True)
+def register():
+    register_parameters()
+
+
+class TestDecision(unittest.TestCase):
+    def test_write_artifact_json(self):
+        data = [{"some": "data"}]
+        tmpdir = tempfile.mkdtemp()
+        try:
+            decision.ARTIFACTS_DIR = os.path.join(tmpdir, "artifacts")
+            decision.write_artifact("artifact.json", data)
+            with open(os.path.join(decision.ARTIFACTS_DIR, "artifact.json")) as f:
+                self.assertEqual(json.load(f), data)
+        finally:
+            if os.path.exists(tmpdir):
+                shutil.rmtree(tmpdir)
+            decision.ARTIFACTS_DIR = "artifacts"
+
+    def test_write_artifact_yml(self):
+        data = [{"some": "data"}]
+        tmpdir = tempfile.mkdtemp()
+        try:
+            decision.ARTIFACTS_DIR = os.path.join(tmpdir, "artifacts")
+            decision.write_artifact("artifact.yml", data)
+            self.assertEqual(load_yaml(decision.ARTIFACTS_DIR, "artifact.yml"), data)
+        finally:
+            if os.path.exists(tmpdir):
+                shutil.rmtree(tmpdir)
+            decision.ARTIFACTS_DIR = "artifacts"
+
+
+class TestGetDecisionParameters(unittest.TestCase):
+
+    ttc_file = os.path.join(os.getcwd(), "try_task_config.json")
+
+    def setUp(self):
+        self.options = {
+            "base_repository": "https://hg.mozilla.org/mozilla-unified",
+            "head_repository": "https://hg.mozilla.org/mozilla-central",
+            "head_rev": "abcd",
+            "head_ref": "ef01",
+            "head_tag": "",
+            "message": "",
+            "project": "mozilla-central",
+            "pushlog_id": "143",
+            "pushdate": 1503691511,
+            "owner": "nobody@mozilla.com",
+            "repository_type": "hg",
+            "tasks_for": "hg-push",
+            "level": "3",
+        }
+
+    @patch("gecko_taskgraph.decision.get_hg_revision_branch")
+    @patch("gecko_taskgraph.decision._determine_more_accurate_base_rev")
+    def test_simple_options(
+        self, mock_determine_more_accurate_base_rev, mock_get_hg_revision_branch
+    ):
+        mock_get_hg_revision_branch.return_value = "default"
+        mock_determine_more_accurate_base_rev.return_value = "baserev"
+        with MockedOpen({self.ttc_file: None}):
+            params = decision.get_decision_parameters(FAKE_GRAPH_CONFIG, self.options)
+        self.assertEqual(params["pushlog_id"], "143")
+        self.assertEqual(params["build_date"], 1503691511)
+        self.assertEqual(params["hg_branch"], "default")
+        self.assertEqual(params["moz_build_date"], "20170825200511")
+        self.assertEqual(params["try_mode"], None)
+        self.assertEqual(params["try_options"], None)
+        self.assertEqual(params["try_task_config"], {})
+
+    @patch("gecko_taskgraph.decision.get_hg_revision_branch")
+    @patch("gecko_taskgraph.decision._determine_more_accurate_base_rev")
+    def test_no_email_owner(
+        self, mock_determine_more_accurate_base_rev, mock_get_hg_revision_branch
+    ):
+        mock_get_hg_revision_branch.return_value = "default"
+        mock_determine_more_accurate_base_rev.return_value = "baserev"
+        self.options["owner"] = "ffxbld"
+        with MockedOpen({self.ttc_file: None}):
+            params = decision.get_decision_parameters(FAKE_GRAPH_CONFIG, self.options)
+        self.assertEqual(params["owner"], "ffxbld@noreply.mozilla.org")
+
+    @patch("gecko_taskgraph.decision.get_hg_revision_branch")
+    @patch("gecko_taskgraph.decision.get_hg_commit_message")
+    @patch("gecko_taskgraph.decision._determine_more_accurate_base_rev")
+    def test_try_options(
+        self,
+        mock_determine_more_accurate_base_rev,
+        mock_get_hg_commit_message,
+        mock_get_hg_revision_branch,
+    ):
+        mock_get_hg_commit_message.return_value = "try: -b do -t all --artifact"
+        mock_get_hg_revision_branch.return_value = "default"
+        mock_determine_more_accurate_base_rev.return_value = "baserev"
+        self.options["project"] = "try"
+        with MockedOpen({self.ttc_file: None}):
+            params = decision.get_decision_parameters(FAKE_GRAPH_CONFIG, self.options)
+        self.assertEqual(params["try_mode"], "try_option_syntax")
+        self.assertEqual(params["try_options"]["build_types"], "do")
+        self.assertEqual(params["try_options"]["unittests"], "all")
+        self.assertEqual(
+            params["try_task_config"],
+            {
+                "gecko-profile": False,
+                "use-artifact-builds": True,
+                "env": {},
+            },
+        )
+
+    @patch("gecko_taskgraph.decision.get_hg_revision_branch")
+    @patch("gecko_taskgraph.decision.get_hg_commit_message")
+    @patch("gecko_taskgraph.decision._determine_more_accurate_base_rev")
+    def test_try_task_config(
+        self,
+        mock_get_hg_commit_message,
+        mock_get_hg_revision_branch,
+        mock_determine_more_accurate_base_rev,
+    ):
+        mock_get_hg_commit_message.return_value = "Fuzzy query=foo"
+        mock_get_hg_revision_branch.return_value = "default"
+        mock_determine_more_accurate_base_rev.return_value = "baserev"
+        ttc = {"tasks": ["a", "b"]}
+        self.options["project"] = "try"
+        with MockedOpen({self.ttc_file: json.dumps(ttc)}):
+            params = decision.get_decision_parameters(FAKE_GRAPH_CONFIG, self.options)
+            self.assertEqual(params["try_mode"], "try_task_config")
+            self.assertEqual(params["try_options"], None)
+            self.assertEqual(params["try_task_config"], ttc)
+
+    def test_try_syntax_from_message_empty(self):
+        self.assertEqual(decision.try_syntax_from_message(""), "")
+
+    def test_try_syntax_from_message_no_try_syntax(self):
+        self.assertEqual(decision.try_syntax_from_message("abc | def"), "")
+
+    def test_try_syntax_from_message_initial_try_syntax(self):
+        self.assertEqual(
+            decision.try_syntax_from_message("try: -f -o -o"), "try: -f -o -o"
+        )
+
+    def test_try_syntax_from_message_initial_try_syntax_multiline(self):
+        self.assertEqual(
+            decision.try_syntax_from_message("try: -f -o -o\nabc\ndef"), "try: -f -o -o"
+        )
+
+    def test_try_syntax_from_message_embedded_try_syntax_multiline(self):
+        self.assertEqual(
+            decision.try_syntax_from_message("some stuff\ntry: -f -o -o\nabc\ndef"),
+            "try: -f -o -o",
+        )
+
+
+if __name__ == "__main__":
+    main()