如何解决用 Flask @auth.login_required 装饰器装饰的端点的单元测试
我使用 Flask-HTTPAuth 进行身份验证,使用 Flask-login 进行登录,并希望为使用 @auth.login_required(scheme="Bearer")
装饰的端点添加单元测试:
endpoints.py
class UserData(Resource):
@auth.login_required
def get(self,user_id):
user = db_models.User.query.get(user_id)
return result
authentic.py
@auth.verify_token
def verify_token(token):
data = token_serializer.loads(token)
return data.get("username",False)
但是被嘲笑卡住了(它不起作用):
test.py
@mock.patch("myapp.auth")
@mock.patch("myapp.db_models.User")
def test_get_user(self,mock_user,mock_login):
with app.test_client() as client:
mock_login.verify_token.return_value = TEST_USER["username"]
mock_login.login_required.return_value = True
mock_user.query.get.return_value = TEST_USER
response = client.get("/user/100000")
self.assertIsNotNone(response)
self.assertisinstance(response.get_json(),dict)
第二种方法:
test.py
@mock.patch("myapp.auth.login_required",return_value = True)
@mock.patch("myapp.db_models.User")
def test_get_user(self,mock_login):
with app.test_client() as client:
mock_user.query.get.return_value = TEST_USER
response = client.get("/user/100000")
self.assertIsNotNone(response)
self.assertisinstance(response.get_json(),dict)
你能帮我找出正确的方法吗?
感谢您的帮助!
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。