mirror of https://github.com/ansible/ansible.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
3.1 KiB
Python
89 lines
3.1 KiB
Python
# 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_dns
|
|
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_dns.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_dnsfilter_profile_modify(fixture_data, mocker):
|
|
mocker.patch("ansible.module_utils.network.fortimanager.fortimanager.FortiManagerHandler.process_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}}
|
|
# 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_modify(fmg_instance, fixture_data[0]['paramgram_used'])
|
|
assert output['raw_response']['status']['code'] == 0
|