40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
import os
|
|
import sys
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
import rizom_unwrap as ru
|
|
|
|
|
|
class TestCandidatePaths(unittest.TestCase):
|
|
def test_priority_order(self):
|
|
self.assertEqual(ru.candidate_paths("a.exe", "b.exe", "c.exe"),
|
|
["a.exe", "b.exe", "c.exe", ru.DEFAULT_RIZOM])
|
|
|
|
def test_none_entries_skipped(self):
|
|
self.assertEqual(ru.candidate_paths(None, None, None), [ru.DEFAULT_RIZOM])
|
|
|
|
def test_env_only(self):
|
|
self.assertEqual(ru.candidate_paths(None, "b.exe", None),
|
|
["b.exe", ru.DEFAULT_RIZOM])
|
|
|
|
|
|
class TestFindRizomuv(unittest.TestCase):
|
|
def test_missing_exe_raises(self):
|
|
with mock.patch.dict(os.environ):
|
|
os.environ.pop("RIZOMUV_EXE", None)
|
|
with self.assertRaises(ru.RizomError):
|
|
ru.find_rizomuv("Z:/no/such/rizomuv.exe", use_registry=False,
|
|
default="Z:/no/default.exe")
|
|
|
|
def test_registry_value_is_real_exe_if_present(self):
|
|
p = ru.registry_rizom_path()
|
|
if p is None:
|
|
self.skipTest("本机无 RizomUV 注册表项")
|
|
self.assertTrue(os.path.isfile(p), "注册表返回值应是真实 exe：%s" % p)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|