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)