T8989: ospf_v3 module

1 files changed, 140 insertions, 0 deletions

diff --git a/tests/unit/modules/test_vyos_ospfv3.py b/tests/unit/modules/test_vyos_ospfv3.py
new file mode 100644
index 0000000..a84041b
--- /dev/null
+++ b/tests/unit/modules/test_vyos_ospfv3.py
@@ -0,0 +1,140 @@
+# -*- coding: utf-8 -*-
+from __future__ import absolute_import, division, print_function
+
+
+__metaclass__ = type
+
+import json
+import os
+import unittest
+
+from unittest.mock import MagicMock
+
+from ansible_collections.vyos.rest.plugins.modules.vyos_ospfv3 import (
+    build_commands,
+    get_running_config,
+)
+
+
+def load_fixture(filename):
+    fixtures_dir = os.path.join(os.path.dirname(__file__), "..", "fixtures")
+    with open(os.path.join(fixtures_dir, filename)) as f:
+        return json.load(f)
+
+
+class VyOSModuleTestCase(unittest.TestCase):
+    def setUp(self):
+        self.mock_vyos = MagicMock()
+        self.mock_vyos.get_config = MagicMock(return_value={})
+
+    def set_running_config(self, data):
+        self.mock_vyos.get_config.return_value = data
+
+
+class TestVyOSOspfv3Parse(VyOSModuleTestCase):
+
+    def setUp(self):
+        super().setUp()
+        self.fixture = load_fixture("ospfv3_running.json")
+
+    def test_parses_parameters(self):
+        self.set_running_config(self.fixture)
+        result = get_running_config(self.mock_vyos)
+        self.assertEqual(result["parameters"]["router_id"], "192.0.2.10")
+
+    def test_parses_redistribute(self):
+        self.set_running_config(self.fixture)
+        result = get_running_config(self.mock_vyos)
+        route_types = [r["route_type"] for r in result["redistribute"]]
+        self.assertIn("bgp", route_types)
+        self.assertIn("connected", route_types)
+        connected = next(r for r in result["redistribute"] if r["route_type"] == "connected")
+        self.assertEqual(connected["route_map"], "RM1")
+
+    def test_parses_areas(self):
+        self.set_running_config(self.fixture)
+        result = get_running_config(self.mock_vyos)
+        self.assertEqual(len(result["areas"]), 2)
+        area2 = next(a for a in result["areas"] if a["area_id"] == "2")
+        self.assertEqual(area2["export_list"], "export1")
+        self.assertEqual(area2["import_list"], "import1")
+        self.assertEqual(len(area2["range"]), 2)
+        not_adv = next(r for r in area2["range"] if r["address"] == "2001:db20::/32")
+        self.assertTrue(not_adv["not_advertise"])
+
+    def test_empty_config_returns_empty(self):
+        self.set_running_config({})
+        result = get_running_config(self.mock_vyos)
+        self.assertEqual(result, {})
+
+
+class TestVyOSOspfv3BuildCommands(unittest.TestCase):
+
+    def test_deleted_with_have(self):
+        have = {"parameters": {"router_id": "192.0.2.10"}}
+        cmds = build_commands({}, have, "deleted")
+        self.assertEqual(cmds, [("delete", ["protocols", "ospfv3"])])
+
+    def test_deleted_without_have(self):
+        cmds = build_commands({}, {}, "deleted")
+        self.assertEqual(cmds, [])
+
+    def test_merged_parameters(self):
+        config = {"parameters": {"router_id": "192.0.2.10"}}
+        cmds = build_commands(config, {}, "merged")
+        self.assertIn(
+            ("set", ["protocols", "ospfv3", "parameters", "router-id", "192.0.2.10"]),
+            cmds,
+        )
+
+    def test_merged_redistribute(self):
+        config = {"redistribute": [{"route_type": "bgp"}]}
+        cmds = build_commands(config, {}, "merged")
+        self.assertIn(
+            ("set", ["protocols", "ospfv3", "redistribute", "bgp"]),
+            cmds,
+        )
+
+    def test_merged_idempotent(self):
+        config = {"parameters": {"router_id": "192.0.2.10"}}
+        have = {"parameters": {"router_id": "192.0.2.10"}}
+        cmds = build_commands(config, have, "merged")
+        self.assertEqual(cmds, [])
+
+    def test_merged_area_range(self):
+        config = {
+            "areas": [{"area_id": "2", "range": [{"address": "2001:db10::/32"}]}],
+        }
+        cmds = build_commands(config, {}, "merged")
+        self.assertIn(
+            ("set", ["protocols", "ospfv3", "area", "2", "range", "2001:db10::/32"]),
+            cmds,
+        )
+
+    def test_replaced_idempotent(self):
+        config = {"parameters": {"router_id": "192.0.2.10"}}
+        have = {"parameters": {"router_id": "192.0.2.10"}}
+        cmds = build_commands(config, have, "replaced")
+        self.assertEqual(cmds, [])
+
+    def test_replaced_rebuilds_on_change(self):
+        config = {"parameters": {"router_id": "192.0.2.11"}}
+        have = {"parameters": {"router_id": "192.0.2.10"}}
+        cmds = build_commands(config, have, "replaced")
+        self.assertEqual(cmds[0], ("delete", ["protocols", "ospfv3"]))
+        self.assertIn(
+            ("set", ["protocols", "ospfv3", "parameters", "router-id", "192.0.2.11"]),
+            cmds,
+        )
+
+    def test_merged_area_export_list(self):
+        config = {"areas": [{"area_id": "2", "export_list": "export1"}]}
+        cmds = build_commands(config, {}, "merged")
+        self.assertIn(
+            ("set", ["protocols", "ospfv3", "area", "2", "export-list", "export1"]),
+            cmds,
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()