6d8c18cd1a* Replace standalone Studio wording with Unsloth Replace the single word Studio with Unsloth wherever it is used as shorthand for Unsloth Studio in docs, CLI output, UI strings, i18n locales, workflow display names, comments and docstrings. Kept unchanged: the full name Unsloth Studio, third party product names (LM Studio, Visual Studio, Mac Studio), feature names (Recipe Studio, Fine-tuning Studio and its translations), and all identifiers such as env vars, commands, paths and filenames. * Address review feedback on the Studio wording rename Use "an" before Unsloth where the rename left the article as "a". Restore the split brand where Unsloth and Studio render as two halves of the full product name: the onboarding sidebar subtitle and the IPv6 localhost warning. Scope two messages to the full name Unsloth Studio where plain Unsloth was misleading: the AMD README bullet and the CLI studio setup error.
62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
"""Guard install.ps1's Unsloth launcher against the AV-heuristic shape (Kaspersky
|
|
HEUR:Trojan.VBS.Agent.gen): a WScript .vbs spawning a hidden ExecutionPolicy-Bypass PowerShell.
|
|
The shortcut must stay windowless via powershell.exe -WindowStyle Hidden over launch-studio.ps1,
|
|
never a .vbs/WScript.Shell.Run wrapper, and any pre-existing .vbs must be deleted on upgrade."""
|
|
|
|
import re
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[3]
|
|
INSTALL_PS1 = REPO_ROOT / "install.ps1"
|
|
|
|
|
|
def _text() -> str:
|
|
return INSTALL_PS1.read_text(encoding = "utf-8")
|
|
|
|
|
|
def test_install_ps1_present():
|
|
assert INSTALL_PS1.is_file(), f"missing {INSTALL_PS1}"
|
|
|
|
|
|
def test_no_vbs_launcher_generated():
|
|
text = _text()
|
|
# No here-string building a .vbs body, no .vbs written (legacy cleanup checked separately).
|
|
assert "$vbsContent" not in text, (
|
|
"install.ps1 must not generate a launch-studio.vbs: a WScript.Shell .vbs "
|
|
"spawning a hidden ExecutionPolicy-Bypass PowerShell is the exact shape "
|
|
"VBS-dropper heuristics flag (Kaspersky HEUR:Trojan.VBS.Agent.gen)."
|
|
)
|
|
assert 'CreateObject("WScript.Shell")' not in text
|
|
assert "shell.Run" not in text
|
|
assert not re.search(r"Set-Content\s+-LiteralPath\s+\$launcherVbs", text)
|
|
assert "//B //Nologo" not in text
|
|
|
|
|
|
def test_legacy_vbs_removed_on_upgrade():
|
|
# An upgrade must DELETE a pre-existing launch-studio.vbs, not just stop generating it,
|
|
# or AV keeps flagging the stale file.
|
|
text = _text()
|
|
assert re.search(
|
|
r"Remove-Item\s+-LiteralPath\s+\$legacyLauncherVbs", text
|
|
), "upgrades must remove a pre-existing launch-studio.vbs so AV stops flagging it"
|
|
|
|
|
|
def test_shortcut_target_is_not_wscript():
|
|
# The .lnk must not launch through wscript.exe (the VBS script host).
|
|
text = _text()
|
|
assert "wscript.exe" not in text.lower()
|
|
|
|
|
|
def test_launcher_is_windowless_powershell():
|
|
# The shortcut runs powershell.exe -WindowStyle Hidden over launch-studio.ps1.
|
|
text = _text()
|
|
assert re.search(
|
|
r"-WindowStyle\s+Hidden", text
|
|
), "the launcher must run powershell.exe with -WindowStyle Hidden over launch-studio.ps1."
|
|
assert "launch-studio.ps1" in text
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(pytest.main([__file__, "-v"]))
|