Remove changes to only count unique hostnames. Fixes https://trello.com/c/HN7S4KF3 again.

This commit is contained in:
Chris Church 2015-05-14 15:48:09 -04:00
parent 74cfca2f21
commit 529b7a1117
2 changed files with 10 additions and 15 deletions

View File

@ -282,16 +282,10 @@ class DashboardView(APIView):
user_hosts = get_user_queryset(request.user, Host)
user_hosts_failed = user_hosts.filter(has_active_failures=True)
try:
user_hosts_count = user_hosts.distinct('name').count()
user_hosts_failed_count = user_hosts_failed.distinct('name').count()
except NotImplementedError: # For unit tests only, SQLite doesn't support distinct('name')
user_hosts_count = len(set(user_hosts.values_list('name', flat=True)))
user_hosts_failed_count = len(set(user_hosts_failed.values_list('name', flat=True)))
data['hosts'] = {'url': reverse('api:host_list'),
'failures_url': reverse('api:host_list') + "?has_active_failures=True",
'total': user_hosts_count,
'failed': user_hosts_failed_count}
'total': user_hosts.count(),
'failed': user_hosts_failed.count()}
user_projects = get_user_queryset(request.user, Project)
user_projects_failed = user_projects.filter(last_job_failed=True)

View File

@ -1098,35 +1098,36 @@ class InventoryTest(BaseTest):
self.assertEqual(response['hosts']['total'], 0)
self.assertEqual(response['hosts']['failed'], 0)
# Create hosts with the same name in different inventories.
# Create hosts with the same name in different inventories. This host
# count should include total hosts, not unique names.
for x in xrange(4):
hostname = 'host-%d' % x
self.inventory_a.hosts.create(name=hostname)
self.inventory_b.hosts.create(name=hostname)
with self.current_user(self.super_django_user):
response = self.get(url, expect=200)
self.assertEqual(response['hosts']['total'], 4)
self.assertEqual(response['hosts']['total'], 8)
self.assertEqual(response['hosts']['failed'], 0)
# Mark all hosts in one inventory as failed. Failed count should
# reflect unique hostnames.
# reflect all hosts, not unique hostnames.
for host in self.inventory_a.hosts.all():
host.has_active_failures = True
host.save()
with self.current_user(self.super_django_user):
response = self.get(url, expect=200)
self.assertEqual(response['hosts']['total'], 4)
self.assertEqual(response['hosts']['total'], 8)
self.assertEqual(response['hosts']['failed'], 4)
# Mark all hosts in the other inventory as failed. Failed count
# should reflect unique hostnames and never be greater than total.
# should reflect all hosts and never be greater than total.
for host in self.inventory_b.hosts.all():
host.has_active_failures = True
host.save()
with self.current_user(self.super_django_user):
response = self.get(url, expect=200)
self.assertEqual(response['hosts']['total'], 4)
self.assertEqual(response['hosts']['failed'], 4)
self.assertEqual(response['hosts']['total'], 8)
self.assertEqual(response['hosts']['failed'], 8)
def test_dashboard_inventory_graph_view(self):
url = reverse('api:dashboard_inventory_graph_view')