diff options
Diffstat (limited to 'tests/unit/modules')
4 files changed, 4 insertions, 4 deletions
diff --git a/tests/unit/modules/network/vyos/test_vyos_interface.py b/tests/unit/modules/network/vyos/test_vyos_interface.py index 1194e3bd..e0e4bfc7 100644 --- a/tests/unit/modules/network/vyos/test_vyos_interface.py +++ b/tests/unit/modules/network/vyos/test_vyos_interface.py @@ -86,7 +86,7 @@ class TestVyosInterfaceModule(TestVyosModule): self.mock_execute_lldp_command.stop() self.mock_execute_interfaces_command.stop() - def load_fixtures(self, commands=None, transport="cli"): + def load_fixtures(self, commands=None, transport="cli", filename=None): self.get_config.return_value = load_fixture( "vyos_interface_config.cfg" ) diff --git a/tests/unit/modules/network/vyos/test_vyos_ping.py b/tests/unit/modules/network/vyos/test_vyos_ping.py index f3ba2ee5..60407ad6 100644 --- a/tests/unit/modules/network/vyos/test_vyos_ping.py +++ b/tests/unit/modules/network/vyos/test_vyos_ping.py @@ -49,7 +49,7 @@ class TestVyosPingModule(TestVyosModule): output = list() for command in commands: - filename = str(command).split(" | ")[0].replace(" ", "_") + filename = str(command).split(" | ", 1)[0].replace(" ", "_") output.append(load_fixture("vyos_ping_%s" % filename)) return output diff --git a/tests/unit/modules/network/vyos/test_vyos_static_route.py b/tests/unit/modules/network/vyos/test_vyos_static_route.py index 21f11394..00985c6a 100644 --- a/tests/unit/modules/network/vyos/test_vyos_static_route.py +++ b/tests/unit/modules/network/vyos/test_vyos_static_route.py @@ -52,7 +52,7 @@ class TestVyosStaticRouteModule(TestVyosModule): self.mock_get_config.stop() self.mock_load_config.stop() - def load_fixtures(self, commands=None, transport="cli"): + def load_fixtures(self, commands=None, transport="cli", filename=None): self.get_config.return_value = "" self.load_config.return_value = dict(diff=None, session="session") diff --git a/tests/unit/modules/network/vyos/test_vyos_user.py b/tests/unit/modules/network/vyos/test_vyos_user.py index d4c2dbe6..616ad095 100644 --- a/tests/unit/modules/network/vyos/test_vyos_user.py +++ b/tests/unit/modules/network/vyos/test_vyos_user.py @@ -50,7 +50,7 @@ class TestVyosUserModule(TestVyosModule): self.mock_get_config.stop() self.mock_load_config.stop() - def load_fixtures(self, commands=None, transport="cli"): + def load_fixtures(self, commands=None, transport="cli", filename=None): self.get_config.return_value = load_fixture("vyos_user_config.cfg") self.load_config.return_value = dict(diff=None, session="session") |