mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-22 16:31:15 -05:00
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
import requests_mock
|
|
import pytest
|
|
|
|
from app.clients.performance_platform.performance_platform_client import PerformancePlatformClient
|
|
|
|
|
|
@pytest.fixture(scope='function')
|
|
def client(mocker):
|
|
client = PerformancePlatformClient()
|
|
current_app = mocker.Mock(config={
|
|
'PERFORMANCE_PLATFORM_ENABLED': True,
|
|
'PERFORMANCE_PLATFORM_URL': 'performance-platform-url',
|
|
'PERFORMANCE_PLATFORM_TOKEN': 'token'
|
|
})
|
|
client.init_app(current_app)
|
|
return client
|
|
|
|
|
|
def test_should_not_call_if_not_enabled(notify_api, client, mocker):
|
|
mocker.patch.object(client, '_send_stats_to_performance_platform')
|
|
client.active = False
|
|
client.send_performance_stats(
|
|
date='2016-10-16T00:00:00+00:00',
|
|
channel='sms',
|
|
count=142,
|
|
period='day'
|
|
)
|
|
|
|
client._send_stats_to_performance_platform.assert_not_called()
|
|
|
|
|
|
def test_should_call_if_enabled(notify_api, client, mocker):
|
|
mocker.patch.object(client, '_send_stats_to_performance_platform')
|
|
client.send_performance_stats(
|
|
date='2016-10-16T00:00:00+00:00',
|
|
channel='sms',
|
|
count=142,
|
|
period='day'
|
|
)
|
|
|
|
assert client._send_stats_to_performance_platform.call_count == 1
|
|
|
|
|
|
def test_send_platform_stats_creates_correct_call(notify_api, client):
|
|
with requests_mock.Mocker() as request_mock:
|
|
request_mock.post(
|
|
client.performance_platform_url,
|
|
json={},
|
|
status_code=200
|
|
)
|
|
client.send_performance_stats(
|
|
date='2016-10-16T00:00:00+00:00',
|
|
channel='sms',
|
|
count=142,
|
|
period='day'
|
|
)
|
|
|
|
assert request_mock.call_count == 1
|
|
|
|
assert request_mock.request_history[0].url == client.performance_platform_url
|
|
assert request_mock.request_history[0].method == 'POST'
|
|
|
|
request_args = request_mock.request_history[0].json()
|
|
assert request_args['dataType'] == 'notifications'
|
|
assert request_args['service'] == 'govuk-notify'
|
|
assert request_args['period'] == 'day'
|
|
assert request_args['channel'] == 'sms'
|
|
assert request_args['_timestamp'] == '2016-10-16T00:00:00+00:00'
|
|
assert request_args['count'] == 142
|
|
expected_base64_id = 'MjAxNi0xMC0xNlQwMDowMDowMCswMDowMGdvdnVrLW5vdGlmeXNtc25vdGlmaWNhdGlvbnNkYXk='
|
|
assert request_args['_id'] == expected_base64_id
|