def setUp(self): super(TestOnyxLinkaggModule, self).setUp() self.mock_get_config = patch.object( onyx_linkagg.OnyxLinkAggModule, "_get_port_channels") self.get_config = self.mock_get_config.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config') self.load_config = self.mock_load_config.start() self.mock_get_version = patch.object( onyx_linkagg.OnyxLinkAggModule, "_get_os_version") self.get_version = self.mock_get_version.start()
def setUp(self): super(TestOnyxVxlanModule, self).setUp() self.mock_get_vxlan_config = patch.object( onyx_vxlan.OnyxVxlanModule, "_show_vxlan_config") self.get_vxlan_config = self.mock_get_vxlan_config.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config') self.load_config = self.mock_load_config.start() self.mock_get_nve_detail = patch.object( onyx_vxlan.OnyxVxlanModule, "_show_nve_detail") self.get_nve_detail = self.mock_get_nve_detail.start()
def setUp(self): self._ptp_enabled = True self._ntp_enabled = True super(TestOnyxPtpModule, self).setUp() self.mock_get_ptp_config = patch.object(onyx_ptp_global.OnyxPtpGlobalModule, "_show_ptp_config") self.get_ptp_config = self.mock_get_ptp_config.start() self.mock_get_ntp_config = patch.object(onyx_ptp_global.OnyxPtpGlobalModule, "_show_ntp_config") self.get_ntp_config = self.mock_get_ntp_config.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config') self.load_config = self.mock_load_config.start()
def setUp(self): super(TestOnyxVlanModule, self).setUp() self.mock_get_config = patch.object(onyx_vlan.OnyxVlanModule, "_get_vlan_config") self.get_config = self.mock_get_config.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config' ) self.load_config = self.mock_load_config.start() self.mock_get_version = patch.object(onyx_vlan.OnyxVlanModule, "_get_os_version") self.get_version = self.mock_get_version.start()
def setUp(self): super(TestOnyxInterfaceModule, self).setUp() self.mock_get_config = patch.object( onyx_l2_interface.OnyxL2InterfaceModule, "_get_switchport_config") self.get_config = self.mock_get_config.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config' ) self.load_config = self.mock_load_config.start() self.mock_get_version = patch.object( onyx_l2_interface.OnyxL2InterfaceModule, "_get_os_version") self.get_version = self.mock_get_version.start()
def setUp(self): super(TestOnyxMlagVipModule, self).setUp() self._mlag_enabled = True self.mock_show_mlag = patch.object( onyx_mlag_vip.OnyxMLagVipModule, "_show_mlag") self.show_mlag = self.mock_show_mlag.start() self.mock_show_mlag_vip = patch.object( onyx_mlag_vip.OnyxMLagVipModule, "_show_mlag_vip") self.show_mlag_vip = self.mock_show_mlag_vip.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config') self.load_config = self.mock_load_config.start()
def setUp(self): super(TestOnyxOspfModule, self).setUp() self._ospf_exists = True self.mock_get_config = patch.object(onyx_ospf.OnyxOspfModule, "_get_ospf_config") self.get_config = self.mock_get_config.start() self.mock_get_interfaces_config = patch.object( onyx_ospf.OnyxOspfModule, "_get_ospf_interfaces_config") self.get_interfaces_config = self.mock_get_interfaces_config.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config' ) self.load_config = self.mock_load_config.start()
def test_ensure_command_called(self): """ Testing that command is executed with correct args :return: """ set_module_args({ 'operation': "Upload", 'opsName': 'Uploading_testi', 'DN': "PLMN-PLMN/MRBTS-746", }) with patch.object(basic.AnsibleModule, 'run_command') as mock_run_command: stdout = 'configuration updated' stderr = '' return_code = 0 mock_run_command.return_value = return_code, stdout, stderr # successful execution with self.assertRaises(AnsibleExitJson) as result: netact_cm_command.main() print(result.exception.args) self.assertTrue(result.exception.args[0] ['changed']) # ensure result is changed mock_run_command.assert_called_once_with([ '/opt/oss/bin/racclimx.sh', '-op', 'Upload', '-opsName', 'Uploading_testi', '-DN', 'PLMN-PLMN/MRBTS-746' ], check_rc=True)
def test_withwrongargs(self): """ Testing that wrong attribute causing error :return: """ set_module_args({ 'operation': "Upload", 'opsName': 'Uploading_testi', 'MR': "PLMN-PLMN/MRBTS-746", 'abc': 'abc' }) with self.assertRaises(AnsibleFailJson): with patch.object(basic.AnsibleModule, 'run_command') as mock_run_command: stdout = 'configuration updated' stderr = '' return_code = 0 mock_run_command.return_value = return_code, stdout, stderr # successful execution with self.assertRaises(AnsibleExitJson) as result: netact_cm_command.main() self.assertTrue(result.exception.args[0] ['changed']) # ensure result is changed self.assertFalse(True) # ensure result is changed
def test_ensure_backupPlanName_outputs_correctly(self): """ Testing that command is executed with correct args :return: """ set_module_args({ 'operation': "Provision", 'opsName': 'Provision_test', 'WS': "PLMN-PLMN/MRBTS-746", 'createBackupPlan': "Yes", 'backupPlanName': "backupPlanName" }) with patch.object(basic.AnsibleModule, 'run_command') as mock_run_command: stdout = 'configuration updated' stderr = '' return_code = 0 mock_run_command.return_value = return_code, stdout, stderr # successful execution with self.assertRaises(AnsibleExitJson) as result: netact_cm_command.main() print(result.exception.args) self.assertTrue(result.exception.args[0] ['changed']) # ensure result is changed mock_run_command.assert_called_once_with([ '/opt/oss/bin/racclimx.sh', '-op', 'Provision', '-opsName', 'Provision_test', '-WS', 'PLMN-PLMN/MRBTS-746', '-createBackupPlan', 'true', '-backupPlanName', 'backupPlanName' ], check_rc=True)
def setUp(self): super(TestOnyxQosModule, self).setUp() self.mock_get_if_qos_config = patch.object( onyx_qos.OnyxQosModule, "_show_interface_qos") self.get_if_qos_config = self.mock_get_if_qos_config.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config') self.load_config = self.mock_load_config.start()
def setUp(self): self.enabled = False super(TestOnyxNTP, self).setUp() self.mock_get_config = patch.object( onyx_ntp.OnyxNTPModule, "_show_ntp_config") self.get_config = self.mock_get_config.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config') self.load_config = self.mock_load_config.start()
def setUp(self): super(TestOnyxInterfaceModule, self).setUp() self.mock_get_config = patch.object(onyx_lldp.OnyxLldpModule, "_get_lldp_config") self.get_config = self.mock_get_config.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config' ) self.load_config = self.mock_load_config.start()
def setUp(self): super(TestOnyxBufferPoolModule, self).setUp() self.mock_get_buffer_pool_config = patch.object( onyx_buffer_pool.OnyxBufferPoolModule, "_show_traffic_pool") self.get_buffer_pool_config = self.mock_get_buffer_pool_config.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config' ) self.load_config = self.mock_load_config.start()
def test_lookup_multiple_obj(dummy_credentials): avi_lookup = lookup_loader.get('community.network.avi') avi_mock = MagicMock() avi_mock.return_value.get.return_value.json.return_value = data[ "mock_multiple_obj"] with patch.object(avi, 'ApiSession', avi_mock): retval = avi_lookup.run([], {}, avi_credentials=dummy_credentials, obj_type="network") assert retval == data["mock_multiple_obj"]["results"]
def setUp(self): self.enabled = False super(TestOnyxSyslogFilesModule, self).setUp() self.mock_get_config = patch.object( onyx_syslog_files.OnyxSyslogFilesModule, "show_logging") self.get_config = self.mock_get_config.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config') self.load_config = self.mock_load_config.start()
def test_lookup_single_obj(dummy_credentials): avi_lookup = lookup_loader.get('community.network.avi') avi_mock = MagicMock() avi_mock.return_value.get_object_by_name.return_value = data[ "mock_single_obj"] with patch.object(avi, 'ApiSession', avi_mock): retval = avi_lookup.run([], {}, avi_credentials=dummy_credentials, obj_type="network", obj_name='PG-123') assert retval[0] == data["mock_single_obj"]
def setUp(self): super(TestOnyxMlagIplModule, self).setUp() self._mlag_enabled = True self.mock_get_config = patch.object( onyx_mlag_ipl.OnyxMlagIplModule, "_show_mlag_data") self.get_config = self.mock_get_config.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config') self.load_config = self.mock_load_config.start()
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() 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() result = exc.exception.args[0] self.assertEqual(result['changed'], changed, result) return result
def setUp(self): self.enabled = False super(TestOnyxIgmpVlan, self).setUp() self.mock_get_igmp_config = patch.object( onyx_igmp_vlan.OnyxIgmpVlanModule, "_show_igmp_vlan") self.get_igmp_config = self.mock_get_igmp_config.start() self.mock_load_config = patch( 'ansible_collections.community.network.plugins.module_utils.network.onyx.onyx.load_config' ) self.load_config = self.mock_load_config.start() self.mock_get_igmp_guerier_config = patch.object( onyx_igmp_vlan.OnyxIgmpVlanModule, "_show_igmp_querier_config") self.get_igmp_guerier_config = self.mock_get_igmp_guerier_config.start( ) self.mock_get_igmp_static_groups_config = patch.object( onyx_igmp_vlan.OnyxIgmpVlanModule, "_show_igmp_snooping_groups_config") self.get_igmp_static_groups_config = self.mock_get_igmp_static_groups_config.start( )
def test_stringify_name_filter(self, test_api_version, expected_result, connection_mock): filters = {"name": "object_name"} with patch.object(BaseConfigurationResource, '_fetch_system_info') as fetch_system_info_mock: fetch_system_info_mock.return_value = { 'databaseInfo': { 'buildVersion': test_api_version } } resource = BaseConfigurationResource(connection_mock, False) assert resource._stringify_name_filter( filters ) == expected_result, "Unexpected result for version %s" % ( test_api_version)
def setUp(self): super(TestOnyxFacts, self).setUp() self.mock_run_command = patch.object(onyx_facts.FactsBase, "_show_cmd") self.run_command = self.mock_run_command.start()
def test_invalid_lookup(dummy_credentials): avi_lookup = lookup_loader.get('community.network.avi') avi_mock = MagicMock() with pytest.raises(AnsibleError): with patch.object(avi, 'ApiSession', avi_mock): avi_lookup.run([], {}, avi_credentials=dummy_credentials)