From 710e61bc17a8354c824dba4ea53a71013fe1b0e1 Mon Sep 17 00:00:00 2001 From: omnom62 Date: Mon, 1 Jun 2026 21:33:44 +1000 Subject: Test framework updates --- tests/unit/modules/__init__.py | 0 tests/unit/modules/test_vyos_ntp_global.py | 22 +++++++++++++++++----- tests/unit/modules/test_vyos_route_maps.py | 21 +++++++++++++++++---- 3 files changed, 34 insertions(+), 9 deletions(-) create mode 100644 tests/unit/modules/__init__.py (limited to 'tests/unit/modules') diff --git a/tests/unit/modules/__init__.py b/tests/unit/modules/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/unit/modules/test_vyos_ntp_global.py b/tests/unit/modules/test_vyos_ntp_global.py index 9aac605..c536141 100644 --- a/tests/unit/modules/test_vyos_ntp_global.py +++ b/tests/unit/modules/test_vyos_ntp_global.py @@ -4,13 +4,11 @@ from __future__ import absolute_import, division, print_function __metaclass__ = type +import json import os -import sys import unittest - -# Allow importing collection modules without full ansible-test runner -sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "..", "..", "..")) +from unittest.mock import MagicMock from ansible_collections.vyos.rest.plugins.modules.vyos_ntp_global import ( build_commands, @@ -19,7 +17,21 @@ from ansible_collections.vyos.rest.plugins.modules.vyos_ntp_global import ( normalize_servers, ) -from tests.unit.modules.base import VyOSModuleTestCase, load_fixture + +def load_fixture(filename): + fixtures_dir = os.path.join(os.path.dirname(__file__), "..", "fixtures") + path = os.path.join(fixtures_dir, filename) + with open(path) 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 TestVyOSNtpGlobalNormalize(unittest.TestCase): diff --git a/tests/unit/modules/test_vyos_route_maps.py b/tests/unit/modules/test_vyos_route_maps.py index 2d45bb7..97d814b 100644 --- a/tests/unit/modules/test_vyos_route_maps.py +++ b/tests/unit/modules/test_vyos_route_maps.py @@ -4,12 +4,11 @@ from __future__ import absolute_import, division, print_function __metaclass__ = type +import json import os -import sys import unittest - -sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "..", "..", "..")) +from unittest.mock import MagicMock from ansible_collections.vyos.rest.plugins.modules.vyos_route_maps import ( _want_to_api_match, @@ -18,7 +17,21 @@ from ansible_collections.vyos.rest.plugins.modules.vyos_route_maps import ( get_running_config, ) -from tests.unit.modules.base import VyOSModuleTestCase, load_fixture + +def load_fixture(filename): + fixtures_dir = os.path.join(os.path.dirname(__file__), "..", "fixtures") + path = os.path.join(fixtures_dir, filename) + with open(path) 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 TestVyOSRouteMapsGetRunning(VyOSModuleTestCase): -- cgit v1.2.3