Refactor test

This commit is contained in:
Ken Tsang
2017-06-23 15:29:09 +01:00
parent bcc45ede9c
commit beffaeaeea
2 changed files with 5 additions and 6 deletions

View File

@@ -98,7 +98,7 @@ def choose_template(service_id, template_type='all'):
('All', 'all'), ('All', 'all'),
('Text message', 'sms'), ('Text message', 'sms'),
('Email', 'email'), ('Email', 'email'),
('Letter', 'letter') if current_service['can_send_letters'] else None, ('Letter', 'letter') if 'letter' in current_service['permissions'] else None,
]) ])
] ]

View File

@@ -62,7 +62,7 @@ def test_should_show_overview(
@pytest.mark.parametrize('permissions, expected_rows', [ @pytest.mark.parametrize('permissions, expected_rows', [
(['email', 'sms', 'inbound_sms'], [ (['email', 'sms', 'inbound_sms', 'international_sms'], [
'Service name service one Change', 'Service name service one Change',
'Email reply to address test@example.com Change', 'Email reply to address test@example.com Change',
'Text message sender elevenchars', 'Text message sender elevenchars',
@@ -75,7 +75,7 @@ def test_should_show_overview(
'Service name service one Change', 'Service name service one Change',
'Email reply to address test@example.com Change', 'Email reply to address test@example.com Change',
'Text message sender elevenchars Change', 'Text message sender elevenchars Change',
'International text messages On Change', 'International text messages Off Change',
'Receive text messages Off Change', 'Receive text messages Off Change',
'Letters Off Change', 'Letters Off Change',
]), ]),
@@ -91,7 +91,7 @@ def test_should_show_overview_for_service_with_more_things_set(
expected_rows expected_rows
): ):
client.login(active_user_with_permissions, mocker, service_with_reply_to_addresses) client.login(active_user_with_permissions, mocker, service_with_reply_to_addresses)
service_with_reply_to_addresses['permissions'] = ['email', 'sms', 'inbound_sms', 'international_sms'] service_with_reply_to_addresses['permissions'] = permissions
response = client.get(url_for( response = client.get(url_for(
'main.service_settings', service_id=service_with_reply_to_addresses['id'] 'main.service_settings', service_id=service_with_reply_to_addresses['id']
)) ))
@@ -158,8 +158,7 @@ def test_if_can_receive_inbound_then_cant_change_sms_sender(
'main.service_settings', service_id=service_one['id'] 'main.service_settings', service_id=service_one['id']
)) ))
page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser') page = BeautifulSoup(response.data.decode('utf-8'), 'html.parser')
rows = page.find_all('tr') rows_as_text = [" ".join(row.text.split()) for row in page.find_all('tr')]
rows_as_text = [" ".join(row.text.split()) for row in rows]
assert 'Text message sender SomeNumber Change' not in rows_as_text assert 'Text message sender SomeNumber Change' not in rows_as_text
assert url_for('main.service_request_to_go_live', service_id=service_one['id'], assert url_for('main.service_request_to_go_live', service_id=service_one['id'],
set_inbound_sms=False) not in response.get_data(as_text=True) set_inbound_sms=False) not in response.get_data(as_text=True)