def test_userid_from_api_token_returns_userid_for_valid_tokens():
token = models.Token('acct:[email protected]')
db.Session.add(token)
result = tokens.userid_from_api_token(token.value)
assert result == 'acct:[email protected]'
def test_userid_from_api_token_returns_None_when_token_doesnt_start_with_prefix():
"""
As a sanity check, don't even attempt to look up tokens that don't start
with the expected prefix.
"""
token = models.Token('acct:[email protected]')
token.value = u'abc123'
db.Session.add(token)
result = tokens.userid_from_api_token(u'abc123')
assert result is None
def token(self, db_session):
token = models.Token(userid='acct:[email protected]')
db_session.add(token)
db_session.flush()
return token