mirror of
https://github.com/ZwareBear/awx.git
synced 2026-04-12 02:51:49 -05:00
118 lines
3.7 KiB
Python
118 lines
3.7 KiB
Python
import pytest
|
|
from unittest import mock
|
|
import json
|
|
|
|
from django.core.exceptions import ValidationError
|
|
|
|
from awx.main.models import (
|
|
UnifiedJob,
|
|
InventoryUpdate,
|
|
Inventory,
|
|
Credential,
|
|
CredentialType,
|
|
InventorySource,
|
|
)
|
|
|
|
|
|
def test_cancel(mocker):
|
|
with mock.patch.object(UnifiedJob, 'cancel', return_value=True) as parent_cancel:
|
|
iu = InventoryUpdate()
|
|
|
|
iu.save = mocker.MagicMock()
|
|
build_job_explanation_mock = mocker.MagicMock()
|
|
iu._build_job_explanation = mocker.MagicMock(return_value=build_job_explanation_mock)
|
|
|
|
iu.cancel()
|
|
|
|
parent_cancel.assert_called_with(is_chain=False, job_explanation=None)
|
|
|
|
|
|
def test__build_job_explanation():
|
|
iu = InventoryUpdate(id=3, name='I_am_an_Inventory_Update')
|
|
|
|
job_explanation = iu._build_job_explanation()
|
|
|
|
assert job_explanation == 'Previous Task Canceled: {"job_type": "%s", "job_name": "%s", "job_id": "%s"}' % \
|
|
('inventory_update', 'I_am_an_Inventory_Update', 3)
|
|
|
|
|
|
def test_valid_clean_insights_credential():
|
|
cred_type = CredentialType.defaults['insights']()
|
|
insights_cred = Credential(credential_type=cred_type)
|
|
inv = Inventory(insights_credential=insights_cred)
|
|
|
|
inv.clean_insights_credential()
|
|
|
|
|
|
def test_invalid_clean_insights_credential():
|
|
cred_type = CredentialType.defaults['scm']()
|
|
cred = Credential(credential_type=cred_type)
|
|
inv = Inventory(insights_credential=cred)
|
|
|
|
with pytest.raises(ValidationError) as e:
|
|
inv.clean_insights_credential()
|
|
|
|
assert json.dumps(str(e.value)) == json.dumps(str([u"Credential kind must be 'insights'."]))
|
|
|
|
|
|
def test_valid_kind_clean_insights_credential():
|
|
inv = Inventory(kind='smart')
|
|
|
|
inv.clean_insights_credential()
|
|
|
|
|
|
def test_invalid_kind_clean_insights_credential():
|
|
cred_type = CredentialType.defaults['insights']()
|
|
insights_cred = Credential(credential_type=cred_type)
|
|
inv = Inventory(kind='smart', insights_credential=insights_cred)
|
|
|
|
with pytest.raises(ValidationError) as e:
|
|
inv.clean_insights_credential()
|
|
|
|
assert json.dumps(str(e.value)) == json.dumps(str([u'Assignment not allowed for Smart Inventory']))
|
|
|
|
|
|
@pytest.mark.parametrize('source_vars,validate_certs', [
|
|
({'ssl_verify': True}, True),
|
|
({'ssl_verify': False}, False),
|
|
({'validate_certs': True}, True),
|
|
({'validate_certs': False}, False)])
|
|
def test_satellite_plugin_backwards_support_for_ssl_verify(source_vars, validate_certs):
|
|
injector = InventorySource.injectors['satellite6']('2.9')
|
|
inv_src = InventorySource(
|
|
name='satellite source', source='satellite6',
|
|
source_vars=source_vars
|
|
)
|
|
|
|
ret = injector.inventory_as_dict(inv_src, '/tmp/foo')
|
|
assert 'validate_certs' in ret
|
|
assert ret['validate_certs'] in (validate_certs, str(validate_certs))
|
|
|
|
|
|
class TestControlledBySCM():
|
|
def test_clean_source_path_valid(self):
|
|
inv_src = InventorySource(source_path='/not_real/',
|
|
source='scm')
|
|
|
|
inv_src.clean_source_path()
|
|
|
|
@pytest.mark.parametrize('source', [
|
|
'ec2',
|
|
'manual',
|
|
])
|
|
def test_clean_source_path_invalid(self, source):
|
|
inv_src = InventorySource(source_path='/not_real/',
|
|
source=source)
|
|
|
|
with pytest.raises(ValidationError):
|
|
inv_src.clean_source_path()
|
|
|
|
def test_clean_update_on_launch_update_on_project_update(self):
|
|
inv_src = InventorySource(update_on_project_update=True,
|
|
update_on_launch=True,
|
|
source='scm')
|
|
|
|
with pytest.raises(ValidationError):
|
|
inv_src.clean_update_on_launch()
|
|
|