simplify query for active_count

This commit is contained in:
AlanCoding
2017-12-12 14:25:58 -05:00
parent 07cfa6cba5
commit 98f8faa349
2 changed files with 34 additions and 6 deletions

View File

@@ -35,6 +35,30 @@ class TestInventoryScript:
}
@pytest.mark.django_db
class TestActiveCount:
def test_host_active_count(self, organization):
inv1 = Inventory.objects.create(name='inv1', organization=organization)
inv2 = Inventory.objects.create(name='inv2', organization=organization)
assert Host.objects.active_count() == 0
inv1.hosts.create(name='host1')
inv2.hosts.create(name='host1')
assert Host.objects.active_count() == 1
inv1.hosts.create(name='host2')
assert Host.objects.active_count() == 2
def test_active_count_minus_tower(self, inventory):
inventory.hosts.create(name='locally-managed-host')
source = inventory.inventory_sources.create(
name='tower-source', source='tower'
)
source.hosts.create(
name='remotely-managed-host', inventory=inventory
)
assert Host.objects.active_count() == 1
@pytest.mark.django_db
class TestSCMUpdateFeatures: