1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

// vim: tw=80
//! Examples of Mockall's generated code
use mockall::{mock, automock};

/// Mock of a basic trait with several kinds of method.
///
/// It is mocked by the [`MockFoo`](struct.MockFoo.html) struct.
#[automock]
pub trait Foo {
    /// A method with a `'static` return type
    fn foo(&self, x: i32, y: i16) -> i32;

    /// A method returning a reference
    fn bar(&self, x: i32) -> &i32;

    /// A method returning a mutable reference
    fn baz(&mut self, x: i32) -> &mut i32;

    /// A method returning a `'static` reference
    fn bean(&self) -> &'static i32;

    /// A static method
    fn bang(x: i32) -> i32;
}

/// A trait implemented by a Struct we want to mock
pub trait Bah {
    /// Some trait method
    fn bah(&self);
}

mock! {
    /// Mock of a struct
    ///
    /// Structs can be mocked with [`mock!`].
    /// Their mock methods have an identical API to the methods generated by
    /// [`#[automock]`](automock).
    pub Boo {
        /// A method on a struct
        fn boo(&self);
    }
    /// An implementation of a trait on a mocked struct
    trait Bah {
        fn bah(&self);
    }
}

#[automock(mod mock_ffi;)]
extern "C" {
    /// A foreign "C" function.
    pub fn ffi_func();
}

#[cfg(feature = "nightly")]
/// Mock this entire module
#[automock]
pub mod my_module {
    /// A function in a mocked module
    pub fn modfunc() {
        unimplemented!()
    }
}