Unit tests: share common code (#31456)

* move set_module_args to units.modules.utils
* unit tests: reuse set_module_args
* unit tests: mock exit/fail_json in module.utils.ModuleTestCase
* unit tests: use module.utils.ModuleTestCase
* unit tests: fix 'import shadowed by loop variable'
This commit is contained in:
Pilou 2017-11-17 18:17:07 +01:00 committed by Matt Clay
commit a5c9726502
154 changed files with 671 additions and 1113 deletions

View file

@ -22,16 +22,9 @@ __metaclass__ = type
import os
import json
from ansible.compat.tests import unittest
from ansible.compat.tests.mock import patch
from ansible.module_utils import basic
from ansible.module_utils._text import to_bytes
from units.modules.utils import AnsibleExitJson, AnsibleFailJson, ModuleTestCase
def set_module_args(args):
args = json.dumps({'ANSIBLE_MODULE_ARGS': args})
basic._ANSIBLE_ARGS = to_bytes(args)
fixture_path = os.path.join(os.path.dirname(__file__), 'fixtures')
fixture_data = {}
@ -54,15 +47,7 @@ def load_fixture(name):
return data
class AnsibleExitJson(Exception):
pass
class AnsibleFailJson(Exception):
pass
class TestIosModule(unittest.TestCase):
class TestIosModule(ModuleTestCase):
def execute_module(self, failed=False, changed=False, commands=None, sort=True, defaults=False):
@ -84,27 +69,16 @@ class TestIosModule(unittest.TestCase):
return result
def failed(self):
def fail_json(*args, **kwargs):
kwargs['failed'] = True
raise AnsibleFailJson(kwargs)
with patch.object(basic.AnsibleModule, 'fail_json', fail_json):
with self.assertRaises(AnsibleFailJson) as exc:
self.module.main()
with self.assertRaises(AnsibleFailJson) as exc:
self.module.main()
result = exc.exception.args[0]
self.assertTrue(result['failed'], result)
return result
def changed(self, changed=False):
def exit_json(*args, **kwargs):
if 'changed' not in kwargs:
kwargs['changed'] = False
raise AnsibleExitJson(kwargs)
with patch.object(basic.AnsibleModule, 'exit_json', exit_json):
with self.assertRaises(AnsibleExitJson) as exc:
self.module.main()
with self.assertRaises(AnsibleExitJson) as exc:
self.module.main()
result = exc.exception.args[0]
self.assertEqual(result['changed'], changed, result)

View file

@ -17,11 +17,10 @@
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type
import json
from ansible.compat.tests.mock import patch
from ansible.modules.network.ios import ios_banner
from .ios_module import TestIosModule, load_fixture, set_module_args
from units.modules.utils import set_module_args
from .ios_module import TestIosModule, load_fixture
class TestIosBannerModule(TestIosModule):
@ -29,6 +28,8 @@ class TestIosBannerModule(TestIosModule):
module = ios_banner
def setUp(self):
super(TestIosBannerModule, self).setUp()
self.mock_exec_command = patch('ansible.modules.network.ios.ios_banner.exec_command')
self.exec_command = self.mock_exec_command.start()
@ -36,6 +37,7 @@ class TestIosBannerModule(TestIosModule):
self.load_config = self.mock_load_config.start()
def tearDown(self):
super(TestIosBannerModule, self).tearDown()
self.mock_exec_command.stop()
self.mock_load_config.stop()

View file

@ -23,7 +23,8 @@ import json
from ansible.compat.tests.mock import patch
from ansible.modules.network.ios import ios_command
from .ios_module import TestIosModule, load_fixture, set_module_args
from units.modules.utils import set_module_args
from .ios_module import TestIosModule, load_fixture
class TestIosCommandModule(TestIosModule):
@ -31,10 +32,13 @@ class TestIosCommandModule(TestIosModule):
module = ios_command
def setUp(self):
super(TestIosCommandModule, self).setUp()
self.mock_run_commands = patch('ansible.modules.network.ios.ios_command.run_commands')
self.run_commands = self.mock_run_commands.start()
def tearDown(self):
super(TestIosCommandModule, self).tearDown()
self.mock_run_commands.stop()
def load_fixtures(self, commands=None):

View file

@ -20,11 +20,10 @@
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type
import json
from ansible.compat.tests.mock import patch
from ansible.modules.network.ios import ios_config
from .ios_module import TestIosModule, load_fixture, set_module_args
from units.modules.utils import set_module_args
from .ios_module import TestIosModule, load_fixture
class TestIosConfigModule(TestIosModule):
@ -32,6 +31,8 @@ class TestIosConfigModule(TestIosModule):
module = ios_config
def setUp(self):
super(TestIosConfigModule, self).setUp()
self.mock_get_config = patch('ansible.modules.network.ios.ios_config.get_config')
self.get_config = self.mock_get_config.start()
@ -42,6 +43,7 @@ class TestIosConfigModule(TestIosModule):
self.run_commands = self.mock_run_commands.start()
def tearDown(self):
super(TestIosConfigModule, self).tearDown()
self.mock_get_config.stop()
self.mock_load_config.stop()
self.mock_run_commands.stop()

View file

@ -25,7 +25,8 @@ import json
from ansible.compat.tests.mock import patch
from ansible.modules.network.ios import ios_logging
from .ios_module import TestIosModule, load_fixture, set_module_args
from units.modules.utils import set_module_args
from .ios_module import TestIosModule, load_fixture
class TestIosLoggingModule(TestIosModule):
@ -33,6 +34,8 @@ class TestIosLoggingModule(TestIosModule):
module = ios_logging
def setUp(self):
super(TestIosLoggingModule, self).setUp()
self.mock_get_config = patch('ansible.modules.network.ios.ios_logging.get_config')
self.get_config = self.mock_get_config.start()
@ -40,6 +43,8 @@ class TestIosLoggingModule(TestIosModule):
self.load_config = self.mock_load_config.start()
def tearDown(self):
super(TestIosLoggingModule, self).tearDown()
self.mock_get_config.stop()
self.mock_load_config.stop()

View file

@ -22,7 +22,8 @@ __metaclass__ = type
from ansible.compat.tests.mock import patch
from ansible.modules.network.ios import ios_ping
from .ios_module import TestIosModule, load_fixture, set_module_args
from units.modules.utils import set_module_args
from .ios_module import TestIosModule, load_fixture
class TestIosPingModule(TestIosModule):
@ -30,10 +31,12 @@ class TestIosPingModule(TestIosModule):
module = ios_ping
def setUp(self):
super(TestIosPingModule, self).setUp()
self.mock_run_commands = patch('ansible.modules.network.ios.ios_ping.run_commands')
self.run_commands = self.mock_run_commands.start()
def tearDown(self):
super(TestIosPingModule, self).tearDown()
self.mock_run_commands.stop()
def load_fixtures(self, commands=None):

View file

@ -20,11 +20,10 @@
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type
import json
from ansible.compat.tests.mock import patch
from ansible.modules.network.ios import ios_system
from .ios_module import TestIosModule, load_fixture, set_module_args
from units.modules.utils import set_module_args
from .ios_module import TestIosModule, load_fixture
class TestIosSystemModule(TestIosModule):
@ -32,6 +31,8 @@ class TestIosSystemModule(TestIosModule):
module = ios_system
def setUp(self):
super(TestIosSystemModule, self).setUp()
self.mock_get_config = patch('ansible.modules.network.ios.ios_system.get_config')
self.get_config = self.mock_get_config.start()
@ -39,6 +40,7 @@ class TestIosSystemModule(TestIosModule):
self.load_config = self.mock_load_config.start()
def tearDown(self):
super(TestIosSystemModule, self).tearDown()
self.mock_get_config.stop()
self.mock_load_config.stop()

View file

@ -23,7 +23,8 @@ import json
from ansible.compat.tests.mock import patch
from ansible.modules.network.ios import ios_user
from .ios_module import TestIosModule, load_fixture, set_module_args
from units.modules.utils import set_module_args
from .ios_module import TestIosModule, load_fixture
class TestIosUserModule(TestIosModule):
@ -31,6 +32,8 @@ class TestIosUserModule(TestIosModule):
module = ios_user
def setUp(self):
super(TestIosUserModule, self).setUp()
self.mock_get_config = patch('ansible.modules.network.ios.ios_user.get_config')
self.get_config = self.mock_get_config.start()
@ -38,6 +41,7 @@ class TestIosUserModule(TestIosModule):
self.load_config = self.mock_load_config.start()
def tearDown(self):
super(TestIosUserModule, self).tearDown()
self.mock_get_config.stop()
self.mock_load_config.stop()

View file

@ -20,11 +20,10 @@
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type
import json
from ansible.compat.tests.mock import patch
from ansible.modules.network.ios import ios_vrf
from .ios_module import TestIosModule, load_fixture, set_module_args
from units.modules.utils import set_module_args
from .ios_module import TestIosModule, load_fixture
class TestIosVrfModule(TestIosModule):
@ -32,6 +31,8 @@ class TestIosVrfModule(TestIosModule):
module = ios_vrf
def setUp(self):
super(TestIosVrfModule, self).setUp()
self.mock_get_config = patch('ansible.modules.network.ios.ios_vrf.get_config')
self.get_config = self.mock_get_config.start()
@ -42,6 +43,7 @@ class TestIosVrfModule(TestIosModule):
self.exec_command = self.mock_exec_command.start()
def tearDown(self):
super(TestIosVrfModule, self).tearDown()
self.mock_get_config.stop()
self.mock_load_config.stop()
self.mock_exec_command.stop()