add tests for main_repair

Author: mayeut

src/auditwheel/main_repair.py

@@ -158,13 +158,13 @@ def execute(args: argparse.Namespace, parser: argparse.ArgumentParser) -> int:
             if libc is None:
                 libc = Libc.GLIBC
             if libc != Libc.GLIBC:
-                msg = f"can't repair wheel {wheel_filename} with {libc} libc to a wheel targeting GLIBC"
+                msg = f"can't repair wheel {wheel_filename} with {libc.name} libc to a wheel targeting GLIBC"
                 parser.error(msg)
         elif plat_base.startswith("musllinux"):
             if libc is None:
                 libc = Libc.MUSL
             if libc != Libc.MUSL:
-                msg = f"can't repair wheel {wheel_filename} with {libc} libc to a wheel targeting MUSL"
+                msg = f"can't repair wheel {wheel_filename} with {libc.name} libc to a wheel targeting MUSL"
                 parser.error(msg)
 
         logger.info("Repairing %s", wheel_filename)

tests/unit/test_main.py

@@ -5,6 +5,7 @@
 
 import pytest
 
+from auditwheel.libc import Libc, LibcVersion
 from auditwheel.main import main
 
 on_supported_platform = pytest.mark.skipif(
@@ -37,14 +38,58 @@ def test_help(monkeypatch, capsys):
     assert "usage: auditwheel [-h] [-V] [-v] command ..." in captured.out
 
 
-def test_show_unexisting_wheel(monkeypatch, capsys, tmp_path):
+@pytest.mark.parametrize("function", ["show", "repair"])
+def test_unexisting_wheel(monkeypatch, capsys, tmp_path, function):
     monkeypatch.setattr(sys, "platform", "linux")
     monkeypatch.setattr(platform, "machine", lambda: "x86_64")
     wheel = str(tmp_path / "not-a-file.whl")
-    monkeypatch.setattr(sys, "argv", ["dummy", "show", wheel])
+    monkeypatch.setattr(sys, "argv", ["auditwheel", function, wheel])
 
     with pytest.raises(SystemExit):
         main()
 
     captured = capsys.readouterr()
     assert "No such file" in captured.err
+
+
+@pytest.mark.parametrize(
+    ("libc", "filename", "plat", "message"),
+    [
+        (
+            Libc.GLIBC,
+            "foo-1.0-py3-none-manylinux1_aarch64.whl",
+            "manylinux_2_28_x86_64",
+            "can't repair wheel foo-1.0-py3-none-manylinux1_aarch64.whl with aarch64 architecture to a wheel targeting x86_64",
+        ),
+        (
+            Libc.GLIBC,
+            "foo-1.0-py3-none-musllinux_1_1_x86_64.whl",
+            "manylinux_2_28_x86_64",
+            "can't repair wheel foo-1.0-py3-none-musllinux_1_1_x86_64.whl with MUSL libc to a wheel targeting GLIBC",
+        ),
+        (
+            Libc.MUSL,
+            "foo-1.0-py3-none-manylinux1_x86_64.whl",
+            "musllinux_1_1_x86_64",
+            "can't repair wheel foo-1.0-py3-none-manylinux1_x86_64.whl with GLIBC libc to a wheel targeting MUSL",
+        ),
+    ],
+)
+def test_repair_wheel_mismatch(
+    monkeypatch, capsys, tmp_path, libc, filename, plat, message
+):
+    monkeypatch.setattr(sys, "platform", "linux")
+    monkeypatch.setattr(platform, "machine", lambda: "x86_64")
+    monkeypatch.setattr(Libc, "detect", lambda: libc)
+    monkeypatch.setattr(Libc, "get_current_version", lambda _: LibcVersion(1, 1))
+    wheel = tmp_path / filename
+    wheel.write_text("")
+    monkeypatch.setattr(
+        sys, "argv", ["auditwheel", "repair", "--plat", plat, str(wheel)]
+    )
+
+    with pytest.raises(SystemExit):
+        main()
+
+    captured = capsys.readouterr()
+    assert message in captured.err