def test_user_status_fails_when_token_blacklisted(self):
with self.client:
# blacklist a valid token
blacklist_token = BlacklistToken(
token=json.loads(self.resp_register.data.decode())['token'])
blacklist_token.__repr__()
db.session.add(blacklist_token)
db.session.commit()
response = self.client.get(
'/auth/user',
headers=dict(
Authorization='Bearer ' + json.loads(
self.resp_register.data.decode()
)['token']
)
)
data = json.loads(response.data.decode())
self.assertTrue(data['status'] == 'fail')
self.assertTrue(data['message'] == 'Token blacklisted. Please log in again.')
self.assertEqual(response.status_code, 401)
class TestBlacklistTokenModel(BaseTestCase):
token_str = 'token'
blt = None
def setUp(self):
BaseTestCase.setUp(self)
self.blt = BlacklistToken(self.token_str)
db.session.add(self.blt)
db.session.commit()
def test_repr(self):
outcome = self.blt.__repr__()
assert (outcome == '<id: token: token')
def test_check_blacklist_found(self):
outcome = BlacklistToken.check_blacklist(self.token_str)
assert (outcome)
def test_check_blacklist_not_found(self):
non_blt_token_str = 'non_blt_token_str'
outcome = BlacklistToken.check_blacklist(non_blt_token_str)
assert (outcome == False)
