255 lines
8.5 KiB
Python
255 lines
8.5 KiB
Python
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
import pytest
|
|
|
|
from awx.main.models import Organization, Inventory, InventorySource, Project
|
|
|
|
|
|
@pytest.fixture
|
|
def base_inventory():
|
|
org = Organization.objects.create(name='test-org')
|
|
inv = Inventory.objects.create(name='test-inv', organization=org)
|
|
return inv
|
|
|
|
|
|
@pytest.fixture
|
|
def project(base_inventory):
|
|
return Project.objects.create(
|
|
name='test-proj',
|
|
organization=base_inventory.organization,
|
|
scm_type='git',
|
|
scm_url='https://github.com/ansible/test-playbooks.git',
|
|
)
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_inventory_source_create(run_module, admin_user, base_inventory, project):
|
|
source_path = '/var/lib/awx/example_source_path/'
|
|
result = run_module('tower_inventory_source', dict(
|
|
name='foo',
|
|
inventory=base_inventory.name,
|
|
state='present',
|
|
source='scm',
|
|
source_path=source_path,
|
|
source_project=project.name
|
|
), admin_user)
|
|
assert result.pop('changed', None), result
|
|
|
|
inv_src = InventorySource.objects.get(name='foo')
|
|
assert inv_src.inventory == base_inventory
|
|
result.pop('invocation')
|
|
assert result == {
|
|
'id': inv_src.id,
|
|
'name': 'foo',
|
|
}
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_create_inventory_source_implied_org(run_module, admin_user):
|
|
org = Organization.objects.create(name='test-org')
|
|
inv = Inventory.objects.create(name='test-inv', organization=org)
|
|
|
|
# Credential is not required for ec2 source, because of IAM roles
|
|
result = run_module('tower_inventory_source', dict(
|
|
name='Test Inventory Source',
|
|
inventory='test-inv',
|
|
source='ec2',
|
|
state='present'
|
|
), admin_user)
|
|
assert result.pop('changed', None), result
|
|
|
|
inv_src = InventorySource.objects.get(name='Test Inventory Source')
|
|
assert inv_src.inventory == inv
|
|
|
|
result.pop('invocation')
|
|
assert result == {
|
|
"name": "Test Inventory Source",
|
|
"id": inv_src.id,
|
|
}
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_create_inventory_source_multiple_orgs(run_module, admin_user):
|
|
org = Organization.objects.create(name='test-org')
|
|
Inventory.objects.create(name='test-inv', organization=org)
|
|
|
|
# make another inventory by same name in another org
|
|
org2 = Organization.objects.create(name='test-org-number-two')
|
|
inv2 = Inventory.objects.create(name='test-inv', organization=org2)
|
|
|
|
result = run_module('tower_inventory_source', dict(
|
|
name='Test Inventory Source',
|
|
inventory=inv2.name,
|
|
organization='test-org-number-two',
|
|
source='ec2',
|
|
state='present'
|
|
), admin_user)
|
|
assert result.pop('changed', None), result
|
|
|
|
inv_src = InventorySource.objects.get(name='Test Inventory Source')
|
|
assert inv_src.inventory == inv2
|
|
|
|
result.pop('invocation')
|
|
assert result == {
|
|
"name": "Test Inventory Source",
|
|
"id": inv_src.id,
|
|
}
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_create_inventory_source_with_venv(run_module, admin_user, base_inventory, mocker, project):
|
|
path = '/var/lib/awx/venv/custom-venv/foobar13489435/'
|
|
source_path = '/var/lib/awx/example_source_path/'
|
|
with mocker.patch('awx.main.models.mixins.get_custom_venv_choices', return_value=[path]):
|
|
result = run_module('tower_inventory_source', dict(
|
|
name='foo',
|
|
inventory=base_inventory.name,
|
|
state='present',
|
|
source='scm',
|
|
source_project=project.name,
|
|
custom_virtualenv=path,
|
|
source_path=source_path
|
|
), admin_user)
|
|
assert result.pop('changed'), result
|
|
|
|
inv_src = InventorySource.objects.get(name='foo')
|
|
assert inv_src.inventory == base_inventory
|
|
result.pop('invocation')
|
|
|
|
assert inv_src.custom_virtualenv == path
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_custom_venv_no_op(run_module, admin_user, base_inventory, mocker, project):
|
|
"""If the inventory source is modified, then it should not blank fields
|
|
unrelated to the params that the user passed.
|
|
This enforces assumptions about the behavior of the AnsibleModule
|
|
default argument_spec behavior.
|
|
"""
|
|
source_path = '/var/lib/awx/example_source_path/'
|
|
inv_src = InventorySource.objects.create(
|
|
name='foo',
|
|
inventory=base_inventory,
|
|
source_project=project,
|
|
source='scm',
|
|
custom_virtualenv='/var/lib/awx/venv/foobar/'
|
|
)
|
|
# mock needed due to API behavior, not incorrect client behavior
|
|
with mocker.patch('awx.main.models.mixins.get_custom_venv_choices', return_value=['/var/lib/awx/venv/foobar/']):
|
|
result = run_module('tower_inventory_source', dict(
|
|
name='foo',
|
|
description='this is the changed description',
|
|
inventory=base_inventory.name,
|
|
source='scm', # is required, but behavior is arguable
|
|
state='present',
|
|
source_project=project.name,
|
|
source_path=source_path
|
|
), admin_user)
|
|
assert result.pop('changed', None), result
|
|
inv_src.refresh_from_db()
|
|
assert inv_src.custom_virtualenv == '/var/lib/awx/venv/foobar/'
|
|
assert inv_src.description == 'this is the changed description'
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_falsy_value(run_module, admin_user, base_inventory):
|
|
result = run_module('tower_inventory_source', dict(
|
|
name='falsy-test',
|
|
inventory=base_inventory.name,
|
|
source='ec2',
|
|
update_on_launch=True
|
|
), admin_user)
|
|
assert not result.get('failed', False), result.get('msg', result)
|
|
assert result.get('changed', None), result
|
|
|
|
inv_src = InventorySource.objects.get(name='falsy-test')
|
|
assert inv_src.update_on_launch is True
|
|
|
|
result = run_module('tower_inventory_source', dict(
|
|
name='falsy-test',
|
|
inventory=base_inventory.name,
|
|
source='ec2',
|
|
update_on_launch=False
|
|
), admin_user)
|
|
|
|
inv_src.refresh_from_db()
|
|
assert inv_src.update_on_launch is False
|
|
|
|
|
|
# Tests related to source-specific parameters
|
|
#
|
|
# We want to let the API return issues with "this doesn't support that", etc.
|
|
#
|
|
# GUI OPTIONS:
|
|
# - - - - - - - manual: file: scm: ec2: gce azure_rm vmware sat openstack rhv tower custom
|
|
# credential ? ? o o r r r r r r r o
|
|
# source_project ? ? r - - - - - - - - -
|
|
# source_path ? ? r - - - - - - - - -
|
|
# verbosity ? ? o o o o o o o o o o
|
|
# overwrite ? ? o o o o o o o o o o
|
|
# overwrite_vars ? ? o o o o o o o o o o
|
|
# update_on_launch ? ? o o o o o o o o o o
|
|
# UoPL ? ? o - - - - - - - - -
|
|
# source_vars* ? ? - o - o o o o - - -
|
|
# environmet vars* ? ? o - - - - - - - - o
|
|
# source_script ? ? - - - - - - - - - r
|
|
#
|
|
# UoPL - update_on_project_launch
|
|
# * - source_vars are labeled environment_vars on project and custom sources
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_missing_required_credential(run_module, admin_user, base_inventory):
|
|
result = run_module('tower_inventory_source', dict(
|
|
name='Test Azure Source',
|
|
inventory=base_inventory.name,
|
|
source='azure_rm',
|
|
state='present'
|
|
), admin_user)
|
|
assert result.pop('failed', None) is True, result
|
|
|
|
assert 'Credential is required for a cloud source' in result.get('msg', '')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_source_project_not_for_cloud(run_module, admin_user, base_inventory, project):
|
|
result = run_module('tower_inventory_source', dict(
|
|
name='Test ec2 Inventory Source',
|
|
inventory=base_inventory.name,
|
|
source='ec2',
|
|
state='present',
|
|
source_project=project.name
|
|
), admin_user)
|
|
assert result.pop('failed', None) is True, result
|
|
|
|
assert 'Cannot set source_project if not SCM type' in result.get('msg', '')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_source_path_not_for_cloud(run_module, admin_user, base_inventory):
|
|
result = run_module('tower_inventory_source', dict(
|
|
name='Test ec2 Inventory Source',
|
|
inventory=base_inventory.name,
|
|
source='ec2',
|
|
state='present',
|
|
source_path='where/am/I'
|
|
), admin_user)
|
|
assert result.pop('failed', None) is True, result
|
|
|
|
assert 'Cannot set source_path if not SCM type' in result.get('msg', '')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_scm_source_needs_project(run_module, admin_user, base_inventory):
|
|
result = run_module('tower_inventory_source', dict(
|
|
name='SCM inventory without project',
|
|
inventory=base_inventory.name,
|
|
state='present',
|
|
source='scm',
|
|
source_path='/var/lib/awx/example_source_path/'
|
|
), admin_user)
|
|
assert result.pop('failed', None), result
|
|
|
|
assert 'Project required for scm type sources' in result.get('msg', '')
|