FortiManager DNS security profile (#46239)

* Initial Commit

* Initial Commit

* Initial Commit

* Fixing syntax errors

* After running sanity tests, fixed pylint and pep8 errors

* After running sanity tests, fixed pylint and pep8 errors

* Fixing more syntax issues

* Fixing more syntax issues

* Adding username to doc block

* PR candidate

* PR candidate

* fixed pep8 and docs issues

* fixed 2.6 function issues

* fixed 2.6 function issues part duex

* Initial commit for security profile group module

* Adding better playbook example to module

* Initial commit for FMG DNS security profile

* Fixing pep8 line too long in unit test

* Removing excess modules and unit tests

* Fixing trailing white space for dns module

* Fixing utility function with nested dictionaries

* Fixing utility function with pep8 issue

* Adding change to allow for multiple list of dictionaries to be submitted via a single playbook

* Fixing review changes

* Adding @ in author names per @Gundalow's request per Ansible's guidelines
This commit is contained in:
ftntcorecse 2018-11-20 22:47:37 -07:00 committed by Sumit Jaiswal
commit 64c4881d58
4 changed files with 653 additions and 192 deletions

View file

@ -0,0 +1,76 @@
{
"fmgr_dnsfilter_profile_addsetdelete": [
{
"paramgram_used": {
"comment": "Created by Ansible Module TEST",
"ftgd-dns": {
"options": null,
"filters": {
"action": null,
"category": null,
"log": null
}
},
"name": "Ansible_DNS_Profile",
"adom": "root",
"redirect-portal": null,
"sdns-ftgd-err-log": null,
"youtube-restrict": null,
"sdns-domain-log": null,
"domain-filter": {
"domain-filter-table": null
},
"log-all-domain": null,
"mode": "delete",
"block-botnet": null,
"safe-search": null,
"external-ip-blocklist": null,
"block-action": null
},
"raw_response": {
"status": {
"message": "OK",
"code": 0
},
"url": "/pm/config/adom/root/obj/dnsfilter/profile/Ansible_DNS_Profile"
},
"post_method": "delete"
},
{
"raw_response": {
"status": {
"message": "OK",
"code": 0
},
"url": "/pm/config/adom/root/obj/dnsfilter/profile"
},
"paramgram_used": {
"comment": "Created by Ansible Module TEST",
"ftgd-dns": {
"options": null,
"filters": {
"action": null,
"category": null,
"log": null
}
},
"adom": "root",
"youtube-restrict": null,
"sdns-domain-log": null,
"block-botnet": null,
"external-ip-blocklist": null,
"block-action": "block",
"name": "Ansible_DNS_Profile",
"redirect-portal": null,
"sdns-ftgd-err-log": null,
"safe-search": null,
"domain-filter": {
"domain-filter-table": null
},
"log-all-domain": null,
"mode": "set"
},
"post_method": "set"
}
]
}

View file

@ -0,0 +1,94 @@
# 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 pyFMG.fortimgr import FortiManager
import pytest
try:
from ansible.modules.network.fortimanager import fmgr_secprof_dns
except ImportError:
pytest.skip("Could not load required modules for testing", allow_module_level=True)
fmg_instance = FortiManager("1.1.1.1", "admin", "")
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(scope="function", params=load_fixtures())
def fixture_data(request):
func_name = request.function.__name__.replace("test_", "")
return request.param.get(func_name, None)
def test_fmgr_dnsfilter_profile_addsetdelete(fixture_data, mocker):
mocker.patch("pyFMG.fortimgr.FortiManager._post_request", side_effect=fixture_data)
# Fixture sets used:###########################
##################################################
# comment: Created by Ansible Module TEST
# ftgd-dns: {'options': None, 'filters': {'action': None, 'category': None, 'log': None}}
# name: Ansible_DNS_Profile
# adom: root
# redirect-portal: None
# sdns-ftgd-err-log: None
# youtube-restrict: None
# sdns-domain-log: None
# domain-filter: {'domain-filter-table': None}
# log-all-domain: None
# mode: delete
# block-botnet: None
# safe-search: None
# external-ip-blocklist: None
# block-action: None
##################################################
##################################################
# comment: Created by Ansible Module TEST
# ftgd-dns: {'options': None, 'filters': {'action': None, 'category': None, 'log': None}}
# adom: root
# youtube-restrict: None
# sdns-domain-log: None
# block-botnet: None
# external-ip-blocklist: None
# block-action: block
# name: Ansible_DNS_Profile
# redirect-portal: None
# sdns-ftgd-err-log: None
# safe-search: None
# domain-filter: {'domain-filter-table': None}
# log-all-domain: None
# mode: set
##################################################
# Test using fixture 1 #
output = fmgr_secprof_dns.fmgr_dnsfilter_profile_addsetdelete(fmg_instance, fixture_data[0]['paramgram_used'])
assert output['raw_response']['status']['code'] == 0
# Test using fixture 2 #
output = fmgr_secprof_dns.fmgr_dnsfilter_profile_addsetdelete(fmg_instance, fixture_data[1]['paramgram_used'])
assert output['raw_response']['status']['code'] == 0