e25269e1b3
* Adding device_facts module for contribution * changes added for pylint error * Updated code to solve ansible-test compile error * Changes to avoide comile error added * Review Comments changes update * Avoided blank line * pylint error changes * Removed ansible_facts return in error case * Updated description * modules renamed * changing from ansible_facts to device_info * avoide pep8 error * Updated sample output * version changed to 2.9 * Changed Copyright license to BSD * Changed 3-clause BSD license to 2-clause BSD * Added unit test support for ome_device_info * version change * removed pylint error in unit test modules * Avoid Sanity error for unit test modules * updated version
78 lines
3.7 KiB
Python
78 lines
3.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
#
|
|
# Dell EMC OpenManage Ansible Modules
|
|
# Version 2.0
|
|
# Copyright (C) 2019 Dell Inc.
|
|
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
# All rights reserved. Dell, EMC, and other trademarks are trademarks of Dell Inc. or its subsidiaries.
|
|
# Other trademarks may be trademarks of their respective owners.
|
|
#
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import pytest
|
|
from ansible.module_utils.urls import ConnectionError, SSLValidationError
|
|
from ansible.module_utils.six.moves.urllib.error import URLError, HTTPError
|
|
from ansible.module_utils.remote_management.dellemc.ome import RestOME
|
|
from units.compat.mock import MagicMock
|
|
import json
|
|
|
|
|
|
class TestRestOME(object):
|
|
@pytest.fixture
|
|
def mock_response(self):
|
|
mock_response = MagicMock()
|
|
mock_response.getcode.return_value = 200
|
|
mock_response.headers = mock_response.getheaders.return_value = {'X-Auth-Token': 'token_id'}
|
|
mock_response.read.return_value = json.dumps({"value": "data"})
|
|
return mock_response
|
|
|
|
def test_invoke_request_with_session(self, mock_response, mocker):
|
|
mocker.patch('ansible.module_utils.remote_management.dellemc.ome.open_url',
|
|
return_value=mock_response)
|
|
module_params = {'hostname': '192.168.0.1', 'username': 'username',
|
|
'password': 'password', "port": 443}
|
|
req_session = True
|
|
with RestOME(module_params, req_session) as obj:
|
|
response = obj.invoke_request("/testpath", "GET")
|
|
assert response.status_code == 200
|
|
assert response.json_data == {"value": "data"}
|
|
assert response.success is True
|
|
|
|
def test_invoke_request_without_session(self, mock_response, mocker):
|
|
mocker.patch('ansible.module_utils.remote_management.dellemc.ome.open_url',
|
|
return_value=mock_response)
|
|
module_params = {'hostname': '192.168.0.1', 'username': 'username',
|
|
'password': 'password', "port": 443}
|
|
req_session = False
|
|
with RestOME(module_params, req_session) as obj:
|
|
response = obj.invoke_request("/testpath", "GET")
|
|
assert response.status_code == 200
|
|
assert response.json_data == {"value": "data"}
|
|
assert response.success is True
|
|
|
|
@pytest.mark.parametrize("exc", [URLError, SSLValidationError, ConnectionError])
|
|
def test_invoke_request_error_case_handling(self, exc, mock_response, mocker):
|
|
open_url_mock = mocker.patch('ansible.module_utils.remote_management.dellemc.ome.open_url',
|
|
return_value=mock_response)
|
|
open_url_mock.side_effect = exc("test")
|
|
module_params = {'hostname': '192.168.0.1', 'username': 'username',
|
|
'password': 'password', "port": 443}
|
|
req_session = False
|
|
with pytest.raises(exc) as e:
|
|
with RestOME(module_params, req_session) as obj:
|
|
obj.invoke_request("/testpath", "GET")
|
|
|
|
def test_invoke_request_http_error_handling(self, mock_response, mocker):
|
|
open_url_mock = mocker.patch('ansible.module_utils.remote_management.dellemc.ome.open_url',
|
|
return_value=mock_response)
|
|
open_url_mock.side_effect = HTTPError('http://testhost.com/', 400,
|
|
'Bad Request Error', {}, None)
|
|
module_params = {'hostname': '192.168.0.1', 'username': 'username',
|
|
'password': 'password', "port": 443}
|
|
req_session = False
|
|
with pytest.raises(HTTPError) as e:
|
|
with RestOME(module_params, req_session) as obj:
|
|
obj.invoke_request("/testpath", "GET")
|