def test_create_user():
#setup
parameters = {'email': '*****@*****.**', 'display_name': 'test_name'}
expected = User(**parameters)
mail_params = {
'subject': 'Welcome the WDL Fantasy League!',
'sender': app.config.get("MAIL_USERNAME"),
'recipients': [parameters['email']],
'html': '<h1>Welcome to the WDL Fantasy League, {}!</h1>'.format(parameters['display_name'])\
+ '<p>Your id is {}</p>'.format(expected._id)\
+ '<p>and your link is http://whodunnitleague.com/fantasy/team.html?id={}'.format(expected._id)
}
mocked_message = Mock()
mocked_message_class = Mock(return_value=mocked_message)
mocked_mail = Mock()
mocked_mail.send = Mock()
module.mail = mocked_mail
module.Message = mocked_message_class
with app.app_context():
current_app.user_service = Mock()
current_app.user_service.create_user = Mock()
#functionality
r = client.post('/users/', data=json.dumps(parameters))
#test
with app.app_context():
current_app.user_service.create_user\
.assert_called_with(expected)
mocked_message_class.assert_called_with(**mail_params)
mocked_mail.send.assert_called_with(mocked_message)
assert r.get_data() == json.dumps(dict(expected))
assert r.status_code == 201
def test_check_user_by_display_name():
#setup
incomplete_parameters = {'display_name': 'test_name'}
expected_check = User(**incomplete_parameters)
exists = True
with app.app_context():
current_app.user_service = Mock()
current_app.user_service.check_user = Mock(return_value=exists)
#functionality
r = client.get('/users/check', query_string=incomplete_parameters)
#test
with app.app_context():
current_app.user_service.check_user\
.assert_called_with(expected_check)
assert r.get_data() == json.dumps({'exists': exists})
assert r.status_code == 200
def test_get_user_failure():
#setup
id = 'test_id'
parameters = None
expected = User()
expected_check = User(_id=id)
with app.app_context():
current_app.user_service = Mock()
current_app.user_service.get_user = Mock(return_value=parameters)
#functionality
r = client.get('/users', query_string={'id': id})
#test
with app.app_context():
current_app.user_service.get_user\
.assert_called_with(expected_check)
assert r.get_data() == json.dumps(dict(expected))
assert r.status_code == 200
def test_get_user_success():
#setup
id = 'test_id'
parameters = {'email': '*****@*****.**', 'display_name': 'tet_name'}
expected = User(**parameters)
expected_check = User(_id=id)
with app.app_context():
current_app.user_service = Mock()
current_app.user_service.get_user = Mock(return_value=parameters)
#functionality
r = client.get('/users', query_string={'id': id})
#test
with app.app_context():
current_app.user_service.get_user\
.assert_called_with(expected_check)
assert r.get_data() == json.dumps(dict(expected))
assert r.status_code == 200