def test_bad_values(self):
"""
When giving anything but a basestring, get the appopriate value_error
"""
self.datautils.create_session()
for i in [x for x in bad_data_typevals_list if not isinstance(x, basestring)]:
with self.assertRaises(ValueError) as e:
get_user_from_token(i, 'valid_token_type')
self.assertEqual(e.exception.args[0], 'dbsession should be a valid db session')
with self.assertRaises(ValueError) as e:
get_user_from_token(self.session, i)
self.assertEqual(e.exception.args[0], 'token must be a valid basestring')
def test_nonexistent_token(self):
"""
If the token doesn't exist, we should get None
"""
self.datautils.create_session()
result = get_user_from_token(self.session, 'invalid_token')
self.assertEqual(result, None)
def test_get_user(self):
"""
When the token is valid, get the right user
"""
user = self.datautils.create_user()
sess = self.datautils.create_session({'user_id': user.id})
result = get_user_from_token(self.session, sess.token)
self.assertEqual(dict_from_row(user), dict_from_row(result))