Added choose_service using services_dao.

This commit is contained in:
Nicholas Staples
2016-01-18 11:15:14 +00:00
parent 4e2019c949
commit a09ac6cbf5
5 changed files with 74 additions and 15 deletions

View File

@@ -1,15 +1,23 @@
from tests import create_test_user
from flask import url_for
from app.models import User
def test_should_show_choose_services_page(app_,
db_,
db_session,
active_user):
active_user,
mock_get_services):
with app_.test_request_context():
with app_.test_client() as client:
client.login(active_user)
user = User.query.first()
client.login(user)
response = client.get(url_for('main.choose_service'))
assert response.status_code == 200
assert 'Choose service' in response.get_data(as_text=True)
resp_data = response.get_data(as_text=True)
assert 'Choose service' in resp_data
services = mock_get_services.side_effect()
assert mock_get_services.called
assert services['data'][0]['name'] in resp_data
assert services['data'][1]['name'] in resp_data