From 1e0f7fc15dc444f07e8eb807716b6fff5a7605b9 Mon Sep 17 00:00:00 2001
From: Daniel Thorn <dthorn@mozilla.com>
Date: Fri, 9 Aug 2024 12:16:17 -0700
Subject: [PATCH] test update-obs-common

---
 tests/test_update_obs_common.py | 40 +++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 tests/test_update_obs_common.py

diff --git a/tests/test_update_obs_common.py b/tests/test_update_obs_common.py
new file mode 100644
index 0000000..c56b90c
--- /dev/null
+++ b/tests/test_update_obs_common.py
@@ -0,0 +1,40 @@
+# 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 https://mozilla.org/MPL/2.0/.
+import os
+import re
+
+import pytest
+
+from obs_common import update_obs_common
+
+
+@pytest.fixture
+def tmp_cwd(tmp_path):
+    orig_wd = os.getcwd()
+    os.chdir(tmp_path)
+    try:
+        yield tmp_path
+    finally:
+        os.chdir(orig_wd)
+
+
+def test_update_from_old(tmp_cwd):
+    req = tmp_cwd / "requirements.in"
+    req.write_text("obs-common @ https://path/to/package.whl\n")
+    EXPECT_PATTERN = re.compile(
+        "obs-common @ https://github.com/mozilla-services/obs-common/"
+        "releases/download/[^/]+/obs_common-[^-]+-py2.py3-none-any.whl\n",
+        re.MULTILINE,
+    )
+    assert update_obs_common.main() == 0
+    assert EXPECT_PATTERN.fullmatch(req.read_text())
+
+
+def test_update_idempotent(tmp_cwd):
+    req = tmp_cwd / "requirements.in"
+    req.write_text("obs-common @ https://path/to/package.whl\n")
+    assert update_obs_common.main() == 0
+    expect = req.read_text()
+    assert update_obs_common.main() == 0
+    assert req.read_text() == expect