darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 09:51:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add method to count of live services and orgs

Browse Source 
So that we don’t have to update the home page manually any more.
This commit is contained in:
Chris Hill-Scott
2019-04-11 13:38:21 +01:00
parent 3845e90ee4
commit 6bfd999de2
5 changed files with 83 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
tests/app/status/test_status.py
View File

@@ -1,6 +1,8 @@
import pytest
from flask import json
from tests.app.db import create_organisation, create_service
@pytest.mark.parametrize('path', ['/', '/_status'])
def test_get_status_all_ok(client, notify_db_session, path):
@@ -12,3 +14,55 @@ def test_get_status_all_ok(client, notify_db_session, path):
assert resp_json['travis_commit']
assert resp_json['travis_build_number']
assert resp_json['build_time']
def test_empty_live_service_and_organisation_counts(admin_request):
assert admin_request.get('status.live_service_and_organisation_counts') == {
'organisations': 0,
'services': 0,
}
def test_populated_live_service_and_organisation_counts(admin_request):
# Org 1 has three real live services and one fake, for a total of 3
org_1 = create_organisation('org 1')
live_service_1 = create_service(service_name='1')
live_service_1.organisation = org_1
live_service_2 = create_service(service_name='2')
live_service_2.organisation = org_1
live_service_3 = create_service(service_name='3')
live_service_3.organisation = org_1
fake_live_service_1 = create_service(service_name='f1', count_as_live=False)
fake_live_service_1.organisation = org_1
inactive_service_1 = create_service(service_name='i1', active=False)
inactive_service_1.organisation = org_1
# This service isn’t associated to an org, but should still be counted as live
create_service(service_name='4')
# Org 2 has no real live services
org_2 = create_organisation('org 2')
trial_service_1 = create_service(service_name='t1', restricted=True)
trial_service_1.organisation = org_2
fake_live_service_2 = create_service(service_name='f2', count_as_live=False)
fake_live_service_2.organisation = org_2
inactive_service_2 = create_service(service_name='i2', active=False)
inactive_service_2.organisation = org_2
# Org 2 has no services at all
create_organisation('org 3')
# This service isn’t associated to an org, and should not be counted as live
# because it’s marked as not counted
create_service(service_name='f3', count_as_live=False)
# This service isn’t associated to an org, and should not be counted as live
# because it’s in trial mode
create_service(service_name='t', restricted=True)
create_service(service_name='i', restricted=False, active=False)
assert admin_request.get('status.live_service_and_organisation_counts') == {
'organisations': 1,
'services': 4,
}