寻求Rust Stable兼容的Mocktopus替代方案（函数Mock需求）

问题背景 #

我们的代码大量用Mocktopus做函数Mock，但切换到Rust Stable后没法用了。试过mockall，但因为Limiter这类结构体是其他宏生成的，没法用mock!宏包裹——比如在基于Substrate的Interlay项目里，之前靠Mocktopus Mock get_minimum_collateral_vault函数来测试register_vault方法，现在急需替代方案。

方案1：手动依赖注入（最稳定的原生方案） #

这是Rust生态里最通用的Mock思路，完全不依赖第三方库，适合长期维护。

• 核心逻辑：把要Mock的函数抽象成trait，通过给结构体传不同的trait实现，切换生产和测试逻辑。

针对helper模块函数的场景 #

1. 为helper里的函数定义 trait：

   pub trait HelperOps {
       fn min(a: u32, b: u32) -> u32;
   }
   

2. 写生产环境的默认实现：

   pub struct ProductionHelper;
   impl HelperOps for ProductionHelper {
       fn min(a: u32, b: u32) -> u32 {
           std::cmp::min(a, b)
       }
   }
   

3. 修改Limiter结构体，让它持有HelperOps的实例（或者通过方法参数传入）：

   pub struct Limiter<H: HelperOps> {
       helper: H,
       // 其他原有字段
   }
   
   impl<H: HelperOps> Limiter<H> {
       pub fn new(helper: H) -> Self {
           Self { helper }
       }
   
       pub fn calculate_limit(&self, a: u32, b: u32) -> u32 {
           self.helper.min(a, b)
       }
   }
   

4. 测试时写Mock实现：

   pub struct MockHelper;
   impl HelperOps for MockHelper {
       fn min(a: u32, b: u32) -> u32 {
           // 自定义Mock逻辑，比如固定返回值、计数调用次数
           100
       }
   }
   
   #[test]
   fn test_limiter_calculate() {
       let limiter = Limiter::new(MockHelper);
       assert_eq!(limiter.calculate_limit(50, 200), 100);
   }
   

针对Substrate Interlay项目的场景 #

把get_minimum_collateral_vault所在的逻辑抽象成trait，让register_vault依赖这个trait：

pub trait VaultCollateralOps {
    fn get_minimum_collateral_vault(&self) -> Balance;
}

// 生产环境用Runtime实现这个trait
impl VaultCollateralOps for Runtime {
    fn get_minimum_collateral_vault(&self) -> Balance {
        // 原有的业务逻辑
    }
}

// 测试用的Mock实现
pub struct MockVaultCollateral;
impl VaultCollateralOps for MockVaultCollateral {
    fn get_minimum_collateral_vault(&self) -> Balance {
        // 测试需要的固定值，比如1000
        1000
    }
}

// 修改register_vault方法，接收实现了VaultCollateralOps的参数
pub fn register_vault<T: VaultCollateralOps>(ops: &T) -> Result<(), Error> {
    let min_collat = ops.get_minimum_collateral_vault();
    // 原有的register逻辑
    Ok(())
}

测试时直接传MockVaultCollateral实例就行，不用改原有生产代码的核心逻辑。

方案2：条件编译+全局变量（快速适配原有代码） #

如果不想大改结构，可以用条件编译替换测试环境的函数实现，适合临时过渡或者简单函数的Mock。

• 核心逻辑：用#[cfg(test)]标记测试环境的函数实现，通过全局变量存Mock逻辑。

比如针对helper::min：

// 生产环境的正常实现
#[cfg(not(test))]
pub fn min(a: u32, b: u32) -> u32 {
    std::cmp::min(a, b)
}

// 测试环境用全局变量存Mock函数
#[cfg(test)]
pub static mut MIN_MOCK: Option<fn(u32, u32) -> u32> = None;

#[cfg(test)]
pub fn min(a: u32, b: u32) -> u32 {
    MIN_MOCK.unwrap()(a, b)
}

// 测试时设置Mock
#[test]
fn test_min_mock() {
    unsafe {
        MIN_MOCK = Some(|_, _| 100);
    }
    assert_eq!(helper::min(50, 200), 100);
    // 测试后重置，避免影响其他测试用例
    unsafe {
        MIN_MOCK = None;
    }
}

⚠️ 注意：这里用了unsafe，一定要在每个测试后重置Mock状态，不然会导致测试间的状态污染。

方案3：mockall的trait层面Mock（适配宏生成的结构体） #

如果还是想用mockall，不用直接包裹宏生成的结构体，而是给结构体的核心行为定义trait，用mockall自动生成Mock实现。

1. 先给Limiter的核心功能定义trait，并加上#[automock]：

   use mockall::automock;
   
   #[automock]
   pub trait LimiterOps {
       fn calculate_limit(&self, a: u32, b: u32) -> u32;
   }
   

2. 让宏生成的Limiter结构体实现这个trait：

   impl LimiterOps for Limiter {
       fn calculate_limit(&self, a: u32, b: u32) -> u32 {
           // 调用原逻辑，依赖helper::min
           helper::min(a, b)
       }
   }
   

3. 测试时用MockLimiterOps，同时结合方案2的条件编译Mock helper函数，或者直接在Mock trait里重写逻辑。

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