def test_login_required(self): h = HTTPAuth() h.get_verify_token_callback = verify_token self.assertEqual(h.login_required(lambda x: x)(1).status_code, 401) token = generate_user_token(self.u) with app.test_request_context('/', headers={'authorization': 'Bearer %s' % (token)}) as c: self.assertEqual(h.login_required(lambda x: x)(1),1)
def test_get_token(self): h = HTTPAuth() token = "test" with app.test_request_context('/', headers={'authorization': 'Bearer %s' % (token)}) as c: self.assertEqual(token, h.get_token())