class TestNiosSRVRecordModule(TestNiosModule): module = nios_srv_record def setUp(self): super(TestNiosSRVRecordModule, self).setUp() self.module = MagicMock( name='ansible.modules.net_tools.nios.nios_srv_record.WapiModule') self.module.check_mode = False self.module.params = {'provider': None} self.mock_wapi = patch( 'ansible.modules.net_tools.nios.nios_srv_record.WapiModule') self.exec_command = self.mock_wapi.start() self.mock_wapi_run = patch( 'ansible.modules.net_tools.nios.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.modules.net_tools.nios.nios_network_view.WapiModule') self.module.check_mode = False self.module.params = {'provider': None} self.mock_wapi = patch( 'ansible.modules.net_tools.nios.nios_network_view.WapiModule') self.exec_command = self.mock_wapi.start() self.mock_wapi_run = patch( 'ansible.modules.net_tools.nios.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 TestNiosAAAARecordModule(TestNiosModule): module = nios_aaaa_record def setUp(self): super(TestNiosAAAARecordModule, self).setUp() self.module = MagicMock( name='ansible.modules.net_tools.nios.nios_aaaa_record.WapiModule') self.module.check_mode = False self.module.params = {'provider': None} self.mock_wapi = patch( 'ansible.modules.net_tools.nios.nios_aaaa_record.WapiModule') self.exec_command = self.mock_wapi.start() self.mock_wapi_run = patch( 'ansible.modules.net_tools.nios.nios_aaaa_record.WapiModule.run') self.mock_wapi_run.start() self.load_config = self.mock_wapi_run.start() def tearDown(self): super(TestNiosAAAARecordModule, 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_aaaa_record_create(self): self.module.params = { 'provider': None, 'state': 'present', 'name': 'aaaa.ansible.com', 'ipv6': '2001:0db8:85a3:0000:0000:8a2e:0370:7334', 'comment': None, 'extattrs': None } test_object = None test_spec = { "name": { "ib_req": True }, "ipv6": { "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__(), 'ipv6': '2001:0db8:85a3:0000:0000:8a2e:0370:7334' }) def test_nios_aaaa_record_update_comment(self): self.module.params = { 'provider': None, 'state': 'present', 'name': 'aaaa.ansible.com', 'ipv6': '2001:0db8:85a3:0000:0000:8a2e:0370:7334', 'comment': 'updated comment', 'extattrs': None } test_object = [{ "comment": "test comment", "_ref": "aaaarecord/ZG5zLm5ldHdvcmtfdmlldyQw:default/true", "name": "aaaa.ansible.com", "ipv6": "2001:0db8:85a3:0000:0000:8a2e:0370:7334", "extattrs": {} }] test_spec = { "name": { "ib_req": True }, "ipv6": { "ib_req": True }, "comment": {}, "extattrs": {} } wapi = self._get_wapi(test_object) res = wapi.run('testobject', test_spec) self.assertTrue(res['changed']) def test_nios_aaaa_record_remove(self): self.module.params = { 'provider': None, 'state': 'absent', 'name': 'aaaa.ansible.com', 'ipv6': '2001:0db8:85a3:0000:0000:8a2e:0370:7334', 'comment': None, 'extattrs': None } ref = "aaaarecord/ZG5zLm5ldHdvcmtfdmlldyQw:default/false" test_object = [{ "comment": "test comment", "_ref": ref, "name": "aaaa.ansible.com", "ipv6": "2001:0db8:85a3:0000:0000:8a2e:0370:7334", "extattrs": { 'Site': { 'value': 'test' } } }] test_spec = { "name": { "ib_req": True }, "ipv6": { "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.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', 'ssl_verify', '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)