summaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/units/modules/network/fortimanager/fixtures/test_fmgr_secprof_spam.json157
-rw-r--r--test/units/modules/network/fortimanager/test_fmgr_secprof_spam.py72
2 files changed, 229 insertions, 0 deletions
diff --git a/test/units/modules/network/fortimanager/fixtures/test_fmgr_secprof_spam.json b/test/units/modules/network/fortimanager/fixtures/test_fmgr_secprof_spam.json
new file mode 100644
index 0000000000..d75156e1de
--- /dev/null
+++ b/test/units/modules/network/fortimanager/fixtures/test_fmgr_secprof_spam.json
@@ -0,0 +1,157 @@
+{
+ "fmgr_spamfilter_profile_modify": [
+ {
+ "paramgram_used": {
+ "comment": null,
+ "smtp": {
+ "local-override": null,
+ "hdrip": null,
+ "log": null,
+ "tag-type": null,
+ "tag-msg": null,
+ "action": null
+ },
+ "spam-log": null,
+ "gmail": {
+ "log": null
+ },
+ "spam-bword-table": null,
+ "mapi": {
+ "action": null,
+ "log": null
+ },
+ "flow-based": null,
+ "spam-mheader-table": null,
+ "spam-log-fortiguard-response": null,
+ "yahoo-mail": {
+ "log": null
+ },
+ "adom": "root",
+ "pop3": {
+ "action": null,
+ "tag-msg": null,
+ "tag-type": null,
+ "log": null
+ },
+ "external": null,
+ "spam-rbl-table": null,
+ "imap": {
+ "action": null,
+ "tag-msg": null,
+ "tag-type": null,
+ "log": null
+ },
+ "spam-iptrust-table": null,
+ "replacemsg-group": null,
+ "name": "Ansible_Spam_Filter_Profile",
+ "spam-bwl-table": null,
+ "spam-filtering": null,
+ "msn-hotmail": {
+ "log": null
+ },
+ "spam-bword-threshold": null,
+ "mode": "delete",
+ "options": null
+ },
+ "datagram_sent": {},
+ "raw_response": {
+ "status": {
+ "message": "OK",
+ "code": 0
+ },
+ "url": "/pm/config/adom/root/obj/spamfilter/profile/Ansible_Spam_Filter_Profile"
+ },
+ "post_method": "delete"
+ },
+ {
+ "raw_response": {
+ "status": {
+ "message": "OK",
+ "code": 0
+ },
+ "url": "/pm/config/adom/root/obj/spamfilter/profile"
+ },
+ "datagram_sent": {
+ "comment": "Created by Ansible",
+ "spam-log-fortiguard-response": "enable",
+ "spam-log": "enable",
+ "name": "Ansible_Spam_Filter_Profile",
+ "spam-filtering": "enable",
+ "flow-based": "enable",
+ "spam-bword-threshold": 10,
+ "external": "enable",
+ "options": [
+ "bannedword",
+ "spamfsip",
+ "spamfsurl",
+ "spamrbl",
+ "spamfsphish",
+ "spambwl"
+ ],
+ "gmail": {
+ "log": "enable"
+ }
+ },
+ "paramgram_used": {
+ "comment": "Created by Ansible",
+ "smtp": {
+ "local-override": null,
+ "hdrip": null,
+ "log": null,
+ "tag-type": null,
+ "tag-msg": null,
+ "action": null
+ },
+ "yahoo-mail": {
+ "log": null
+ },
+ "gmail": {
+ "log": "enable"
+ },
+ "spam-bword-table": null,
+ "mapi": {
+ "action": null,
+ "log": null
+ },
+ "flow-based": "enable",
+ "spam-mheader-table": null,
+ "spam-log-fortiguard-response": "enable",
+ "spam-log": "enable",
+ "adom": "root",
+ "pop3": {
+ "action": null,
+ "tag-type": null,
+ "log": null,
+ "tag-msg": null
+ },
+ "external": "enable",
+ "spam-rbl-table": null,
+ "imap": {
+ "action": null,
+ "tag-type": null,
+ "log": null,
+ "tag-msg": null
+ },
+ "spam-iptrust-table": null,
+ "name": "Ansible_Spam_Filter_Profile",
+ "replacemsg-group": null,
+ "spam-bwl-table": null,
+ "spam-filtering": "enable",
+ "msn-hotmail": {
+ "log": null
+ },
+ "spam-bword-threshold": 10,
+ "mode": "set",
+ "options": [
+ "bannedword",
+ "spamfsip",
+ "spamfsurl",
+ "spamrbl",
+ "spamfsphish",
+ "spambwl"
+ ]
+ },
+ "post_method": "set"
+ }
+ ]
+}
diff --git a/test/units/modules/network/fortimanager/test_fmgr_secprof_spam.py b/test/units/modules/network/fortimanager/test_fmgr_secprof_spam.py
new file mode 100644
index 0000000000..9db77c7430
--- /dev/null
+++ b/test/units/modules/network/fortimanager/test_fmgr_secprof_spam.py
@@ -0,0 +1,72 @@
+# Copyright 2018 Fortinet, Inc.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Ansible. If not, see <https://www.gnu.org/licenses/>.
+
+# Make coding more python3-ish
+from __future__ import (absolute_import, division, print_function)
+__metaclass__ = type
+
+import os
+import json
+from ansible.module_utils.network.fortimanager.fortimanager import FortiManagerHandler
+import pytest
+
+try:
+ from ansible.modules.network.fortimanager import fmgr_secprof_spam
+except ImportError:
+ pytest.skip("Could not load required modules for testing", allow_module_level=True)
+
+
+def load_fixtures():
+ fixture_path = os.path.join(os.path.dirname(__file__), 'fixtures') + "/{filename}.json".format(
+ filename=os.path.splitext(os.path.basename(__file__))[0])
+ try:
+ with open(fixture_path, "r") as fixture_file:
+ fixture_data = json.load(fixture_file)
+ except IOError:
+ return []
+ return [fixture_data]
+
+
+@pytest.fixture(autouse=True)
+def module_mock(mocker):
+ connection_class_mock = mocker.patch('ansible.module_utils.basic.AnsibleModule')
+ return connection_class_mock
+
+
+@pytest.fixture(autouse=True)
+def connection_mock(mocker):
+ connection_class_mock = mocker.patch('ansible.modules.network.fortimanager.fmgr_secprof_spam.Connection')
+ return connection_class_mock
+
+
+@pytest.fixture(scope="function", params=load_fixtures())
+def fixture_data(request):
+ func_name = request.function.__name__.replace("test_", "")
+ return request.param.get(func_name, None)
+
+
+fmg_instance = FortiManagerHandler(connection_mock, module_mock)
+
+
+def test_fmgr_spamfilter_profile_modify(fixture_data, mocker):
+ mocker.patch("ansible.module_utils.network.fortimanager.fortimanager.FortiManagerHandler.process_request",
+ side_effect=fixture_data)
+
+ # Test using fixture 1 #
+ output = fmgr_secprof_spam.fmgr_spamfilter_profile_modify(fmg_instance, fixture_data[0]['paramgram_used'])
+ assert output['raw_response']['status']['code'] == 0
+ # Test using fixture 2 #
+ output = fmgr_secprof_spam.fmgr_spamfilter_profile_modify(fmg_instance, fixture_data[1]['paramgram_used'])
+ assert output['raw_response']['status']['code'] == 0
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-30 08:33:19 +0100