用 Flask @auth.login_required 装饰器装饰的端点的单元测试

编程问答 2022-05-27

问题描述

我使用 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 (将#修改为@）

decoratordecoratorflaskmockingmockingpythonunit-testing