External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-05-10 19:07:36 -02:30
Code Issues Packages Projects Releases Wiki Activity

Refactor Host manager and dynamic Inventory tests and update

Browse Source 
validation/serialization
This commit is contained in:
Wayne Witzel III
2017-05-01 12:40:54 -04:00
parent 8a599d9754
commit 1750e5bd2a
4 changed files with 53 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
awx/main/tests/functional/models/test_inventory.py
View File

@@ -37,11 +37,8 @@ class TestSCMUpdateFeatures:
assert not mck_update.called
@pytest.mark.django_db
def test_host_objects_manager(organization):
dynamic_inventory = Inventory(organization=organization, name='dynamic', host_filter='inventory_sources__source=ec2')
dynamic_inventory.save()
@pytest.fixture
def setup_ec2_gce(organization):
ec2_inv = Inventory(name='test_ec2', organization=organization)
ec2_inv.save()
@@ -59,7 +56,35 @@ def test_host_objects_manager(organization):
gce_host.inventory_sources.add(gce_source)
gce_inv.save()
hosts = dynamic_inventory.hosts.all()
assert len(hosts) == 2
assert hosts[0].inventory_sources.first() == ec2_source
assert hosts[1].inventory_sources.first() == ec2_source
@pytest.mark.django_db
class TestHostManager:
def test_host_filter_change(self, setup_ec2_gce, organization):
dynamic_inventory = Inventory(name='dynamic',
kind='dynamic',
organization=organization,
host_filter='inventory_sources__source=ec2')
dynamic_inventory.save()
assert len(dynamic_inventory.hosts.all()) == 2
dynamic_inventory.host_filter = 'inventory_sources__source=gce'
dynamic_inventory.save()
assert len(dynamic_inventory.hosts.all()) == 1
def test_host_filter_not_dynamic(self, setup_ec2_gce, organization):
dynamic_inventory = Inventory(name='dynamic',
organization=organization,
host_filter='inventory_sources__source=ec2')
assert len(dynamic_inventory.hosts.all()) == 0
def test_host_objects_manager(self, setup_ec2_gce, organization):
dynamic_inventory = Inventory(kind='dynamic',
name='dynamic',
organization=organization,
host_filter='inventory_sources__source=ec2')
dynamic_inventory.save()
hosts = dynamic_inventory.hosts.all()
assert len(hosts) == 2
assert hosts[0].inventory_sources.first().source == 'ec2'
assert hosts[1].inventory_sources.first().source == 'ec2'