feat: add ops center and node onboarding flow

This commit is contained in:
Your Name
2026-04-18 23:52:51 +08:00
parent 246838ae4c
commit b9c29481b5
142 changed files with 89727 additions and 186 deletions

View File

@@ -0,0 +1,118 @@
import subprocess
import tempfile
import unittest
from pathlib import Path
from unittest.mock import patch
from app.services.ops_runtime_executor_service import execute_local_support_action, execute_ssh_action
class OpsRuntimeExecutorServiceTests(unittest.TestCase):
@patch("app.services.ops_runtime_executor_service.subprocess.run")
def test_execute_ssh_action_parses_structured_json_result(self, mock_run) -> None:
mock_run.return_value = subprocess.CompletedProcess(
args=["ssh"],
returncode=0,
stdout='login banner\n{"ok": true, "message": "health snapshot collected", "result": {"checks": {"api": {"state": "active"}}}}\n',
stderr="",
)
ok, message, data = execute_ssh_action(
{
"node_code": "mainland-worker-01",
"ssh_host": "121.204.244.248",
"ssh_port": 22,
"ssh_user": "root",
},
"health.snapshot",
{},
)
self.assertTrue(ok)
self.assertEqual("health snapshot collected", message)
self.assertEqual("active", data["checks"]["api"]["state"])
self.assertEqual("ssh", data["transport"]["executor"])
self.assertEqual("121.204.244.248", data["transport"]["ssh_host"])
issued_command = mock_run.call_args.args[0]
self.assertEqual("ssh", issued_command[0])
self.assertIn("root@121.204.244.248", issued_command)
@patch("app.services.ops_runtime_executor_service.subprocess.run")
def test_execute_local_support_action_collects_logs_with_payload(self, mock_run) -> None:
mock_run.return_value = subprocess.CompletedProcess(
args=["journalctl"],
returncode=0,
stdout="log line 1\nlog line 2",
stderr="",
)
ok, message, data = execute_local_support_action(
"logs.collect",
{
"service_name": "domaincheck-worker",
"lines": 180,
},
)
self.assertTrue(ok)
self.assertEqual("log line 1\nlog line 2", message)
self.assertEqual("domaincheck-worker", data["service_name"])
self.assertEqual(180, data["lines"])
issued_command = mock_run.call_args.args[0]
self.assertEqual(
["journalctl", "-u", "domaincheck-worker", "-n", "180", "--no-pager"],
issued_command,
)
@patch("app.services.ops_runtime_executor_service.subprocess.run")
def test_execute_local_support_action_falls_back_to_worker_log_file(self, mock_run) -> None:
mock_run.return_value = subprocess.CompletedProcess(
args=["journalctl"],
returncode=1,
stdout="",
stderr="No journal files were opened due to insufficient permissions.",
)
with tempfile.TemporaryDirectory() as temp_dir:
log_path = Path(temp_dir) / "detect_worker.log"
log_path.write_text("fallback-1\nfallback-2\n", encoding="utf-8")
with patch("app.services.ops_action_executor_core.settings.domain_root", temp_dir):
ok, message, data = execute_local_support_action(
"logs.collect",
{
"service_name": "domaincheck-worker",
"lines": 180,
},
)
self.assertTrue(ok)
self.assertEqual("fallback-1\nfallback-2", message)
self.assertEqual("file_fallback", data["collection_source"])
self.assertTrue(data["fallback_used"])
@patch("app.services.ops_runtime_executor_service.subprocess.run")
def test_execute_local_support_action_restarts_service(self, mock_run) -> None:
mock_run.return_value = subprocess.CompletedProcess(
args=["systemctl"],
returncode=0,
stdout="",
stderr="",
)
ok, message, data = execute_local_support_action(
"service.restart",
{
"service_name": "domaincheck-worker",
},
)
self.assertTrue(ok)
self.assertEqual("domaincheck-worker restart completed", message)
self.assertEqual("domaincheck-worker", data["service_name"])
self.assertEqual("restart", data["systemctl_action"])
issued_command = mock_run.call_args.args[0]
self.assertEqual(["systemctl", "restart", "domaincheck-worker"], issued_command)
if __name__ == "__main__":
unittest.main()