Refactor test to use client fixture

This commit is contained in:
Imdad Ahad
2017-01-24 11:01:44 +00:00
parent 343bf9ab98
commit 4d124a5711

View File

@@ -324,7 +324,7 @@ def test_shows_archived_label_instead_of_live_or_research_mode_label(
def test_should_show_correct_sent_totals_for_platform_admin( def test_should_show_correct_sent_totals_for_platform_admin(
app_, client,
platform_admin_user, platform_admin_user,
mocker, mocker,
mock_get_detailed_services, mock_get_detailed_services,
@@ -341,11 +341,9 @@ def test_should_show_correct_sent_totals_for_platform_admin(
) )
mock_get_detailed_services.return_value = {'data': services} mock_get_detailed_services.return_value = {'data': services}
with app_.test_request_context(): mock_get_user(mocker, user=platform_admin_user)
with app_.test_client() as client: client.login(platform_admin_user)
mock_get_user(mocker, user=platform_admin_user) response = client.get(url_for('main.platform_admin'))
client.login(platform_admin_user)
response = client.get(url_for('main.platform_admin'))
assert response.status_code == 200 assert response.status_code == 200
page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser') page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser')