Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Backport PR for Check Point Unit tests for the following module from PR 62214 #62405

Merged
merged 2 commits into from Sep 17, 2019
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
20 changes: 20 additions & 0 deletions changelogs/fragments/62214-unit-tests-for-cp-modules.yaml
@@ -0,0 +1,20 @@
---
bugfixes:
- "Unit tests for cp_mgmt_global_assignment"
- "Unit tests for cp_mgmt_global_assignment_facts"
- "Unit tests for cp_mgmt_group"
- "Unit tests for cp_mgmt_group_facts"
- "Unit tests for cp_mgmt_group_with_exclusion"
- "Unit tests for cp_mgmt_group_with_exclusion_facts"
- "Unit tests for cp_mgmt_host"
- "Unit tests for cp_mgmt_host_facts"
- "Unit tests for cp_mgmt_multicast_address_range"
- "Unit tests for cp_mgmt_multicast_address_range_facts"
- "Unit tests for cp_mgmt_package"
- "Unit tests for cp_mgmt_package_facts"
- "Unit tests for cp_mgmt_security_zone"
- "Unit tests for cp_mgmt_security_zone_facts"
- "Unit tests for cp_mgmt_service_dce_rpc"
- "Unit tests for cp_mgmt_service_dce_rpc_facts"
- "Unit tests for cp_mgmt_service_group"
- "Unit tests for cp_mgmt_service_group_facts(https://github.com/ansible/ansible/pull/62214)"
@@ -0,0 +1,117 @@
# Ansible module to manage CheckPoint Firewall (c) 2019
#
# Ansible 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.
#
# Ansible 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 <http://www.gnu.org/licenses/>.
#

from __future__ import absolute_import, division, print_function
__metaclass__ = type

import pytest
from units.modules.utils import set_module_args, exit_json, fail_json, AnsibleExitJson

from ansible.module_utils import basic
from ansible.modules.network.check_point import cp_mgmt_global_assignment

OBJECT = {
"global_domain": "Global",
"dependent_domain": "domain2",
"global_access_policy": "standard",
"global_threat_prevention_policy": "standard",
"manage_protection_actions": True
}

CREATE_PAYLOAD = {
"global_domain": "Global",
"dependent_domain": "domain2",
"global_access_policy": "standard",
"global_threat_prevention_policy": "standard",
"manage_protection_actions": True
}

UPDATE_PAYLOAD = {
"global_domain": "Global2",
"dependent_domain": "domain1",
"global_threat_prevention_policy": "",
"manage_protection_actions": False
}

OBJECT_AFTER_UPDATE = UPDATE_PAYLOAD

DELETE_PAYLOAD = {}

function_path = 'ansible.modules.network.check_point.cp_mgmt_global_assignment.api_call'
api_call_object = 'global-assignment'


class TestCheckpointGlobalAssignment(object):
module = cp_mgmt_global_assignment

@pytest.fixture(autouse=True)
def module_mock(self, mocker):
return mocker.patch.multiple(basic.AnsibleModule, exit_json=exit_json, fail_json=fail_json)

@pytest.fixture
def connection_mock(self, mocker):
connection_class_mock = mocker.patch('ansible.module_utils.network.checkpoint.checkpoint.Connection')
return connection_class_mock.return_value

def test_create(self, mocker, connection_mock):
mock_function = mocker.patch(function_path)
mock_function.return_value = {'changed': True, api_call_object: OBJECT}
result = self._run_module(CREATE_PAYLOAD)

assert result['changed']
assert OBJECT.items() == result[api_call_object].items()

def test_create_idempotent(self, mocker, connection_mock):
mock_function = mocker.patch(function_path)
mock_function.return_value = {'changed': False, api_call_object: OBJECT}
result = self._run_module(CREATE_PAYLOAD)

assert not result['changed']

def test_update(self, mocker, connection_mock):
mock_function = mocker.patch(function_path)
mock_function.return_value = {'changed': True, api_call_object: OBJECT_AFTER_UPDATE}
result = self._run_module(UPDATE_PAYLOAD)

assert result['changed']
assert OBJECT_AFTER_UPDATE.items() == result[api_call_object].items()

def test_update_idempotent(self, mocker, connection_mock):
mock_function = mocker.patch(function_path)
mock_function.return_value = {'changed': False, api_call_object: OBJECT_AFTER_UPDATE}
result = self._run_module(UPDATE_PAYLOAD)

assert not result['changed']

def test_delete(self, mocker, connection_mock):
mock_function = mocker.patch(function_path)
mock_function.return_value = {'changed': True}
result = self._run_module(DELETE_PAYLOAD)

assert result['changed']

def test_delete_idempotent(self, mocker, connection_mock):
mock_function = mocker.patch(function_path)
mock_function.return_value = {'changed': False}
result = self._run_module(DELETE_PAYLOAD)

assert not result['changed']

def _run_module(self, module_args):
set_module_args(module_args)
with pytest.raises(AnsibleExitJson) as ex:
self.module.main()
return ex.value.args[0]
@@ -0,0 +1,80 @@
# Ansible module to manage CheckPoint Firewall (c) 2019
#
# Ansible 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.
#
# Ansible 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 <http://www.gnu.org/licenses/>.
#

from __future__ import absolute_import, division, print_function
__metaclass__ = type

import pytest
from units.modules.utils import set_module_args, exit_json, fail_json, AnsibleExitJson

from ansible.module_utils import basic
from ansible.modules.network.check_point import cp_mgmt_global_assignment_facts

OBJECT = {
"from": 1,
"to": 1,
"total": 6,
"objects": [
"53de74b7-8f19-4cbe-99fc-a81ef0759bad"
]
}

SHOW_PLURAL_PAYLOAD = {
'limit': 1,
'details_level': 'uid'
}

SHOW_SINGLE_PAYLOAD = {}

api_call_object = 'global-assignment'
api_call_object_plural_version = 'global-assignments'
failure_msg = '''{u'message': u'Requested object [object_which_is_not_exist] not found', u'code': u'generic_err_object_not_found'}'''


class TestCheckpointGlobalAssignmentFacts(object):
module = cp_mgmt_global_assignment_facts

@pytest.fixture(autouse=True)
def module_mock(self, mocker):
return mocker.patch.multiple(basic.AnsibleModule, exit_json=exit_json, fail_json=fail_json)

@pytest.fixture
def connection_mock(self, mocker):
connection_class_mock = mocker.patch('ansible.module_utils.network.checkpoint.checkpoint.Connection')
return connection_class_mock.return_value

def test_show_single_object_which_is_not_exist(self, mocker, connection_mock):
connection_mock.send_request.return_value = (404, failure_msg)
try:
result = self._run_module(SHOW_SINGLE_PAYLOAD)
except Exception as e:
result = e.args[0]

assert result['failed']
assert 'Checkpoint device returned error 404 with message ' + failure_msg == result['msg']

def test_show_few_objects(self, mocker, connection_mock):
connection_mock.send_request.return_value = (200, OBJECT)
result = self._run_module(SHOW_PLURAL_PAYLOAD)

assert not result['changed']
assert OBJECT == result['ansible_facts'][api_call_object_plural_version]

def _run_module(self, module_args):
set_module_args(module_args)
with pytest.raises(AnsibleExitJson) as ex:
self.module.main()
return ex.value.args[0]
117 changes: 117 additions & 0 deletions test/units/modules/network/check_point/test_cp_mgmt_group.py
@@ -0,0 +1,117 @@
# Ansible module to manage CheckPoint Firewall (c) 2019
#
# Ansible 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.
#
# Ansible 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 <http://www.gnu.org/licenses/>.
#

from __future__ import absolute_import, division, print_function
__metaclass__ = type

import pytest
from units.modules.utils import set_module_args, exit_json, fail_json, AnsibleExitJson

from ansible.module_utils import basic
from ansible.modules.network.check_point import cp_mgmt_group

OBJECT = {
"name": "New Group 5",
"members": [
"New Host 1",
"My Test Host 3"
]
}

CREATE_PAYLOAD = {
"name": "New Group 5",
"members": [
"New Host 1",
"My Test Host 3"
]
}

UPDATE_PAYLOAD = {
"name": "New Group 5"
}

OBJECT_AFTER_UPDATE = UPDATE_PAYLOAD

DELETE_PAYLOAD = {
"name": "New Group 5",
"state": "absent"
}

function_path = 'ansible.modules.network.check_point.cp_mgmt_group.api_call'
api_call_object = 'group'


class TestCheckpointGroup(object):
module = cp_mgmt_group

@pytest.fixture(autouse=True)
def module_mock(self, mocker):
return mocker.patch.multiple(basic.AnsibleModule, exit_json=exit_json, fail_json=fail_json)

@pytest.fixture
def connection_mock(self, mocker):
connection_class_mock = mocker.patch('ansible.module_utils.network.checkpoint.checkpoint.Connection')
return connection_class_mock.return_value

def test_create(self, mocker, connection_mock):
mock_function = mocker.patch(function_path)
mock_function.return_value = {'changed': True, api_call_object: OBJECT}
result = self._run_module(CREATE_PAYLOAD)

assert result['changed']
assert OBJECT.items() == result[api_call_object].items()

def test_create_idempotent(self, mocker, connection_mock):
mock_function = mocker.patch(function_path)
mock_function.return_value = {'changed': False, api_call_object: OBJECT}
result = self._run_module(CREATE_PAYLOAD)

assert not result['changed']

def test_update(self, mocker, connection_mock):
mock_function = mocker.patch(function_path)
mock_function.return_value = {'changed': True, api_call_object: OBJECT_AFTER_UPDATE}
result = self._run_module(UPDATE_PAYLOAD)

assert result['changed']
assert OBJECT_AFTER_UPDATE.items() == result[api_call_object].items()

def test_update_idempotent(self, mocker, connection_mock):
mock_function = mocker.patch(function_path)
mock_function.return_value = {'changed': False, api_call_object: OBJECT_AFTER_UPDATE}
result = self._run_module(UPDATE_PAYLOAD)

assert not result['changed']

def test_delete(self, mocker, connection_mock):
mock_function = mocker.patch(function_path)
mock_function.return_value = {'changed': True}
result = self._run_module(DELETE_PAYLOAD)

assert result['changed']

def test_delete_idempotent(self, mocker, connection_mock):
mock_function = mocker.patch(function_path)
mock_function.return_value = {'changed': False}
result = self._run_module(DELETE_PAYLOAD)

assert not result['changed']

def _run_module(self, module_args):
set_module_args(module_args)
with pytest.raises(AnsibleExitJson) as ex:
self.module.main()
return ex.value.args[0]