diff options
Diffstat (limited to 'test/units/modules/network/iosxr/test_iosxr_user.py')
-rw-r--r-- | test/units/modules/network/iosxr/test_iosxr_user.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/test/units/modules/network/iosxr/test_iosxr_user.py b/test/units/modules/network/iosxr/test_iosxr_user.py index 7f392e7a1c..2254249498 100644 --- a/test/units/modules/network/iosxr/test_iosxr_user.py +++ b/test/units/modules/network/iosxr/test_iosxr_user.py @@ -19,11 +19,10 @@ from __future__ import (absolute_import, division, print_function) __metaclass__ = type -import json - from ansible.compat.tests.mock import patch from ansible.modules.network.iosxr import iosxr_user -from .iosxr_module import TestIosxrModule, load_fixture, set_module_args +from units.modules.utils import set_module_args +from .iosxr_module import TestIosxrModule, load_fixture class TestIosxrUserModule(TestIosxrModule): @@ -31,6 +30,8 @@ class TestIosxrUserModule(TestIosxrModule): module = iosxr_user def setUp(self): + super(TestIosxrUserModule, self).setUp() + self.mock_get_config = patch('ansible.modules.network.iosxr.iosxr_user.get_config') self.get_config = self.mock_get_config.start() @@ -38,6 +39,8 @@ class TestIosxrUserModule(TestIosxrModule): self.load_config = self.mock_load_config.start() def tearDown(self): + super(TestIosxrUserModule, self).tearDown() + self.mock_get_config.stop() self.mock_load_config.stop() |