#[async_mock::async_mock]
#[async_trait::async_trait]
trait SomeAsyncTrait {
async fn foo(&self, x: i32) -> i32;
fn bar(&self, x: i32) -> i32;
}
#[derive(Default)]
struct SomeStruct;
impl SomeStruct {
async fn foo_di(&self, t: &impl SomeAsyncTrait, x: i32) -> i32 {
t.foo(x).await
}
async fn bar_di(&self, t: &impl SomeAsyncTrait, x: i32) -> i32 {
t.bar(x)
}
}
#[tokio::test]
async fn async_ok() {
let mut mock = MockSomeAsyncTrait::default();
mock.expect_foo().once().returning(|x| x + 1);
let sut = SomeStruct::default();
let result = sut.foo_di(&mock, 3).await;
assert_eq!(result, 4);
}
#[tokio::test]
async fn non_async_ok() {
let mut mock = MockSomeAsyncTrait::default();
mock.expect_bar().once().returning(|x| x + 1);
let sut = SomeStruct::default();
let result = sut.bar_di(&mock, 3).await;
assert_eq!(result, 4);
}