104 lines
3.8 KiB
Python
104 lines
3.8 KiB
Python
|
import subprocess
|
||
|
|
import tempfile
|
||
|
|
import unittest
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from deploy_tool.environment import check_environment, load_tool_paths, save_tool_paths
|
||
|
|
|
||
|
|
|
||
|
|
class EnvironmentTests(unittest.TestCase):
|
||
|
|
def test_reports_installed_command(self):
|
||
|
|
def runner(command):
|
||
|
|
return subprocess.CompletedProcess(command, 0, "java 21\n", "")
|
||
|
|
|
||
|
|
results = check_environment(runner=runner, checks={"JDK": ["java", "-version"]})
|
||
|
|
|
||
|
|
self.assertTrue(results[0].installed)
|
||
|
|
self.assertEqual(results[0].name, "JDK")
|
||
|
|
self.assertEqual(results[0].source, "PATH")
|
||
|
|
self.assertIn("java 21", results[0].version)
|
||
|
|
|
||
|
|
def test_reports_missing_command(self):
|
||
|
|
def runner(command):
|
||
|
|
raise FileNotFoundError(command[0])
|
||
|
|
|
||
|
|
results = check_environment(runner=runner, checks={"MySQL": ["mysql", "--version"]})
|
||
|
|
|
||
|
|
self.assertFalse(results[0].installed)
|
||
|
|
self.assertEqual(results[0].name, "MySQL")
|
||
|
|
self.assertIn("未安装", results[0].message)
|
||
|
|
|
||
|
|
def test_uses_manual_path_when_command_is_not_in_path(self):
|
||
|
|
with tempfile.TemporaryDirectory() as tmp:
|
||
|
|
mysql = Path(tmp) / "mysql.exe"
|
||
|
|
mysql.write_text("", encoding="utf-8")
|
||
|
|
|
||
|
|
def runner(command):
|
||
|
|
if command[0] == "mysql":
|
||
|
|
raise FileNotFoundError(command[0])
|
||
|
|
return subprocess.CompletedProcess(command, 0, "mysql 8.4\n", "")
|
||
|
|
|
||
|
|
results = check_environment(
|
||
|
|
runner=runner,
|
||
|
|
checks={"MySQL": ["mysql", "--version"]},
|
||
|
|
manual_paths={"MySQL": mysql},
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertTrue(results[0].installed)
|
||
|
|
self.assertEqual(results[0].source, "手动指定")
|
||
|
|
self.assertEqual(results[0].command[0], str(mysql))
|
||
|
|
|
||
|
|
def test_discovers_executable_from_known_search_paths(self):
|
||
|
|
with tempfile.TemporaryDirectory() as tmp:
|
||
|
|
mysql = Path(tmp) / "MySQL" / "MySQL Server 8.4" / "bin" / "mysql.exe"
|
||
|
|
mysql.parent.mkdir(parents=True)
|
||
|
|
mysql.write_text("", encoding="utf-8")
|
||
|
|
|
||
|
|
def runner(command):
|
||
|
|
if command[0] == "mysql":
|
||
|
|
raise FileNotFoundError(command[0])
|
||
|
|
return subprocess.CompletedProcess(command, 0, "mysql 8.4\n", "")
|
||
|
|
|
||
|
|
results = check_environment(
|
||
|
|
runner=runner,
|
||
|
|
checks={"MySQL": ["mysql", "--version"]},
|
||
|
|
search_roots=[Path(tmp)],
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertTrue(results[0].installed)
|
||
|
|
self.assertEqual(results[0].source, "自动发现")
|
||
|
|
self.assertEqual(results[0].command[0], str(mysql))
|
||
|
|
|
||
|
|
def test_does_not_recursively_scan_arbitrary_deep_directories(self):
|
||
|
|
with tempfile.TemporaryDirectory() as tmp:
|
||
|
|
mysql = Path(tmp) / "random" / "deep" / "folder" / "mysql.exe"
|
||
|
|
mysql.parent.mkdir(parents=True)
|
||
|
|
mysql.write_text("", encoding="utf-8")
|
||
|
|
|
||
|
|
def runner(command):
|
||
|
|
if command[0] == "mysql":
|
||
|
|
raise FileNotFoundError(command[0])
|
||
|
|
return subprocess.CompletedProcess(command, 0, "mysql 8.4\n", "")
|
||
|
|
|
||
|
|
results = check_environment(
|
||
|
|
runner=runner,
|
||
|
|
checks={"MySQL": ["mysql", "--version"]},
|
||
|
|
search_roots=[Path(tmp)],
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertFalse(results[0].installed)
|
||
|
|
|
||
|
|
def test_saves_and_loads_manual_paths(self):
|
||
|
|
with tempfile.TemporaryDirectory() as tmp:
|
||
|
|
config = Path(tmp) / "tool-paths.json"
|
||
|
|
mysql = Path(tmp) / "mysql.exe"
|
||
|
|
|
||
|
|
save_tool_paths({"MySQL": mysql}, config)
|
||
|
|
loaded = load_tool_paths(config)
|
||
|
|
|
||
|
|
self.assertEqual(loaded, {"MySQL": mysql})
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|