mirror of
https://github.com/kunkundi/crossdesk.git
synced 2026-10-17 01:13:19 +08:00
85 lines
4.1 KiB
Python
85 lines
4.1 KiB
Python
"""Validate generated Task Scheduler XML on macOS/Linux without changing startup.
|
|
|
|
Run: python3 apps/desktop/tests/windows_autostart_task_test.py
|
|
Native registration/migration coverage lives in windows_autostart_test.cpp.
|
|
"""
|
|
import pathlib
|
|
import shutil
|
|
import subprocess
|
|
import tempfile
|
|
import unittest
|
|
import xml.etree.ElementTree as ET
|
|
|
|
ROOT = pathlib.Path(__file__).resolve().parents[3]
|
|
NAMESPACE = {"t": "http://schemas.microsoft.com/windows/2004/02/mit/task"}
|
|
|
|
|
|
class AutostartTaskTest(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.temporary = tempfile.TemporaryDirectory(prefix="crossdesk-autostart-test-")
|
|
cls.addClassCleanup(cls.temporary.cleanup)
|
|
directory = pathlib.Path(cls.temporary.name)
|
|
source = directory / "fixture.cpp"
|
|
source.write_text(r'''
|
|
#include <iostream>
|
|
#include "platform/windows/autostart_task.h"
|
|
int main() {
|
|
std::cout << crossdesk::platform::kAutostartRequiresElevation << '\n';
|
|
const auto xml = crossdesk::platform::BuildAutostartTaskXml(
|
|
L"S-1-5-21-123-456-789-1001", L"C:\\程序 & Tools\\CrossDesk.exe",
|
|
L"C:\\程序 & Tools");
|
|
for (wchar_t character : xml) std::cout << static_cast<unsigned>(character) << ' ';
|
|
}
|
|
''', encoding="utf-8")
|
|
compiler = shutil.which("c++") or shutil.which("clang++")
|
|
if not compiler:
|
|
raise unittest.SkipTest("C++17 compiler is unavailable")
|
|
cls.variants = {}
|
|
for name, defines in {
|
|
"installed": [], "portable": ["-DCROSSDESK_PORTABLE=1"],
|
|
"debug": ["-DCROSSDESK_DEBUG"],
|
|
"portable_debug": ["-DCROSSDESK_PORTABLE=1", "-DCROSSDESK_DEBUG"],
|
|
}.items():
|
|
binary = directory / name
|
|
subprocess.run([compiler, "-std=c++17", "-Wall", "-Wextra", "-Werror",
|
|
"-I", str(ROOT / "apps/desktop/src"), *defines,
|
|
str(source), "-o", str(binary)], check=True)
|
|
policy, codepoints = subprocess.check_output([str(binary)], text=True).split("\n", 1)
|
|
xml = "".join(chr(int(value)) for value in codepoints.split())
|
|
cls.variants[name] = (policy, ET.fromstring(xml))
|
|
|
|
def text(self, path):
|
|
return self.variants["installed"][1].findtext(path, namespaces=NAMESPACE)
|
|
|
|
def test_privilege_policy_matches_build_type(self):
|
|
self.assertEqual({name: result[0] for name, result in self.variants.items()},
|
|
{"installed": "1", "portable": "0", "debug": "0", "portable_debug": "0"})
|
|
|
|
def test_login_targets_same_interactive_user_without_password(self):
|
|
sid = "S-1-5-21-123-456-789-1001"
|
|
self.assertEqual(self.text("t:Triggers/t:LogonTrigger/t:UserId"), sid)
|
|
self.assertEqual(self.text("t:Principals/t:Principal/t:UserId"), sid)
|
|
self.assertEqual(self.text("t:Principals/t:Principal/t:LogonType"), "InteractiveToken")
|
|
self.assertEqual(self.text("t:Principals/t:Principal/t:RunLevel"), "HighestAvailable")
|
|
self.assertEqual(self.text("t:Triggers/t:LogonTrigger/t:Enabled"), "true")
|
|
|
|
def test_command_is_not_shell_interpreted_and_unicode_round_trips(self):
|
|
self.assertEqual(self.text("t:Actions/t:Exec/t:Command"), r"C:\程序 & Tools\CrossDesk.exe")
|
|
self.assertEqual(self.text("t:Actions/t:Exec/t:WorkingDirectory"), r"C:\程序 & Tools")
|
|
self.assertIsNone(self.text("t:Actions/t:Exec/t:Arguments"))
|
|
task = self.variants["installed"][1]
|
|
self.assertEqual(task.find("t:Actions", NAMESPACE).get("Context"),
|
|
task.find("t:Principals/t:Principal", NAMESPACE).get("id"))
|
|
|
|
def test_remote_client_can_run_indefinitely_on_battery(self):
|
|
self.assertEqual(self.text("t:Triggers/t:LogonTrigger/t:ExecutionTimeLimit"), "PT0S")
|
|
self.assertEqual(self.text("t:Settings/t:DisallowStartIfOnBatteries"), "false")
|
|
self.assertEqual(self.text("t:Settings/t:StopIfGoingOnBatteries"), "false")
|
|
self.assertEqual(self.text("t:Settings/t:ExecutionTimeLimit"), "PT0S")
|
|
self.assertEqual(self.text("t:Settings/t:MultipleInstancesPolicy"), "IgnoreNew")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|