mozilla-taskgraph 2.0.0__tar.gz → 2.0.1__tar.gz

{mozilla-taskgraph-2.0.0 → mozilla-taskgraph-2.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mozilla-taskgraph
-Version: 2.0.0
+Version: 2.0.1
 Summary: Mozilla-specific transforms and utilities for Taskgraph
 Home-page: https://github.com/mozilla-releng/mozilla-taskgraph
 Classifier: Development Status :: 4 - Beta

{mozilla-taskgraph-2.0.0 → mozilla-taskgraph-2.0.1}/src/mozilla_taskgraph/worker_types.py

@@ -80,24 +80,26 @@ def build_bitrise_payload(config, task, task_def):
         "branch_repo_owner": config.params["head_repository"],
     }
 
-    head_ref = normref(config.params["head_ref"])
-    head_tag = normref(config.params["head_tag"], type="tags")
-    base_ref = normref(config.params["base_ref"])
-
-    if head_ref:
+    if head_ref := normref(config.params["head_ref"]):
         global_params["branch"] = head_ref
 
-    if head_tag:
+    if head_tag := normref(config.params["head_tag"], type="tags"):
         global_params["tag"] = head_tag
 
+    if commit_message := config.params.get("commit_message"):
+        global_params["commit_message"] = commit_message
+
+    if pull_request_number := config.params.get("pull_request_number"):
+        global_params["pull_request_id"] = pull_request_number
+
     if config.params["tasks_for"] == "github-pull-request":
         global_params["pull_request_author"] = config.params["owner"]
 
-        if base_ref:
+        if base_ref := normref(config.params["base_ref"]):
             global_params["branch_dest"] = base_ref
 
-        if config.params["base_repository"]:
-            global_params["branch_dest_repo_owner"] = config.params["base_repository"]
+        if base_repository := config.params["base_repository"]:
+            global_params["branch_dest_repo_owner"] = base_repository
 
     task_def["payload"] = {"global_params": global_params}
     if workflow_params:

{mozilla-taskgraph-2.0.0 → mozilla-taskgraph-2.0.1}/src/mozilla_taskgraph.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mozilla-taskgraph
-Version: 2.0.0
+Version: 2.0.1
 Summary: Mozilla-specific transforms and utilities for Taskgraph
 Home-page: https://github.com/mozilla-releng/mozilla-taskgraph
 Classifier: Development Status :: 4 - Beta

{mozilla-taskgraph-2.0.0 → mozilla-taskgraph-2.0.1}/test/test_worker_types.py

@@ -58,7 +58,11 @@ from mozilla_taskgraph import worker_types
         ),
         pytest.param(
             {"bitrise": {"app": "some-app", "workflows": ["bar"]}},
-            {"tasks_for": "github-pull-request"},
+            {
+                "tasks_for": "github-pull-request",
+                "commit_message": "This is a commit",
+                "pull_request_number": "123",
+            },
             {
                 "payload": {
                     "global_params": {
@@ -68,6 +72,8 @@ from mozilla_taskgraph import worker_types
                         "branch_repo_owner": "http://example.com/head/repo",
                         "commit_hash": "abcdef",
                         "pull_request_author": "some-owner",
+                        "commit_message": "This is a commit",
+                        "pull_request_id": "123",
                     },
                 },
                 "scopes": ["foo:bitrise:app:some-app", "foo:bitrise:workflow:bar"],
@@ -83,6 +89,8 @@ from mozilla_taskgraph import worker_types
                 "head_ref": "",
                 "head_tag": "some-tag",
                 "tasks_for": "github-pull-request",
+                "commit_message": "This is a commit",
+                "pull_request_number": "123",
             },
             {
                 "payload": {
@@ -91,6 +99,8 @@ from mozilla_taskgraph import worker_types
                         "commit_hash": "abcdef",
                         "pull_request_author": "some-owner",
                         "tag": "some-tag",
+                        "commit_message": "This is a commit",
+                        "pull_request_id": "123",
                     }
                 },
                 "scopes": ["foo:bitrise:app:some-app", "foo:bitrise:workflow:bar"],
@@ -106,6 +116,8 @@ from mozilla_taskgraph import worker_types
                 "head_ref": "refs/heads/bar",
                 "head_tag": "refs/tags/some-tag",
                 "tasks_for": "github-pull-request",
+                "commit_message": "This is a commit",
+                "pull_request_number": "123",
             },
             {
                 "payload": {
@@ -116,6 +128,8 @@ from mozilla_taskgraph import worker_types
                         "commit_hash": "abcdef",
                         "pull_request_author": "some-owner",
                         "tag": "some-tag",
+                        "commit_message": "This is a commit",
+                        "pull_request_id": "123",
                     }
                 },
                 "scopes": ["foo:bitrise:app:some-app", "foo:bitrise:workflow:bar"],
@@ -131,6 +145,8 @@ from mozilla_taskgraph import worker_types
                 "head_ref": "refs/tags/bar",
                 "head_tag": "refs/heads/some-tag",
                 "tasks_for": "github-pull-request",
+                "commit_message": "This is a commit",
+                "pull_request_number": "123",
             },
             {
                 "payload": {
@@ -138,6 +154,8 @@ from mozilla_taskgraph import worker_types
                         "branch_repo_owner": "http://example.com/head/repo",
                         "commit_hash": "abcdef",
                         "pull_request_author": "some-owner",
+                        "commit_message": "This is a commit",
+                        "pull_request_id": "123",
                     }
                 },
                 "scopes": ["foo:bitrise:app:some-app", "foo:bitrise:workflow:bar"],