notifications-api/tests/app/status/test_status.py

import pytest
from flask import json

from tests.app.db import create_organization, create_service


@pytest.mark.parametrize('path', ['/', '/_status'])
def test_get_status_all_ok(client, notify_db_session, path):
    response = client.get(path)
    assert response.status_code == 200
    resp_json = json.loads(response.get_data(as_text=True))
    assert resp_json['status'] == 'ok'
    assert resp_json['db_version']
    assert resp_json['git_commit']
    assert resp_json['build_time']


def test_empty_live_service_and_organization_counts(admin_request):
    assert admin_request.get('status.live_service_and_organization_counts') == {
        'organizations': 0,
        'services': 0,
    }


def test_populated_live_service_and_organization_counts(admin_request):

    # Org 1 has three real live services and one fake, for a total of 3
    org_1 = create_organization('org 1')
    live_service_1 = create_service(service_name='1')
    live_service_1.organization = org_1
    live_service_2 = create_service(service_name='2')
    live_service_2.organization = org_1
    live_service_3 = create_service(service_name='3')
    live_service_3.organization = org_1
    fake_live_service_1 = create_service(service_name='f1', count_as_live=False)
    fake_live_service_1.organization = org_1
    inactive_service_1 = create_service(service_name='i1', active=False)
    inactive_service_1.organization = 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_organization('org 2')
    trial_service_1 = create_service(service_name='t1', restricted=True)
    trial_service_1.organization = org_2
    fake_live_service_2 = create_service(service_name='f2', count_as_live=False)
    fake_live_service_2.organization = org_2
    inactive_service_2 = create_service(service_name='i2', active=False)
    inactive_service_2.organization = org_2

    # Org 2 has no services at all
    create_organization('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_organization_counts') == {
        'organizations': 1,
        'services': 4,
    }