1 files changed, 27 insertions, 0 deletions
diff --git a/tests/test_output_filename.py b/tests/test_output_filename.py
new file mode 100644
index 0000000..fab0e98
--- /dev/null
+++ b/tests/test_output_filename.py
@@ -0,0 +1,27 @@
+from pathlib import Path
+
+import pytest
+
+from debputy.util import compute_output_filename
+
+
+def write_unpacked_deb(root: Path, package: str, version: str, arch: str):
+    (root / "control").write_text(
+        f"Package: {package}\nVersion: {version}\nArchitecture: {arch}\n"
+    )
+
+
+@pytest.mark.parametrize(
+    "package,version,arch,is_udeb,expected",
+    [
+        ("fake", "1.0", "amd64", False, "fake_1.0_amd64.deb"),
+        ("fake", "1.0", "amd64", True, "fake_1.0_amd64.udeb"),
+        ("fake", "2:1.0", "amd64", False, "fake_1.0_amd64.deb"),
+        ("fake", "2:1.0", "amd64", True, "fake_1.0_amd64.udeb"),
+        ("fake", "3.0", "all", False, "fake_3.0_all.deb"),
+        ("fake", "3.0", "all", True, "fake_3.0_all.udeb"),
+    ],
+)
+def test_generate_deb_filename(tmp_path, package, version, arch, is_udeb, expected):
+    write_unpacked_deb(tmp_path, package, version, arch)
+    assert compute_output_filename(str(tmp_path), is_udeb) == expected