class TestNiosNAPTRRecordModule(TestNiosModule): module = nios_naptr_record def setUp(self): super(TestNiosNAPTRRecordModule, self).setUp() self.module = MagicMock(name='ansible_collections.misc.not_a_real_collection.plugins.modules.nios_naptr_record.WapiModule') self.module.check_mode = False self.module.params = {'provider': None} self.mock_wapi = patch('ansible_collections.misc.not_a_real_collection.plugins.modules.nios_naptr_record.WapiModule') self.exec_command = self.mock_wapi.start() self.mock_wapi_run = patch('ansible_collections.misc.not_a_real_collection.plugins.modules.nios_naptr_record.WapiModule.run') self.mock_wapi_run.start() self.load_config = self.mock_wapi_run.start() def tearDown(self): super(TestNiosNAPTRRecordModule, self).tearDown() self.mock_wapi.stop() self.mock_wapi_run.stop() def _get_wapi(self, test_object): wapi = api.WapiModule(self.module) wapi.get_object = Mock(name='get_object', return_value=test_object) wapi.create_object = Mock(name='create_object') wapi.update_object = Mock(name='update_object') wapi.delete_object = Mock(name='delete_object') return wapi def load_fixtures(self, commands=None): self.exec_command.return_value = (0, load_fixture('nios_result.txt').strip(), None) self.load_config.return_value = dict(diff=None, session='session') def test_nios_naptr_record_create(self): self.module.params = {'provider': None, 'state': 'present', 'name': '*.subscriber-100.ansiblezone.com', 'order': '1000', 'preference': '10', 'replacement': 'replacement1.network.ansiblezone.com', 'comment': None, 'extattrs': None} test_object = None test_spec = { "name": {"ib_req": True}, "order": {"ib_req": True}, "preference": {"ib_req": True}, "replacement": {"ib_req": True}, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) print("WAPI: ", wapi.__dict__) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.create_object.assert_called_once_with('testobject', {'name': self.module._check_type_dict().__getitem__(), 'order': '1000', 'preference': '10', 'replacement': 'replacement1.network.ansiblezone.com'}) def test_nios_naptr_record_update_comment(self): self.module.params = {'provider': None, 'state': 'present', 'name': '*.subscriber-100.ansiblezone.com', 'order': '1000', 'preference': '10', 'replacement': 'replacement1.network.ansiblezone.com', 'comment': 'updated comment', 'extattrs': None} test_object = [ { "comment": "test comment", "_ref": "naptrrecord/ZG5zLm5ldHdvcmtfdmlldyQw:default/true", "name": "*.subscriber-100.ansiblezone.com", "order": "1000", "preference": "10", "replacement": "replacement1.network.ansiblezone.com", "extattrs": {} } ] test_spec = { "name": {"ib_req": True}, "order": {"ib_req": True}, "preference": {"ib_req": True}, "replacement": {"ib_req": True}, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) def test_nios_naptr_record_remove(self): self.module.params = {'provider': None, 'state': 'absent', 'name': '*.subscriber-100.ansiblezone.com', 'order': '1000', 'preference': '10', 'replacement': 'replacement1.network.ansiblezone.com', 'comment': None, 'extattrs': None} ref = "naptrrecord/ZG5zLm5ldHdvcmtfdmlldyQw:default/false" test_object = [{ "comment": "test comment", "_ref": ref, "name": "*.subscriber-100.ansiblezone.com", "order": "1000", "preference": "10", "replacement": "replacement1.network.ansiblezone.com", "extattrs": {'Site': {'value': 'test'}} }] test_spec = { "name": {"ib_req": True}, "order": {"ib_req": True}, "preference": {"ib_req": True}, "replacement": {"ib_req": True}, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.delete_object.assert_called_once_with(ref)
class TestNiosSRVRecordModule(TestNiosModule): module = nios_srv_record def setUp(self): super(TestNiosSRVRecordModule, self).setUp() self.module = MagicMock( name= 'ansible_collections.misc.not_a_real_collection.plugins.modules.nios_srv_record.WapiModule' ) self.module.check_mode = False self.module.params = {'provider': None} self.mock_wapi = patch( 'ansible_collections.misc.not_a_real_collection.plugins.modules.nios_srv_record.WapiModule' ) self.exec_command = self.mock_wapi.start() self.mock_wapi_run = patch( 'ansible_collections.misc.not_a_real_collection.plugins.modules.nios_srv_record.WapiModule.run' ) self.mock_wapi_run.start() self.load_config = self.mock_wapi_run.start() def tearDown(self): super(TestNiosSRVRecordModule, self).tearDown() self.mock_wapi.stop() self.mock_wapi_run.stop() def _get_wapi(self, test_object): wapi = api.WapiModule(self.module) wapi.get_object = Mock(name='get_object', return_value=test_object) wapi.create_object = Mock(name='create_object') wapi.update_object = Mock(name='update_object') wapi.delete_object = Mock(name='delete_object') return wapi def load_fixtures(self, commands=None): self.exec_command.return_value = ( 0, load_fixture('nios_result.txt').strip(), None) self.load_config.return_value = dict(diff=None, session='session') def test_nios_srv_record_create(self): self.module.params = { 'provider': None, 'state': 'present', 'name': '_sip._tcp.service.ansible.com', 'port': 5080, 'target': 'service1.ansible.com', 'priority': 10, 'weight': 10, 'comment': None, 'extattrs': None } test_object = None test_spec = { "name": { "ib_req": True }, "port": { "ib_req": True }, "target": { "ib_req": True }, "priority": { "ib_req": True }, "weight": { "ib_req": True }, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) print("WAPI: ", wapi) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.create_object.assert_called_once_with( 'testobject', { 'name': self.module._check_type_dict().__getitem__(), 'port': 5080, 'target': 'service1.ansible.com', 'priority': 10, 'weight': 10 }) def test_nios_srv_record_update_comment(self): self.module.params = { 'provider': None, 'state': 'present', 'name': '_sip._tcp.service.ansible.com', 'port': 5080, 'target': 'service1.ansible.com', 'priority': 10, 'weight': 10, 'comment': None, 'extattrs': None } test_object = [{ "comment": "test comment", "_ref": "srvrecord/ZG5zLm5ldHdvcmtfdmlldyQw:default/true", "name": "_sip._tcp.service.ansible.com", 'port': 5080, "target": "mailhost.ansible.com", "priority": 10, 'weight': 10, "extattrs": {} }] test_spec = { "name": { "ib_req": True }, "port": { "ib_req": True }, "target": { "ib_req": True }, "priority": { "ib_req": True }, "weight": { "ib_req": True }, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) def test_nios_srv_record_remove(self): self.module.params = { 'provider': None, 'state': 'absent', 'name': '_sip._tcp.service.ansible.com', 'port': 5080, 'target': 'service1.ansible.com', 'priority': 10, 'weight': 10, 'comment': None, 'extattrs': None } ref = "srvrecord/ZG5zLm5ldHdvcmtfdmlldyQw:default/false" test_object = [{ "comment": "test comment", "_ref": ref, "name": "_sip._tcp.service.ansible.com", "port": 5080, "target": "mailhost.ansible.com", "priority": 10, "weight": 10, "extattrs": { 'Site': { 'value': 'test' } } }] test_spec = { "name": { "ib_req": True }, "port": { "ib_req": True }, "target": { "ib_req": True }, "priority": { "ib_req": True }, "weight": { "ib_req": True }, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.delete_object.assert_called_once_with(ref)
class TestNiosNetworkViewModule(TestNiosModule): module = nios_network_view def setUp(self): super(TestNiosNetworkViewModule, self).setUp() self.module = MagicMock(name='ansible_collections.misc.not_a_real_collection.plugins.modules.nios_network_view.WapiModule') self.module.check_mode = False self.module.params = {'provider': None} self.mock_wapi = patch('ansible_collections.misc.not_a_real_collection.plugins.modules.nios_network_view.WapiModule') self.exec_command = self.mock_wapi.start() self.mock_wapi_run = patch('ansible_collections.misc.not_a_real_collection.plugins.modules.nios_network_view.WapiModule.run') self.mock_wapi_run.start() self.load_config = self.mock_wapi_run.start() def tearDown(self): super(TestNiosNetworkViewModule, self).tearDown() self.mock_wapi.stop() self.mock_wapi_run.stop() def _get_wapi(self, test_object): wapi = api.WapiModule(self.module) wapi.get_object = Mock(name='get_object', return_value=test_object) wapi.create_object = Mock(name='create_object') wapi.update_object = Mock(name='update_object') wapi.delete_object = Mock(name='delete_object') return wapi def load_fixtures(self, commands=None): self.exec_command.return_value = (0, load_fixture('nios_result.txt').strip(), None) self.load_config.return_value = dict(diff=None, session='session') def test_nios_network_view_create(self): self.module.params = {'provider': None, 'state': 'present', 'name': 'ansible', 'comment': None, 'extattrs': None} test_object = None test_spec = { "name": {"ib_req": True}, "comment": {}, "extattrs": {}, } wapi = self._get_wapi(test_object) print("WAPI: ", wapi) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.create_object.assert_called_once_with('testobject', {'name': self.module._check_type_dict().__getitem__()}) def test_nios_network_view_update_comment(self): self.module.params = {'provider': None, 'state': 'present', 'name': 'default', 'comment': 'updated comment', 'extattrs': None, 'network_view': 'default'} test_object = [ { "comment": "test comment", "_ref": "networkview/ZG5zLm5ldHdvcmtfdmlldyQw:default/true", "name": "default", "extattrs": {}, "network_view": "default" } ] test_spec = { "name": {"ib_req": True}, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.update_object.called_once_with(test_object) def test_nios_network_view_update_name(self): self.module.params = {'provider': None, 'state': 'present', 'name': 'default', 'old_name': 'old_default', 'comment': 'updated comment', 'extattrs': None, 'network_view': 'default'} test_object = [ { "comment": "test comment", "_ref": "networkview/ZG5zLm5ldHdvcmtfdmlldyQw:default/true", "name": "default", "old_name": "old_default", "extattrs": {}, "network_view": "default" } ] test_spec = { "name": {"ib_req": True}, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.update_object.called_once_with(test_object) def test_nios_network_view_remove(self): self.module.params = {'provider': None, 'state': 'absent', 'name': 'ansible', 'comment': None, 'extattrs': None} ref = "networkview/ZG5zLm5ldHdvcmtfdmlldyQw:ansible/false" test_object = [{ "comment": "test comment", "_ref": ref, "name": "ansible", "extattrs": {'Site': {'value': 'test'}} }] test_spec = { "name": {"ib_req": True}, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.delete_object.assert_called_once_with(ref)
class TestNiosApi(unittest.TestCase): def setUp(self): super(TestNiosApi, self).setUp() self.module = MagicMock(name='AnsibleModule') self.module.check_mode = False self.module.params = {'provider': None} self.mock_connector = patch('ansible_collections.misc.not_a_real_collection.plugins.module_utils.net_tools.nios.api.get_connector') self.mock_connector.start() def tearDown(self): super(TestNiosApi, self).tearDown() self.mock_connector.stop() def test_get_provider_spec(self): provider_options = ['host', 'username', 'password', 'validate_certs', 'silent_ssl_warnings', 'http_request_timeout', 'http_pool_connections', 'http_pool_maxsize', 'max_retries', 'wapi_version', 'max_results'] res = api.WapiBase.provider_spec self.assertIsNotNone(res) self.assertIn('provider', res) self.assertIn('options', res['provider']) returned_options = res['provider']['options'] self.assertEqual(sorted(provider_options), sorted(returned_options.keys())) def _get_wapi(self, test_object): wapi = api.WapiModule(self.module) wapi.get_object = Mock(name='get_object', return_value=test_object) wapi.create_object = Mock(name='create_object') wapi.update_object = Mock(name='update_object') wapi.delete_object = Mock(name='delete_object') return wapi def test_wapi_no_change(self): self.module.params = {'provider': None, 'state': 'present', 'name': 'default', 'comment': 'test comment', 'extattrs': None} test_object = [ { "comment": "test comment", "_ref": "networkview/ZG5zLm5ldHdvcmtfdmlldyQw:default/true", "name": self.module._check_type_dict().__getitem__(), "extattrs": {} } ] test_spec = { "name": {"ib_req": True}, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertFalse(res['changed']) def test_wapi_change(self): self.module.params = {'provider': None, 'state': 'present', 'name': 'default', 'comment': 'updated comment', 'extattrs': None} test_object = [ { "comment": "test comment", "_ref": "networkview/ZG5zLm5ldHdvcmtfdmlldyQw:default/true", "name": "default", "extattrs": {} } ] test_spec = { "name": {"ib_req": True}, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.update_object.called_once_with(test_object) def test_wapi_change_false(self): self.module.params = {'provider': None, 'state': 'present', 'name': 'default', 'comment': 'updated comment', 'extattrs': None, 'fqdn': 'foo'} test_object = [ { "comment": "test comment", "_ref": "networkview/ZG5zLm5ldHdvcmtfdmlldyQw:default/true", "name": "default", "extattrs": {} } ] test_spec = { "name": {"ib_req": True}, "fqdn": {"ib_req": True, 'update': False}, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.update_object.called_once_with(test_object) def test_wapi_extattrs_change(self): self.module.params = {'provider': None, 'state': 'present', 'name': 'default', 'comment': 'test comment', 'extattrs': {'Site': 'update'}} ref = "networkview/ZG5zLm5ldHdvcmtfdmlldyQw:default/true" test_object = [{ "comment": "test comment", "_ref": ref, "name": "default", "extattrs": {'Site': {'value': 'test'}} }] test_spec = { "name": {"ib_req": True}, "comment": {}, "extattrs": {} } kwargs = copy.deepcopy(test_object[0]) kwargs['extattrs']['Site']['value'] = 'update' kwargs['name'] = self.module._check_type_dict().__getitem__() del kwargs['_ref'] wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.update_object.assert_called_once_with(ref, kwargs) def test_wapi_extattrs_nochange(self): self.module.params = {'provider': None, 'state': 'present', 'name': 'default', 'comment': 'test comment', 'extattrs': {'Site': 'test'}} test_object = [{ "comment": "test comment", "_ref": "networkview/ZG5zLm5ldHdvcmtfdmlldyQw:default/true", "name": self.module._check_type_dict().__getitem__(), "extattrs": {'Site': {'value': 'test'}} }] test_spec = { "name": {"ib_req": True}, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertFalse(res['changed']) def test_wapi_create(self): self.module.params = {'provider': None, 'state': 'present', 'name': 'ansible', 'comment': None, 'extattrs': None} test_object = None test_spec = { "name": {"ib_req": True}, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.create_object.assert_called_once_with('testobject', {'name': self.module._check_type_dict().__getitem__()}) def test_wapi_delete(self): self.module.params = {'provider': None, 'state': 'absent', 'name': 'ansible', 'comment': None, 'extattrs': None} ref = "networkview/ZG5zLm5ldHdvcmtfdmlldyQw:ansible/false" test_object = [{ "comment": "test comment", "_ref": ref, "name": "ansible", "extattrs": {'Site': {'value': 'test'}} }] test_spec = { "name": {"ib_req": True}, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.delete_object.assert_called_once_with(ref) def test_wapi_strip_network_view(self): self.module.params = {'provider': None, 'state': 'present', 'name': 'ansible', 'comment': 'updated comment', 'extattrs': None, 'network_view': 'default'} test_object = [{ "comment": "test comment", "_ref": "view/ZG5zLm5ldHdvcmtfdmlldyQw:ansible/true", "name": "ansible", "extattrs": {}, "network_view": "default" }] test_spec = { "name": {"ib_req": True}, "network_view": {"ib_req": True}, "comment": {}, "extattrs": {} } kwargs = test_object[0].copy() ref = kwargs.pop('_ref') kwargs['comment'] = 'updated comment' kwargs['name'] = self.module._check_type_dict().__getitem__() del kwargs['network_view'] del kwargs['extattrs'] wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.update_object.assert_called_once_with(ref, kwargs)
class TestNiosNSGroupModule(TestNiosModule): module = nios_nsgroup def setUp(self): super(TestNiosNSGroupModule, self).setUp() self.module = MagicMock( name= 'ansible_collections.misc.not_a_real_collection.plugins.modules.nios_nsgroup.WapiModule' ) self.module.check_mode = False self.module.params = {'provider': None} self.mock_wapi = patch( 'ansible_collections.misc.not_a_real_collection.plugins.modules.nios_nsgroup.WapiModule' ) self.exec_command = self.mock_wapi.start() self.mock_wapi_run = patch( 'ansible_collections.misc.not_a_real_collection.plugins.modules.nios_nsgroup.WapiModule.run' ) self.mock_wapi_run.start() self.load_config = self.mock_wapi_run.start() def tearDown(self): super(TestNiosNSGroupModule, self).tearDown() self.mock_wapi.stop() def _get_wapi(self, test_object): wapi = api.WapiModule(self.module) wapi.get_object = Mock(name='get_object', return_value=test_object) wapi.create_object = Mock(name='create_object') wapi.update_object = Mock(name='update_object') wapi.delete_object = Mock(name='delete_object') return wapi def load_fixtures(self, commands=None): self.exec_command.return_value = ( 0, load_fixture('nios_result.txt').strip(), None) self.load_config.return_value = dict(diff=None, session='session') def test_nios_nsgroup_create(self): self.module.params = { 'provider': None, 'state': 'present', 'name': 'my-simple-group', 'comment': None, 'grid_primary': None } test_object = None test_spec = { "name": { "ib_req": True }, "comment": {}, "grid_primary": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.create_object.assert_called_once_with( 'testobject', {'name': self.module._check_type_dict().__getitem__()}) def test_nios_nsgroup_remove(self): self.module.params = { 'provider': None, 'state': 'absent', 'name': 'my-simple-group', 'comment': None, 'grid_primary': None } ref = "nsgroup/ZG5zLm5ldHdvcmtfdmlldyQw:ansible/false" test_object = [{ "comment": "test comment", "_ref": ref, "name": "my-simple-group", "grid_primary": { 'name': 'infoblox-test.example.com' } }] test_spec = { "name": { "ib_req": True }, "comment": {}, "grid_primary": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.delete_object.assert_called_once_with(ref) def test_nios_nsgroup_update_comment(self): self.module.params = { 'provider': None, 'state': 'present', 'name': 'default', 'comment': 'updated comment', 'grid_primary': None } test_object = [{ "comment": "test comment", "_ref": "nsgroup/ZG5zLm5ldHdvcmtfdmlldyQw:default/true", "name": "default", "grid_primary": {} }] test_spec = { "name": { "ib_req": True }, "comment": {}, "grid_primary": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) wapi.update_object.called_once_with(test_object)