def test_waf_main_class_filter_foreign_attributes(mocker):
schema_method_mock = mocker.patch(
'ansible_collections.notmintest.not_a_real_collection.plugins.module_utils.network.fortios.fortios.FortiOSHandler.schema'
)
set_method_result = {
'status': 'success',
'http_method': 'POST',
'http_status': 200
}
set_method_mock = mocker.patch(
'ansible_collections.notmintest.not_a_real_collection.plugins.module_utils.network.fortios.fortios.FortiOSHandler.set',
return_value=set_method_result)
input_data = {
'username': '******',
'state': 'present',
'waf_main_class': {
'random_attribute_not_valid': 'tag',
'id': '3',
'name': 'default_name_4'
},
'vdom': 'root'
}
is_error, changed, response = fortios_waf_main_class.fortios_waf(
input_data, fos_instance)
expected_data = {'id': '3', 'name': 'default_name_4'}
set_method_mock.assert_called_with('waf',
'main-class',
data=expected_data,
vdom='root')
schema_method_mock.assert_not_called()
assert not is_error
assert changed
assert response['status'] == 'success'
assert response['http_status'] == 200
def test_waf_main_class_idempotent(mocker):
schema_method_mock = mocker.patch(
'ansible_collections.notmintest.not_a_real_collection.plugins.module_utils.network.fortios.fortios.FortiOSHandler.schema'
)
set_method_result = {
'status': 'error',
'http_method': 'DELETE',
'http_status': 404
}
set_method_mock = mocker.patch(
'ansible_collections.notmintest.not_a_real_collection.plugins.module_utils.network.fortios.fortios.FortiOSHandler.set',
return_value=set_method_result)
input_data = {
'username': '******',
'state': 'present',
'waf_main_class': {
'id': '3',
'name': 'default_name_4'
},
'vdom': 'root'
}
is_error, changed, response = fortios_waf_main_class.fortios_waf(
input_data, fos_instance)
expected_data = {'id': '3', 'name': 'default_name_4'}
set_method_mock.assert_called_with('waf',
'main-class',
data=expected_data,
vdom='root')
schema_method_mock.assert_not_called()
assert not is_error
assert not changed
assert response['status'] == 'error'
assert response['http_status'] == 404
def test_waf_main_class_deletion_fails(mocker):
schema_method_mock = mocker.patch(
'ansible_collections.notmintest.not_a_real_collection.plugins.module_utils.network.fortios.fortios.FortiOSHandler.schema'
)
delete_method_result = {
'status': 'error',
'http_method': 'POST',
'http_status': 500
}
delete_method_mock = mocker.patch(
'ansible_collections.notmintest.not_a_real_collection.plugins.module_utils.network.fortios.fortios.FortiOSHandler.delete',
return_value=delete_method_result)
input_data = {
'username': '******',
'state': 'absent',
'waf_main_class': {
'id': '3',
'name': 'default_name_4'
},
'vdom': 'root'
}
is_error, changed, response = fortios_waf_main_class.fortios_waf(
input_data, fos_instance)
delete_method_mock.assert_called_with('waf',
'main-class',
mkey=ANY,
vdom='root')
schema_method_mock.assert_not_called()
assert is_error
assert not changed
assert response['status'] == 'error'
assert response['http_status'] == 500