【每周一库】- Mockall 一个强大的Rust对象模拟库（第四部分——完结）

小编在去云南疯玩了一周后短暂忘记了自己的责任与价值，导致每周一库没有及时更新，非常抱歉！

这次是Mockall (对象模拟库) 系列的最后一篇。

关联类型

包含关联类型的特征也可以模拟。与通用特征不同，模拟出的结构型不是通用的，在定义模拟结构型时必须要指定关联的类型，这可以通过#[automock\]属性中的元项目来达成：

#[automock(type Key=u16; type Value=i32;)]
pub trait A {type Key;type Value;fn foo(&self, k: Self::Key) -> Self::Value;
}let mut mock = MockA::new();
mock.expect_foo().returning(|x: u16| i32::from(x));
assert_eq!(4, mock.foo(4));

多重特征与继承特征

在创建一个可以实现多种特征的模拟结构型时，无论有没有继承，都需要使用mock!宏。但在创建之后，模拟对象的使用方法无异。

pub trait A {fn foo(&self);
}pub trait B: A {fn bar(&self);
}mock! {// 需要模拟的结构型C {}// C 中需要实现的第一个特征trait A {fn foo(&self);}// C中需要实现的第二个特征trait B: A {fn bar(&self);}
}
let mut mock = MockC::new();
mock.expect_foo().returning(|| ());
mock.expect_bar().returning(|| ());
mock.foo();
mock.bar();

外部特征

Mockall还可以模拟在你控制之外的外部包中定义的特征与结构型，但是必须使用mock! 宏，而不是#[automock\]。你可以这样去模拟一个外部特征：

mock! {MyStruct {}     // 模拟结构型的名称, 不需要 "Mock" 前缀trait Clone {   // 需要模拟的特征的定义fn clone(&self) -> Self;}
}let mut mock1 = MockMyStruct::new();
let mock2 = MockMyStruct::new();
mock1.expect_clone().return_once(move || mock2);
let cloned = mock1.clone();

静态方法

Mockall还可以模拟静态的方法，但是需要注意，在模拟静态方法时的”期望“是全局的。如果你需要在多个不同测试中使用一个静态方法，必须要提供专属的同步。普通的方法中，期望是设置在模拟对象之上的。但是静态方法没有对应的模拟对象，所以必须要通过创建一个”语境“对象来设置这些期望：

#[automock]
pub trait A {fn foo() -> u32;
}let ctx = MockA::foo_context();
ctx.expect().returning(|| 99);
assert_eq!(99, MockA::foo());

一个常用的模式是用构造方法来模拟一个特征。在这种情况下，你可以利用构造方法来返回一个模拟对象。

struct Foo{}
#[automock]
impl Foo {fn from_i32(x: i32) -> Self {// ...}fn foo(&self) -> i32 {// ...}
}let ctx = MockFoo::from_i32_context();
ctx.expect().returning(|x| {let mut mock = MockFoo::default();mock.expect_foo().return_const(x);mock});
let foo = MockFoo::from_i32(42);
assert_eq!(42, foo.foo());

通用静态方法

模拟通用的结构型或特征的静态方法时，无论方法本身是不是通用的，都能无缝运行。

#[automock]
trait Foo<T: 'static> {fn new(t: T) -> MockFoo<T>;
}let ctx = MockFoo::<u32>::new_context();
ctx.expect().returning(|_| MockFoo::default());
let mock = MockFoo::<u32>::new(42u32);

语境检查点

当语境对象离开作用域时，所有的期望都会被清空。语境对象与其他模拟对象一样有一个checkpoint 方法。

#[automock]
pub trait A {fn foo() -> u32;
}let ctx = MockA::foo_context();
ctx.expect().times(1).returning(|| 99);
ctx.checkpoint();   // Panics!

模拟对象的检查点方法不会对静态方法进行检查。这种行为对同时使用多个模拟对象会比较有用，例如：

#[automock]
pub trait A {fn build() -> Self;fn bar(&self) -> i32;
}let ctx = MockA::build_context();
ctx.expect().times(2).returning(|| MockA::default());
let mut mock0 = MockA::build();
mock0.expect_bar().return_const(4);
mock0.bar();
mock0.checkpoint();     // 不会对build方法进行checkpoint
let mock1 = MockA::build();

还有一点值得注意的时：Mockall一般情况下会为每一个模拟结构型创建一个没有参数的new方法。但是在模拟一个已经包含名为new的方法的结构型时，Mockall则不会自动创建。

外部方程

Mockall还可以模拟外部方程。和静态方法一样，模拟外部方程时的期望是全局的。与模拟结构型时类似，为了使模拟的方程可被访问，需要对导入进行特别的调试。与关联类型一样，需要在#[automock\]属性中提供一些额外的信息模拟外部方程才可正常使用。

mod ffi {use mockall::automock;#[automock(mod mock;)]extern "C" {pub fn foo(x: u32) -> i64;}
}cfg_if! {if #[cfg(test)] {use self::ffi::mock::foo;} else {use self::ffi::foo;}
}fn do_stuff() -> i64 {unsafe{ foo(42) }
}#[cfg(test)]
mod t {use super::*;#[test]fn test_foo() {let ctx = ffi::mock::foo_context();ctx.expect().returning(|x| i64::from(x + 1));assert_eq!(43, do_stuff());}
}

模块

除了可以模拟外部方程之外，Mockall还可以为所有Rust方程的模块派生出模拟，这个功能需要使用每日构建功能，在你的包中需要启用feature(proc_macro_hygiene)。使用方法与模拟外部功能一样，但是模拟模块的命名是自动生成的。

#![feature(proc_macro_hygiene)]
mod outer {use mockall::automock;#[automock()]pub(super) mod inner {pub fn bar(x: u32) -> i64 {// ...}}
}cfg_if! {if #[cfg(test)] {use outer::mock_inner as inner;} else {use outer::inner;}
}#[cfg(test)]
mod t {use super::*;#[test]fn test_foo_bar() {let ctx = inner::bar_context();ctx.expect().returning(|x| i64::from(x + 1));assert_eq!(5, inner::bar(4));}
}