如何为Rust trait提供Mock实现（不使用mockall工具）

方案1：条件编译切换实现 #

通过#[cfg]属性控制不同环境下的Trait实现，直接避免冲突。修改services/discord.rs中的WebService实现：

// services/discord.rs
// 生产环境用实际HTTP请求实现
#[cfg(not(test))]
impl WebService for DiscordService {
    fn send_request(req: RequestBuilder) -> Option<Value> {
        // 实际HTTP请求逻辑
    }
}

// 测试环境用Mock实现
#[cfg(test)]
impl WebService for DiscordService {
    fn send_request(req: RequestBuilder) -> Option<Value> {
        // Mock返回预设值，比如从文件读取或直接构造
        Some(json!({"ok": true}))
    }
}

这种方式改动最小，无需调整原有业务逻辑，编译时会自动根据环境选择对应实现。

方案2：依赖注入解耦（推荐） #

通过泛型将WebService作为依赖注入到DiscordService中，彻底解耦业务逻辑与传输层，测试时可自由替换实现。

1. 调整DiscordService结构，引入泛型约束：

// services/discord.rs
pub struct DiscordService<W: WebService> {
    client: Client,
    web_service: W,
}

// 实现MessagingService时，调用注入的WebService
impl<W: WebService> MessagingService for DiscordService<W> {
    fn new(web_service: W) -> Self {
        DiscordService {
            client: W::create_client(),
            web_service,
        }
    }

    fn update_task_status(&self, task: &mut Task) {
        let body = json!({"what":"ever"});
        let _ = self.web_service.send_request(
            self.client
                .post(format!(
                    "{BASE_URL}/channels/{}/messages",
                    CONF.discord_channel
                ))
                .json(&body),
        );
    }
}

2. 测试时创建Mock的WebService实现并传入：

#[cfg(test)]
mod tests {
    use super::*;
    use crate::traits::WebService;
    use reqwest::blocking::{Client, RequestBuilder};
    use serde_json::Value;

    // 自定义Mock传输层
    struct MockDiscordWebService;

    impl WebService for MockDiscordWebService {
        fn create_client() -> Client {
            // 返回测试用Client，比如reqwest的模拟客户端
            Client::new()
        }

        fn send_request(_req: RequestBuilder) -> Option<Value> {
            // Mock返回预设响应
            Some(json!({"ok": true}))
        }
    }

    #[test]
    fn test_update_task_status() {
        let notifier = DiscordService::new(MockDiscordWebService);
        let mut test_task = Task { /* 填充测试数据 */ };
        
        notifier.update_task_status(&mut test_task);
        
        // 添加断言，验证任务状态是否正确更新
    }
}

这种方式让代码更符合解耦原则，不仅方便测试，后续替换传输层实现也更灵活。

方案3：使用Mock库简化实现 #

针对复杂测试场景，用mockall这类库自动生成Trait的Mock实现，减少手动编写Mock代码的工作量。

1. 在Cargo.toml中添加开发依赖：

[dev-dependencies]
mockall = "0.11"

2. 给WebService trait添加自动Mock属性：

// traits.rs
use mockall::automock;

#[automock]
pub trait WebService {
    fn create_client() -> Client;
    fn send_request(req: RequestBuilder) -> Option<Value>;
}

3. 测试时直接使用生成的MockWebService：

#[cfg(test)]
mod tests {
    use super::*;
    use crate::traits::{MessagingService, MockWebService};
    use serde_json::Value;

    #[test]
    fn test_update_task_status() {
        // 配置Mock行为：调用send_request时返回预设值
        let mut mock_web = MockWebService::new();
        mock_web.expect_send_request()
            .returning(|_| Some(json!({"ok": true})));

        let notifier = DiscordService::new(mock_web);
        let mut test_task = Task { /* 测试数据 */ };
        
        notifier.update_task_status(&mut test_task);
        
        // 添加断言逻辑
    }
}

这种方案适合需要灵活配置Mock行为的场景，比如验证方法调用次数、参数是否正确等。

内容的提问来源于stack exchange，提问作者zar3bski