def test_get_network_rules(self, gatherfacts_module_mock):
network_rules = MockGatherfactsApi.get_network_rules_response('api')
self.get_module_args.update({
'gather_subset': ['network_rules']
})
gatherfacts_module_mock.module.params = self.get_module_args
gatherfacts_module_mock.network_api.get_network_rules = MagicMock(return_value=MockSDKResponse(network_rules))
gatherfacts_module_mock.perform_module_operation()
assert MockGatherfactsApi.get_network_rules_response('module') == gatherfacts_module_mock.module.exit_json.call_args[1]['NetworkRules']
def test_get_network_rules_api_exception(self, gatherfacts_module_mock):
self.get_module_args.update({
'gather_subset': ['network_rules']
})
gatherfacts_module_mock.module.params = self.get_module_args
gatherfacts_module_mock.network_api.get_network_rules = MagicMock(side_effect=MockApiException)
gatherfacts_module_mock.perform_module_operation()
assert MockGatherfactsApi.get_network_rules_response('error') == gatherfacts_module_mock.module.fail_json.call_args[1]['msg']