如何在单位测试中模拟功能Actix Web

Question

我有一个具有2层的API服务：API和存储库。 API层是请求处理和业务逻辑定义的地方，而存储库是数据库调用功能。我想在我的API层的单元测试上模拟数据库调用功能，有什么办法可以使用Actix Web？

pub fn login(db: web::Data<Pool>, payload: Json<LoginInput>) -> Result<Token, Error> {

    // Function i want to mock:
    let user =
        match get_user_by_email(&db.get().unwrap(), payload.email.to_string()) {
            // This return status code of bad request
            Err(e) => return Error::InvalidUser,
            Ok(user) => user,
        };

    let user_claims = UserClaims {
        id: user.id,
        email: user.email
    };

    match verify(payload.password.to_string(), &user.password) {
        Err(e) => return Error::BadRequest(e),
        Ok(result) => match result {
            false => return Error::InvalidUser,
            _ => {}
        },
    };

    let token = generate_access_and_refresh_token(user_claims);

    token
}

测试：

#[actix_web::test]
async fn test_login() {
    let payload = create_mock_login_payload();
    let req = test::TestRequest::default()
        .set_json(payload)
        .to_http_request();

    let resp = login(req).await.unwrap();
    assert_eq!(resp.status(), StatusCode::BadRequest);
}

我想要的模拟功能行为：

fn get_user_by_email(db: Pool, email: String) -> Result<User, Error> {
   if !email_valid(email) {
       return Err(Error::InvalidEmail);
   }
   Ok(create_mock_user())
}