ae0054a79e
Refs: * https://github.com/ansible/ansible/issues/36348 * https://github.com/ansible/ansible/pull/41875
323 lines
9.6 KiB
Python
323 lines
9.6 KiB
Python
from __future__ import (absolute_import, division, print_function)
|
|
|
|
import json
|
|
import os
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
linode_apiv4 = pytest.importorskip('linode_api4')
|
|
mandatory_py_version = pytest.mark.skipif(
|
|
sys.version_info < (2, 7),
|
|
reason='The linode_api4 dependency requires python2.7 or higher'
|
|
)
|
|
|
|
from linode_api4.errors import ApiError as LinodeApiError
|
|
from linode_api4 import LinodeClient
|
|
|
|
from ansible.modules.cloud.linode import linode_v4
|
|
from ansible.module_utils.linode import get_user_agent
|
|
from units.modules.utils import set_module_args
|
|
from units.compat import mock
|
|
|
|
|
|
def test_mandatory_state_is_validated(capfd):
|
|
with pytest.raises(SystemExit):
|
|
set_module_args({'label': 'foo'})
|
|
linode_v4.initialise_module()
|
|
|
|
out, err = capfd.readouterr()
|
|
results = json.loads(out)
|
|
|
|
assert all(txt in results['msg'] for txt in ('state', 'required'))
|
|
assert results['failed'] is True
|
|
|
|
|
|
def test_mandatory_label_is_validated(capfd):
|
|
with pytest.raises(SystemExit):
|
|
set_module_args({'state': 'present'})
|
|
linode_v4.initialise_module()
|
|
|
|
out, err = capfd.readouterr()
|
|
results = json.loads(out)
|
|
|
|
assert all(txt in results['msg'] for txt in ('label', 'required'))
|
|
assert results['failed'] is True
|
|
|
|
|
|
def test_mandatory_access_token_is_validated(default_args,
|
|
no_access_token_in_env,
|
|
capfd):
|
|
with pytest.raises(SystemExit):
|
|
set_module_args(default_args)
|
|
linode_v4.initialise_module()
|
|
|
|
out, err = capfd.readouterr()
|
|
results = json.loads(out)
|
|
|
|
assert results['failed'] is True
|
|
assert all(txt in results['msg'] for txt in (
|
|
'missing',
|
|
'required',
|
|
'access_token',
|
|
))
|
|
|
|
|
|
def test_mandatory_access_token_passed_in_env(default_args,
|
|
access_token):
|
|
set_module_args(default_args)
|
|
|
|
try:
|
|
module = linode_v4.initialise_module()
|
|
except SystemExit:
|
|
pytest.fail("'access_token' is passed in environment")
|
|
|
|
now_set_token = module.params['access_token']
|
|
assert now_set_token == os.environ['LINODE_ACCESS_TOKEN']
|
|
|
|
|
|
def test_mandatory_access_token_passed_in_as_parameter(default_args,
|
|
no_access_token_in_env):
|
|
default_args.update({'access_token': 'foo'})
|
|
set_module_args(default_args)
|
|
|
|
try:
|
|
module = linode_v4.initialise_module()
|
|
except SystemExit:
|
|
pytest.fail("'access_token' is passed in as parameter")
|
|
|
|
assert module.params['access_token'] == 'foo'
|
|
|
|
|
|
def test_instance_by_label_cannot_authenticate(capfd, access_token,
|
|
default_args):
|
|
set_module_args(default_args)
|
|
module = linode_v4.initialise_module()
|
|
client = LinodeClient(module.params['access_token'])
|
|
|
|
target = 'linode_api4.linode_client.LinodeGroup.instances'
|
|
with mock.patch(target, side_effect=LinodeApiError('foo')):
|
|
with pytest.raises(SystemExit):
|
|
linode_v4.maybe_instance_from_label(module, client)
|
|
|
|
out, err = capfd.readouterr()
|
|
results = json.loads(out)
|
|
|
|
assert results['failed'] is True
|
|
assert 'Unable to query the Linode API' in results['msg']
|
|
|
|
|
|
def test_no_instances_found_with_label_gives_none(default_args,
|
|
access_token):
|
|
set_module_args(default_args)
|
|
module = linode_v4.initialise_module()
|
|
client = LinodeClient(module.params['access_token'])
|
|
|
|
target = 'linode_api4.linode_client.LinodeGroup.instances'
|
|
with mock.patch(target, return_value=[]):
|
|
result = linode_v4.maybe_instance_from_label(module, client)
|
|
|
|
assert result is None
|
|
|
|
|
|
def test_optional_region_is_validated(default_args, capfd, access_token):
|
|
default_args.update({'type': 'foo', 'image': 'bar'})
|
|
set_module_args(default_args)
|
|
|
|
with pytest.raises(SystemExit):
|
|
linode_v4.initialise_module()
|
|
|
|
out, err = capfd.readouterr()
|
|
results = json.loads(out)
|
|
|
|
assert results['failed'] is True
|
|
assert all(txt in results['msg'] for txt in (
|
|
'required',
|
|
'together',
|
|
'region'
|
|
))
|
|
|
|
|
|
def test_optional_type_is_validated(default_args, capfd, access_token):
|
|
default_args.update({'region': 'foo', 'image': 'bar'})
|
|
set_module_args(default_args)
|
|
|
|
with pytest.raises(SystemExit):
|
|
linode_v4.initialise_module()
|
|
|
|
out, err = capfd.readouterr()
|
|
results = json.loads(out)
|
|
|
|
assert results['failed'] is True
|
|
assert all(txt in results['msg'] for txt in (
|
|
'required',
|
|
'together',
|
|
'type'
|
|
))
|
|
|
|
|
|
def test_optional_image_is_validated(default_args, capfd, access_token):
|
|
default_args.update({'type': 'foo', 'region': 'bar'})
|
|
set_module_args(default_args)
|
|
|
|
with pytest.raises(SystemExit):
|
|
linode_v4.initialise_module()
|
|
|
|
out, err = capfd.readouterr()
|
|
results = json.loads(out)
|
|
|
|
assert results['failed'] is True
|
|
assert all(txt in results['msg'] for txt in (
|
|
'required',
|
|
'together',
|
|
'image'
|
|
))
|
|
|
|
|
|
def test_instance_already_created(default_args,
|
|
mock_linode,
|
|
capfd,
|
|
access_token):
|
|
default_args.update({
|
|
'type': 'foo',
|
|
'region': 'bar',
|
|
'image': 'baz'
|
|
})
|
|
set_module_args(default_args)
|
|
|
|
target = 'linode_api4.linode_client.LinodeGroup.instances'
|
|
with mock.patch(target, return_value=[mock_linode]):
|
|
with pytest.raises(SystemExit) as sys_exit_exc:
|
|
linode_v4.main()
|
|
|
|
assert sys_exit_exc.value.code == 0
|
|
|
|
out, err = capfd.readouterr()
|
|
results = json.loads(out)
|
|
|
|
assert results['changed'] is False
|
|
assert 'root_password' not in results['instance']
|
|
assert (
|
|
results['instance']['label'] ==
|
|
mock_linode._raw_json['label']
|
|
)
|
|
|
|
|
|
def test_instance_to_be_created_without_root_pass(default_args,
|
|
mock_linode,
|
|
capfd,
|
|
access_token):
|
|
default_args.update({
|
|
'type': 'foo',
|
|
'region': 'bar',
|
|
'image': 'baz'
|
|
})
|
|
set_module_args(default_args)
|
|
|
|
target = 'linode_api4.linode_client.LinodeGroup.instances'
|
|
with mock.patch(target, return_value=[]):
|
|
with pytest.raises(SystemExit) as sys_exit_exc:
|
|
target = 'linode_api4.linode_client.LinodeGroup.instance_create'
|
|
with mock.patch(target, return_value=(mock_linode, 'passw0rd')):
|
|
linode_v4.main()
|
|
|
|
assert sys_exit_exc.value.code == 0
|
|
|
|
out, err = capfd.readouterr()
|
|
results = json.loads(out)
|
|
|
|
assert results['changed'] is True
|
|
assert (
|
|
results['instance']['label'] ==
|
|
mock_linode._raw_json['label']
|
|
)
|
|
assert results['instance']['root_pass'] == 'passw0rd'
|
|
|
|
|
|
def test_instance_to_be_created_with_root_pass(default_args,
|
|
mock_linode,
|
|
capfd,
|
|
access_token):
|
|
default_args.update({
|
|
'type': 'foo',
|
|
'region': 'bar',
|
|
'image': 'baz',
|
|
'root_pass': 'passw0rd',
|
|
})
|
|
set_module_args(default_args)
|
|
|
|
target = 'linode_api4.linode_client.LinodeGroup.instances'
|
|
with mock.patch(target, return_value=[]):
|
|
with pytest.raises(SystemExit) as sys_exit_exc:
|
|
target = 'linode_api4.linode_client.LinodeGroup.instance_create'
|
|
with mock.patch(target, return_value=mock_linode):
|
|
linode_v4.main()
|
|
|
|
assert sys_exit_exc.value.code == 0
|
|
|
|
out, err = capfd.readouterr()
|
|
results = json.loads(out)
|
|
|
|
assert results['changed'] is True
|
|
assert (
|
|
results['instance']['label'] ==
|
|
mock_linode._raw_json['label']
|
|
)
|
|
assert 'root_pass' not in results['instance']
|
|
|
|
|
|
def test_instance_to_be_deleted(default_args,
|
|
mock_linode,
|
|
capfd,
|
|
access_token):
|
|
default_args.update({'state': 'absent'})
|
|
set_module_args(default_args)
|
|
|
|
target = 'linode_api4.linode_client.LinodeGroup.instances'
|
|
with mock.patch(target, return_value=[mock_linode]):
|
|
with pytest.raises(SystemExit) as sys_exit_exc:
|
|
linode_v4.main()
|
|
|
|
assert sys_exit_exc.value.code == 0
|
|
|
|
out, err = capfd.readouterr()
|
|
results = json.loads(out)
|
|
|
|
assert results['changed'] is True
|
|
assert (
|
|
results['instance']['label'] ==
|
|
mock_linode._raw_json['label']
|
|
)
|
|
|
|
|
|
def test_instance_already_deleted_no_change(default_args,
|
|
mock_linode,
|
|
capfd,
|
|
access_token):
|
|
default_args.update({'state': 'absent'})
|
|
set_module_args(default_args)
|
|
|
|
target = 'linode_api4.linode_client.LinodeGroup.instances'
|
|
with mock.patch(target, return_value=[]):
|
|
with pytest.raises(SystemExit) as sys_exit_exc:
|
|
linode_v4.main()
|
|
|
|
assert sys_exit_exc.value.code == 0
|
|
|
|
out, err = capfd.readouterr()
|
|
results = json.loads(out)
|
|
|
|
assert results['changed'] is False
|
|
assert results['instance'] == {}
|
|
|
|
|
|
def test_user_agent_created_properly():
|
|
try:
|
|
from ansible.module_utils.ansible_release import (
|
|
__version__ as ansible_version
|
|
)
|
|
except ImportError:
|
|
ansible_version = 'unknown'
|
|
|
|
expected_user_agent = 'Ansible-linode_v4_module/%s' % ansible_version
|
|
assert expected_user_agent == get_user_agent('linode_v4_module')
|