1 files changed, 128 insertions, 0 deletions
diff --git a/taskcluster/gecko_taskgraph/test/test_util_partials.py b/taskcluster/gecko_taskgraph/test/test_util_partials.py
new file mode 100644
index 0000000000..3630d7b0ec
--- /dev/null
+++ b/taskcluster/gecko_taskgraph/test/test_util_partials.py
@@ -0,0 +1,128 @@
+# 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 unittest
+from unittest import mock
+
+from mozunit import main
+
+from gecko_taskgraph.util import partials
+
+release_blob = {
+    "fileUrls": {
+        "release-localtest": {
+            "completes": {
+                "*": "%OS_FTP%/%LOCALE%/firefox-92.0.1.complete.mar",
+            }
+        }
+    },
+    "platforms": {
+        "WINNT_x86_64-msvc": {
+            "locales": {
+                "en-US": {
+                    "buildID": "20210922161155",
+                }
+            }
+        }
+    },
+}
+
+
+def nightly_blob(release):
+    return {
+        "platforms": {
+            "WINNT_x86_64-msvc": {
+                "locales": {
+                    "en-US": {
+                        "buildID": release[-14:],
+                        "completes": [{"fileUrl": release}],
+                    }
+                }
+            }
+        }
+    }
+
+
+class TestReleaseHistory(unittest.TestCase):
+    @mock.patch("gecko_taskgraph.util.partials.get_release_builds")
+    @mock.patch("gecko_taskgraph.util.partials.get_sorted_releases")
+    def test_populate_release_history(self, get_sorted_releases, get_release_builds):
+        self.assertEqual(
+            partials.populate_release_history(
+                "Firefox", "mozilla-release", partial_updates={}
+            ),
+            {},
+        )
+        get_sorted_releases.assert_not_called()
+        get_release_builds.assert_not_called()
+
+        def patched_get_sorted_releases(product, branch):
+            assert branch == "mozilla-central"
+            return [
+                "Firefox-mozilla-central-nightly-20211003201113",
+                "Firefox-mozilla-central-nightly-20211003100640",
+                "Firefox-mozilla-central-nightly-20211002213629",
+                "Firefox-mozilla-central-nightly-20211002095048",
+                "Firefox-mozilla-central-nightly-20211001214601",
+                "Firefox-mozilla-central-nightly-20211001093323",
+            ]
+
+        def patched_get_release_builds(release, branch):
+            if branch == "mozilla-central":
+                return nightly_blob(release)
+            if branch == "mozilla-release":
+                return release_blob
+
+        get_sorted_releases.side_effect = patched_get_sorted_releases
+        get_release_builds.side_effect = patched_get_release_builds
+
+        self.assertEqual(
+            partials.populate_release_history(
+                "Firefox",
+                "mozilla-release",
+                partial_updates={"92.0.1": {"buildNumber": 1}},
+            ),
+            {
+                "WINNT_x86_64-msvc": {
+                    "en-US": {
+                        "target-92.0.1.partial.mar": {
+                            "buildid": "20210922161155",
+                            "mar_url": "win64/en-US/firefox-92.0.1.complete.mar",
+                            "previousVersion": "92.0.1",
+                            "previousBuildNumber": 1,
+                            "product": "Firefox",
+                        }
+                    }
+                }
+            },
+        )
+        self.assertEqual(
+            partials.populate_release_history("Firefox", "mozilla-central"),
+            {
+                "WINNT_x86_64-msvc": {
+                    "en-US": {
+                        "target.partial-1.mar": {
+                            "buildid": "20211003201113",
+                            "mar_url": "Firefox-mozilla-central-nightly-20211003201113",
+                        },
+                        "target.partial-2.mar": {
+                            "buildid": "20211003100640",
+                            "mar_url": "Firefox-mozilla-central-nightly-20211003100640",
+                        },
+                        "target.partial-3.mar": {
+                            "buildid": "20211002213629",
+                            "mar_url": "Firefox-mozilla-central-nightly-20211002213629",
+                        },
+                        "target.partial-4.mar": {
+                            "buildid": "20211002095048",
+                            "mar_url": "Firefox-mozilla-central-nightly-20211002095048",
+                        },
+                    }
+                }
+            },
+        )
+
+
+if __name__ == "__main__":
+    main()