Added test_unmanaged_py_shebang

Author: Jesse205

tests/conftest.py

@@ -1,4 +1,5 @@
 import os
+from manage.tagutils import CompanyTag
 import pytest
 import random
 import re
@@ -158,8 +159,10 @@ def __init__(self, global_dir, installs=[]):
         self.shebang_can_run_anything_silently = False
         self.scratch = {}
 
-    def get_installs(self, *, include_unmanaged=True, set_default=True):
-        return self.installs
+    def get_installs(self, *, include_unmanaged=False, set_default=True):
+        if include_unmanaged:
+            return self.installs
+        return [i for i in self.installs if not i.get("unmanaged", 0)]
 
     def get_install_to_run(self, tag, *, windowed=False):
         if windowed:
@@ -171,7 +174,7 @@ def get_install_to_run(self, tag, *, windowed=False):
 
         company, _, tag = tag.replace("/", "\\").rpartition("\\")
         return [i for i in self.installs
-                if i["tag"] == tag and (not company or i["company"] == company)][0]
+                if (not tag or i["tag"] == tag) and (not company or i["company"] == company)][0]
 
 
 @pytest.fixture

tests/test_scriptutils.py

@@ -157,6 +157,25 @@ def t(n):
     assert t("pythonw1.0")["executable"].match("pythonw.exe")
 
 
+def test_unmanaged_py_shebang(fake_config, tmp_path):
+    inst = _fake_install("1.0", company="PythonCore", prefix=PurePath("C:\\TestRoot"))
+    inst["unmanaged"] = 1
+    inst["run-for"] = [
+        dict(name="python.exe", target=".\\python.exe"),
+        dict(name="pythonw.exe", target=".\\pythonw.exe", windowed=1),
+    ]
+    fake_config.installs[:] = [inst]
+
+    def t(n):
+        return _find_shebang_command(fake_config, n, windowed=False)
+
+    # Finds the install's default executable
+    assert t("python")["executable"].match("test-binary-1.0.exe")
+    assert t("python1.0")["executable"].match("test-binary-1.0.exe")
+    # Finds the install's run-for executable with windowed=1
+    assert t("pythonw")["executable"].match("pythonw.exe")
+    assert t("pythonw1.0")["executable"].match("pythonw.exe")
+
 
 @pytest.mark.parametrize("script, expect", [
     ("# not a coding comment", None),