separate detailed and normal service_api_client.get_service

to make it easier to mock and control return values
This commit is contained in:
Leo Hemsted
2016-07-20 11:46:29 +01:00
parent 3e6eedd079
commit 3ffd6c744c
8 changed files with 45 additions and 25 deletions

View File

@@ -148,8 +148,7 @@ def get_dashboard_partials(service_id):
lambda job: job['original_file_name'] != current_app.config['TEST_MESSAGE_FILENAME'], lambda job: job['original_file_name'] != current_app.config['TEST_MESSAGE_FILENAME'],
job_api_client.get_job(service_id, limit_days=7)['data'] job_api_client.get_job(service_id, limit_days=7)['data']
)) ))
service = service_api_client.get_detailed_service(service_id)
service = service_api_client.get_service(service_id, detailed=True)
return { return {
'totals': render_template( 'totals': render_template(
@@ -178,10 +177,10 @@ def get_dashboard_partials(service_id):
def get_dashboard_totals(service): def get_dashboard_totals(service):
for msg_type in service['statistics']: for msg_type in service['data']['statistics'].values():
msg_type['failed_percentage'] = get_formatted_percentage(msg_type['failed'], msg_type['requested']) msg_type['failed_percentage'] = get_formatted_percentage(msg_type['failed'], msg_type['requested'])
msg_type['show_warning'] = msg_type['failed_percentage'] > 3 msg_type['show_warning'] = float(msg_type['failed_percentage']) > 3
return service['statistics'] return service['data']['statistics']
def calculate_usage(usage): def calculate_usage(usage):

View File

@@ -41,7 +41,13 @@ class ServiceAPIClient(NotificationsAPIClient):
data = _attach_current_user({}) data = _attach_current_user({})
return self.delete(endpoint, data) return self.delete(endpoint, data)
def get_service(self, service_id, detailed=False): def get_service(self, service_id):
return self._get_service(service_id, False)
def get_detailed_service(self, service_id):
return self._get_service(service_id, True)
def _get_service(self, service_id, detailed):
""" """
Retrieve a service. Retrieve a service.
""" """

View File

@@ -31,9 +31,9 @@ def sum_of_statistics(delivery_statistics):
def add_rates_to(delivery_statistics): def add_rates_to(delivery_statistics):
return dict( return dict(
email_failure_rate=get_formatted_percentage( emails_failure_rate=get_formatted_percentage(
delivery_statistics['email_failed'], delivery_statistics['emails_failed'],
delivery_statistics['email_requested']), delivery_statistics['emails_requested']),
sms_failure_rate=get_formatted_percentage( sms_failure_rate=get_formatted_percentage(
delivery_statistics['sms_failed'], delivery_statistics['sms_failed'],
delivery_statistics['sms_requested']), delivery_statistics['sms_requested']),

View File

@@ -9,8 +9,8 @@
statistics['email']['failed'], statistics['email']['failed'],
statistics['email']['failed_percentage'], statistics['email']['failed_percentage'],
statistics['email']['show_warning'], statistics['email']['show_warning'],
failure_link=url_for(".view_notifications", service_id=id, message_type='email', status='failed'), failure_link=url_for(".view_notifications", service_id=service_id, message_type='email', status='failed'),
link=url_for(".view_notifications", service_id=id, message_type='email', status='sending,delivered,failed') link=url_for(".view_notifications", service_id=service_id, message_type='email', status='sending,delivered,failed')
) }} ) }}
</div> </div>
<div class="column-half"> <div class="column-half">

View File

@@ -13,11 +13,11 @@ class TestClient(FlaskClient):
session['user_id'] = user.id session['user_id'] = user.id
session['_fresh'] = True session['_fresh'] = True
if mocker: if mocker:
get_user_mock = mocker.patch('app.user_api_client.get_user', return_value=user) mocker.patch('app.user_api_client.get_user', return_value=user)
create_event_mock = mocker.patch('app.events_api_client.create_event') mocker.patch('app.events_api_client.create_event')
if service: if mocker and service:
session['service_id'] = service['id'] session['service_id'] = service['id']
get_service_mock = mocker.patch('app.service_api_client.get_service', return_value={'data': service}) mocker.patch('app.service_api_client.get_service', return_value={'data': service})
login_user(user, remember=True) login_user(user, remember=True)
def login_fresh(self): def login_fresh(self):

View File

@@ -6,14 +6,14 @@ from tests import validate_route_permission
def test_should_show_empty_api_keys_page(app_, def test_should_show_empty_api_keys_page(app_,
api_user_active, api_user_pending,
mock_login, mock_login,
mock_get_no_api_keys, mock_get_no_api_keys,
mock_get_service, mock_get_service,
mock_has_permissions): mock_has_permissions):
with app_.test_request_context(): with app_.test_request_context():
with app_.test_client() as client: with app_.test_client() as client:
client.login(api_user_active) client.login(api_user_pending)
service_id = str(uuid.uuid4()) service_id = str(uuid.uuid4())
response = client.get(url_for('main.api_keys', service_id=service_id)) response = client.get(url_for('main.api_keys', service_id=service_id))

View File

@@ -421,8 +421,13 @@ def test_service_dashboard_updates_gets_dashboard_totals(mocker,
app_, app_,
active_user_with_permissions, active_user_with_permissions,
service_one, service_one,
mock_get_user,
mock_get_service_templates,
mock_get_template_statistics,
mock_get_jobs,
mock_get_service_statistics, mock_get_service_statistics,
mock_get_usage): mock_get_usage,
mock_get_detailed_service):
dashboard_totals = mocker.patch('app.main.views.dashboard.get_dashboard_totals', return_value={ dashboard_totals = mocker.patch('app.main.views.dashboard.get_dashboard_totals', return_value={
'email': {'requested': 0, 'delivered': 0, 'failed': 0}, 'email': {'requested': 0, 'delivered': 0, 'failed': 0},
'sms': {'requested': 0, 'delivered': 0, 'failed': 0} 'sms': {'requested': 0, 'delivered': 0, 'failed': 0}

View File

@@ -55,20 +55,30 @@ def fake_uuid():
@pytest.fixture(scope='function') @pytest.fixture(scope='function')
def mock_get_service(mocker, api_user_active): def mock_get_service(mocker, api_user_active):
def _get(service_id, detailed=False): def _get(service_id):
service = service_json( service = service_json(
service_id, "Test Service", [api_user_active.id], message_limit=1000, service_id, "Test Service", [api_user_active.id], message_limit=1000,
active=False, restricted=True) active=False, restricted=True)
if detailed:
service['statistics'] = {
'email': {'requested': 0, 'delivered': 0, 'failed': 0},
'sms': {'requested': 0, 'delivered': 0, 'failed': 0}
}
return {'data': service} return {'data': service}
return mocker.patch('app.service_api_client.get_service', side_effect=_get) return mocker.patch('app.service_api_client.get_service', side_effect=_get)
@pytest.fixture(scope='function')
def mock_get_detailed_service(mocker, api_user_active):
def _get(service_id):
service = service_json(
service_id, "Test Service", [api_user_active.id], message_limit=1000,
active=False, restricted=True)
service['statistics'] = {
'email': {'requested': 0, 'delivered': 0, 'failed': 0},
'sms': {'requested': 0, 'delivered': 0, 'failed': 0}
}
return {'data': service}
return mocker.patch('app.service_api_client.get_detailed_service', side_effect=_get)
@pytest.fixture(scope='function') @pytest.fixture(scope='function')
def mock_get_live_service(mocker, api_user_active): def mock_get_live_service(mocker, api_user_active):
def _get(service_id): def _get(service_id):