From 6acc7838e250512957392fe3577f24f97c712cd2 Mon Sep 17 00:00:00 2001
From: David McDonald <david.mcdonald@digital.cabinet-office.gov.uk>
Date: Wed, 24 Nov 2021 18:07:43 +0000
Subject: [PATCH] Add tests for existing status api client

Essentially copies the tests found in the performance_platform
api client.
---
 .../notify_client/test_status_api_client.py   | 59 +++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 tests/app/notify_client/test_status_api_client.py

diff --git a/tests/app/notify_client/test_status_api_client.py b/tests/app/notify_client/test_status_api_client.py
new file mode 100644
index 000000000..f836056bb
--- /dev/null
+++ b/tests/app/notify_client/test_status_api_client.py
@@ -0,0 +1,59 @@
+from app.notify_client.status_api_client import StatusApiClient
+
+
+def test_get_count_of_live_services_and_organisations(mocker):
+    mocker.patch('app.extensions.RedisClient.get', return_value=None)
+    client = StatusApiClient()
+    mock = mocker.patch.object(client, 'get', return_value={})
+
+    client.get_count_of_live_services_and_organisations()
+
+    mock.assert_called_once_with(url='/_status/live-service-and-organisation-counts')
+
+
+def test_sets_value_in_cache(mocker):
+    client = StatusApiClient()
+
+    mock_redis_get = mocker.patch(
+        'app.extensions.RedisClient.get',
+        return_value=None
+    )
+    mock_api_get = mocker.patch(
+        'app.notify_client.NotifyAdminAPIClient.get',
+        return_value={'data_from': 'api'},
+    )
+    mock_redis_set = mocker.patch(
+        'app.extensions.RedisClient.set',
+    )
+
+    assert client.get_count_of_live_services_and_organisations() == {'data_from': 'api'}
+
+    mock_redis_get.assert_called_once_with('live-service-and-organisation-counts')
+    mock_api_get.assert_called_once_with(url='/_status/live-service-and-organisation-counts')
+    mock_redis_set.assert_called_once_with(
+        'live-service-and-organisation-counts',
+        '{"data_from": "api"}',
+        ex=604800
+    )
+
+
+def test_returns_value_from_cache(mocker):
+    client = StatusApiClient()
+
+    mock_redis_get = mocker.patch(
+        'app.extensions.RedisClient.get',
+        return_value=b'{"data_from": "cache"}',
+    )
+    mock_api_get = mocker.patch(
+        'app.notify_client.NotifyAdminAPIClient.get',
+    )
+    mock_redis_set = mocker.patch(
+        'app.extensions.RedisClient.set',
+    )
+
+    assert client.get_count_of_live_services_and_organisations() == {'data_from': 'cache'}
+
+    mock_redis_get.assert_called_once_with('live-service-and-organisation-counts')
+
+    assert mock_api_get.called is False
+    assert mock_redis_set.called is False