125 lines
3.9 KiB
Python
125 lines
3.9 KiB
Python
# Copyright (C) 2025 Cetmix OÜ
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
|
|
|
|
"""
|
|
Tests for cx.tower.server YAML export/import covering command_ids and plan_ids.
|
|
"""
|
|
import yaml
|
|
|
|
from odoo.tests import TransactionCase, tagged
|
|
|
|
|
|
@tagged("post_install", "-at_install")
|
|
class TestServerYAML(TransactionCase):
|
|
"""YAML export/import tests for cx.tower.server with commands and plans."""
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super().setUpClass()
|
|
env = cls.env
|
|
cls.Server = env["cx.tower.server"]
|
|
cls.Command = env["cx.tower.command"]
|
|
cls.Plan = env["cx.tower.plan"]
|
|
|
|
# Create a command to attach (use defaults for access_level)
|
|
cls.command = cls.Command.create(
|
|
{
|
|
"name": "Test Command",
|
|
"reference": "test_command",
|
|
"action": "ssh_command",
|
|
"allow_parallel_run": False,
|
|
}
|
|
)
|
|
|
|
# Create a flight plan to attach
|
|
cls.plan = cls.Plan.create(
|
|
{
|
|
"name": "Test Flight Plan",
|
|
"reference": "test_plan",
|
|
"allow_parallel_run": False,
|
|
"color": 0,
|
|
}
|
|
)
|
|
|
|
# Create server and link command and plan
|
|
cls.server = cls.Server.create(
|
|
{
|
|
"name": "Server YAML Test",
|
|
"reference": "srv_yaml_test",
|
|
"ip_v4_address": "127.0.0.1",
|
|
"ssh_username": "admin",
|
|
"ssh_port": 22,
|
|
"ssh_auth_mode": "p",
|
|
"ssh_password": "dummy",
|
|
"use_sudo": False,
|
|
# Link the m2m fields
|
|
"command_ids": [(6, 0, [cls.command.id])],
|
|
"plan_ids": [(6, 0, [cls.plan.id])],
|
|
}
|
|
)
|
|
|
|
def test_yaml_export_contains_command_and_plan(self):
|
|
"""Exported YAML include command_ids and plan_ids with correct references."""
|
|
data = yaml.safe_load(self.server.yaml_code)
|
|
# Check command_ids
|
|
self.assertIn(
|
|
"command_ids",
|
|
data,
|
|
"`command_ids` is missing from YAML export",
|
|
)
|
|
self.assertIsInstance(
|
|
data["command_ids"], list, "`command_ids` should be a list in YAML"
|
|
)
|
|
self.assertTrue(
|
|
data["command_ids"],
|
|
"`command_ids` list should not be empty",
|
|
)
|
|
# Only reference should be exported
|
|
self.assertEqual(
|
|
data["command_ids"][0],
|
|
self.command.reference,
|
|
"Exported command reference does not match",
|
|
)
|
|
|
|
# Check plan_ids
|
|
self.assertIn(
|
|
"plan_ids",
|
|
data,
|
|
"`plan_ids` is missing from YAML export",
|
|
)
|
|
self.assertIsInstance(
|
|
data["plan_ids"], list, "`plan_ids` should be a list in YAML"
|
|
)
|
|
self.assertTrue(
|
|
data["plan_ids"],
|
|
"`plan_ids` list should not be empty",
|
|
)
|
|
self.assertEqual(
|
|
data["plan_ids"][0],
|
|
self.plan.reference,
|
|
"Exported plan reference does not match",
|
|
)
|
|
|
|
def test_yaml_roundtrip_restores_command_and_plan(self):
|
|
"""A full export→delete→import cycle must restore the m2m relations."""
|
|
yaml_dict = yaml.safe_load(self.server.yaml_code)
|
|
# Remove original server
|
|
self.server.unlink()
|
|
# Prepare values and import
|
|
vals = self.Server._post_process_yaml_dict_values(yaml_dict)
|
|
restored = self.Server.with_context(
|
|
from_yaml=True, skip_ssh_settings_check=True
|
|
).create(vals)
|
|
|
|
# Verify m2m links restored
|
|
self.assertEqual(
|
|
restored.command_ids.ids,
|
|
[self.command.id],
|
|
"`command_ids` were not restored correctly",
|
|
)
|
|
self.assertEqual(
|
|
restored.plan_ids.ids,
|
|
[self.plan.id],
|
|
"`plan_ids` were not restored correctly",
|
|
)
|