Auto Commit for: fmgr_secprof_spam (#53021)

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