Add tests

Author: zooba

src/manage/install_command.py

@@ -216,7 +216,7 @@ def _calc(prefix, filename, calculate_dest=calculate_dest):
 
 
 def _if_exists(launcher, plat):
-    plat_launcher = launcher.parent / f"{launcher.stem}{plat}.exe"
+    plat_launcher = launcher.parent / f"{launcher.stem}{plat}{launcher.suffix}"
     if plat_launcher.is_file():
         return plat_launcher
     return launcher
@@ -238,7 +238,7 @@ def _write_alias(cmd, install, alias, target):
         launcher = _if_exists(launcher, cmd.default_platform)
     if not launcher.is_file():
         LOGGER.debug("Checking for launcher for -64")
-        launcher = _if_exists(launcher, "64")
+        launcher = _if_exists(launcher, "-64")
     LOGGER.debug("Create %s linking to %s using %s", alias["name"], target, launcher)
     if not launcher or not launcher.is_file():
         LOGGER.warn("Skipping %s alias because the launcher template was not found.", alias["name"])

tests/test_install_command.py

@@ -0,0 +1,97 @@
+import pytest
+import secrets
+from manage import install_command as IC
+
+
+@pytest.fixture
+def alias_checker(tmp_path):
+    with AliasChecker(tmp_path) as checker:
+        yield checker
+
+class AliasChecker:
+    class Cmd:
+        global_dir = "out"
+        launcher_exe = "launcher.txt"
+        launcherw_exe = "launcherw.txt"
+        default_platform = "-64"
+
+        def __init__(self, platform=None):
+            if platform:
+                self.default_platform = platform
+
+
+    def __init__(self, tmp_path):
+        self.Cmd.global_dir = tmp_path / "out"
+        self.Cmd.launcher_exe = tmp_path / "launcher.txt"
+        self.Cmd.launcherw_exe = tmp_path / "launcherw.txt"
+        self._expect_target = "target-" + secrets.token_hex(32)
+        self._expect = {
+            "-32": "-32-" + secrets.token_hex(32),
+            "-64": "-64-" + secrets.token_hex(32),
+            "-arm64": "-arm64-" + secrets.token_hex(32),
+            "w-32": "w-32-" + secrets.token_hex(32),
+            "w-64": "w-64-" + secrets.token_hex(32),
+            "w-arm64": "w-arm64-" + secrets.token_hex(32),
+        }
+        for k, v in self._expect.items():
+            (tmp_path / f"launcher{k}.txt").write_text(v)
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *exc_info):
+        pass
+
+    def check(self, cmd, tag, name, expect, windowed=0):
+        IC._write_alias(
+            cmd,
+            {"tag": tag},
+            {"name": f"{name}.txt", "windowed": windowed},
+            self._expect_target,
+        )
+        print(*cmd.global_dir.glob("*"), sep="\n")
+        assert (cmd.global_dir / f"{name}.txt").is_file()
+        assert (cmd.global_dir / f"{name}.txt.__target__").is_file()
+        assert (cmd.global_dir / f"{name}.txt").read_text() == expect
+        assert (cmd.global_dir / f"{name}.txt.__target__").read_text() == self._expect_target
+
+    def check_32(self, cmd, tag, name):
+        self.check(cmd, tag, name, self._expect["-32"])
+
+    def check_w32(self, cmd, tag, name):
+        self.check(cmd, tag, name, self._expect["w-32"], windowed=1)
+
+    def check_64(self, cmd, tag, name):
+        self.check(cmd, tag, name, self._expect["-64"])
+
+    def check_w64(self, cmd, tag, name):
+        self.check(cmd, tag, name, self._expect["w-64"], windowed=1)
+
+    def check_arm64(self, cmd, tag, name):
+        self.check(cmd, tag, name, self._expect["-arm64"])
+
+    def check_warm64(self, cmd, tag, name):
+        self.check(cmd, tag, name, self._expect["w-arm64"], windowed=1)
+
+
+def test_write_alias_tag_with_platform(alias_checker):
+    alias_checker.check_32(alias_checker.Cmd, "1.0-32", "testA")
+    alias_checker.check_w32(alias_checker.Cmd, "1.0-32", "testB")
+    alias_checker.check_64(alias_checker.Cmd, "1.0-64", "testC")
+    alias_checker.check_w64(alias_checker.Cmd, "1.0-64", "testD")
+    alias_checker.check_arm64(alias_checker.Cmd, "1.0-arm64", "testE")
+    alias_checker.check_warm64(alias_checker.Cmd, "1.0-arm64", "testF")
+
+
+def test_write_alias_default_platform(alias_checker):
+    alias_checker.check_32(alias_checker.Cmd("-32"), "1.0", "testA")
+    alias_checker.check_w32(alias_checker.Cmd("-32"), "1.0", "testB")
+    alias_checker.check_64(alias_checker.Cmd, "1.0", "testC")
+    alias_checker.check_w64(alias_checker.Cmd, "1.0", "testD")
+    alias_checker.check_arm64(alias_checker.Cmd("-arm64"), "1.0", "testE")
+    alias_checker.check_warm64(alias_checker.Cmd("-arm64"), "1.0", "testF")
+
+
+def test_write_alias_fallback_platform(alias_checker):
+    alias_checker.check_64(alias_checker.Cmd("-spam"), "1.0", "testA")
+    alias_checker.check_w64(alias_checker.Cmd("-spam"), "1.0", "testB")