Python 用Flask@auth.login\u所需装饰器装饰端点的单元测试

我正在使用Flask HTTPAuth进行身份验证，Flask login进行登录，并希望为使用

@auth.login\u required（scheme=“Bearer”）

修饰的端点添加单元测试：

端点.py

class UserData(Resource):
    @auth.login_required
    def get(self, user_id):
        user = db_models.User.query.get(user_id)
        return result

@auth.verify_token
def verify_token(token):
    data = token_serializer.loads(token)
    return data.get("username", False)

@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)

@mock.patch("myapp.auth.login_required", return_value = True)
@mock.patch("myapp.db_models.User")
def test_get_user(self, mock_user, 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)

authent.py

class UserData(Resource):
    @auth.login_required
    def get(self, user_id):
        user = db_models.User.query.get(user_id)
        return result

@auth.verify_token
def verify_token(token):
    data = token_serializer.loads(token)
    return data.get("username", False)

@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)

@mock.patch("myapp.auth.login_required", return_value = True)
@mock.patch("myapp.db_models.User")
def test_get_user(self, mock_user, 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)

但是被嘲笑所困扰（这不起作用）：

test.py

class UserData(Resource):
    @auth.login_required
    def get(self, user_id):
        user = db_models.User.query.get(user_id)
        return result

@auth.verify_token
def verify_token(token):
    data = token_serializer.loads(token)
    return data.get("username", False)

@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)

@mock.patch("myapp.auth.login_required", return_value = True)
@mock.patch("myapp.db_models.User")
def test_get_user(self, mock_user, 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)

第二种方法是：

test.py

class UserData(Resource):
    @auth.login_required
    def get(self, user_id):
        user = db_models.User.query.get(user_id)
        return result

@auth.verify_token
def verify_token(token):
    data = token_serializer.loads(token)
    return data.get("username", False)

@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)

@mock.patch("myapp.auth.login_required", return_value = True)
@mock.patch("myapp.db_models.User")
def test_get_user(self, mock_user, 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)

你能帮我找出正确的方法吗

---

谢谢你的帮助

这两种方法都是完全有效的，因为您不测试

auth

库，无论您的模拟方法有多精细。注意细节；）