Rust中Mock结构体实现的Trait方法：规避动态分派性能损耗

我有一个函数important_func，接收TestStruct作为参数，该结构体实现了TestTrait，其中包含耗时的IO方法some_long_running_io_method。测试时想Mock这个方法返回预设结果，但直接用mockall生成的MockTestStruct传入函数时出现类型不匹配错误。

原代码如下：

struct TestStruct {
    a: u32,
    b: u32,
}
pub trait TestTrait {
    fn some_long_running_io_method(&self) -> u32 {
        156
    }
}
fn important_func(a: TestStruct) {
    println!("a: {}", a.some_long_running_io_method());
}
impl TestTrait for TestStruct {
    fn some_long_running_io_method(&self) -> u32 {
        self.a + self.b
    }
}
#[cfg(test)]
mod tests {
    use super::*;
    use mockall::predicate::*;
    use mockall::*;
    #[cfg(test)]
    mock! {
        pub TestStruct {}
        impl TestTrait for TestStruct {
            fn some_long_running_io_method(&self) -> u32;
        }
    }
    #[test]
    fn test_important_func() {
        let mut mock = MockTestStruct::new();
        mock.expect_some_long_running_io_method()
            .returning(|| 1);
        important_func(mock);
    }
}

运行后报错：

error[E0308]: mismatched types
  --> src/test.rs:35:24
   |
35 |         important_func(mock);
   |         -------------- ^^^^ expected struct `TestStruct`, found struct `MockTestStruct`
   |         |
   |         arguments to this function are incorrect

核心思路是将目标函数改为泛型，通过Trait约束限定参数类型，既保留生产环境的静态分派性能，又支持测试时传入Mock结构体。

修改后的完整代码如下：

生产代码 #

struct TestStruct {
    a: u32,
    b: u32,
}

pub trait TestTrait {
    fn some_long_running_io_method(&self) -> u32 {
        156
    }
}

// 修改为泛型函数，约束参数类型实现TestTrait
fn important_func<T: TestTrait>(a: T) {
    println!("a: {}", a.some_long_running_io_method());
}

impl TestTrait for TestStruct {
    fn some_long_running_io_method(&self) -> u32 {
        self.a + self.b
    }
}

测试代码 #

#[cfg(test)]
mod tests {
    use super::*;
    use mockall::predicate::*;
    use mockall::*;

    // 命名Mock结构体为TestMock，避免与原结构体冲突
    mock! {
        pub TestMock {}
        impl TestTrait for TestMock {
            fn some_long_running_io_method(&self) -> u32;
        }
    }

    #[test]
    fn test_important_func() {
        let mut mock = MockTestMock::new();
        mock.expect_some_long_running_io_method()
            .returning(|| 1);
        important_func(mock);
    }
}

关键说明 #

• 泛型函数important_func会在编译时为TestStruct和MockTestMock分别生成特化版本，完全是静态分派，无动态分派的性能损耗。
• Mock结构体使用独立命名，避免与原生产结构体产生命名冲突，同时清晰区分测试与生产类型。
• 这种方式既满足生产环境的性能要求，又能在测试中轻松Mock依赖的IO方法，完全适配你的需求。

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