mirror of
https://github.com/GSA/notifications-api.git
synced 2026-07-17 21:20:59 -04:00
89f4f5173e
so that it doesn't appear generic when it's actually specific to sending the daily notification totals. To do this, split it out into a separate performance_platform directory, containing the business logic, and make the performance_platform_client incredibly thin - all it handles is adding ids to payloads, and sending stats. Also, some changes to the config (not all done yet) since there is one token per endpoint, not one for the whole platform as we'd previously coded
60 lines
2.2 KiB
Python
60 lines
2.2 KiB
Python
import requests
|
|
import requests_mock
|
|
import pytest
|
|
|
|
from app.clients.performance_platform.performance_platform_client import PerformancePlatformClient
|
|
|
|
|
|
@pytest.fixture(scope='function')
|
|
def perf_client(mocker):
|
|
perf_client = PerformancePlatformClient()
|
|
current_app = mocker.Mock(config={
|
|
'PERFORMANCE_PLATFORM_ENABLED': True,
|
|
'PERFORMANCE_PLATFORM_ENDPOINTS': {
|
|
'foo': 'my_token',
|
|
'bar': 'other_token'
|
|
},
|
|
'PERFORMANCE_PLATFORM_URL': 'https://performance-platform-url/'
|
|
})
|
|
perf_client.init_app(current_app)
|
|
return perf_client
|
|
|
|
|
|
def test_should_not_call_if_not_enabled(perf_client):
|
|
with requests_mock.Mocker() as request_mock:
|
|
request_mock.post('https://performance-platform-url/foo', json={}, status_code=200)
|
|
perf_client._active = False
|
|
perf_client.send_stats_to_performance_platform(dataset='foo', payload={})
|
|
|
|
assert request_mock.called is False
|
|
|
|
|
|
|
|
def test_should_call_if_enabled(perf_client):
|
|
with requests_mock.Mocker() as request_mock:
|
|
request_mock.post('https://performance-platform-url/foo', json={}, status_code=200)
|
|
perf_client.send_stats_to_performance_platform(dataset='foo', payload={})
|
|
|
|
assert request_mock.call_count == 1
|
|
assert request_mock.last_request.method == 'POST'
|
|
|
|
|
|
@pytest.mark.parametrize('dataset, token', [
|
|
('foo', 'my_token'),
|
|
('bar', 'other_token')
|
|
])
|
|
def test_should_use_correct_token(perf_client, dataset, token):
|
|
with requests_mock.Mocker() as request_mock:
|
|
request_mock.post('https://performance-platform-url/foo', json={}, status_code=200)
|
|
request_mock.post('https://performance-platform-url/bar', json={}, status_code=200)
|
|
perf_client.send_stats_to_performance_platform(dataset=dataset, payload={})
|
|
|
|
assert request_mock.call_count == 1
|
|
assert request_mock.last_request.headers.get('authorization') == 'Bearer {}'.format(token)
|
|
|
|
|
|
def test_should_raise_for_status(perf_client):
|
|
with pytest.raises(requests.HTTPError), requests_mock.Mocker() as request_mock:
|
|
request_mock.post('https://performance-platform-url/foo', json={}, status_code=403)
|
|
perf_client.send_stats_to_performance_platform(dataset='foo', payload={})
|