use async_mock::async_mock;

#[async_mock]
trait SomeTrait {
    fn foo(&self, x: i32) -> i32;
}

#[async_mock]
trait OtherTrait {
    fn foo(&self, some: &impl SomeTrait, x: i32) -> i32;
    fn bar(&self, some: &(impl SomeTrait + Send + Sync), x: i32) -> i32;
}

#[test]
fn test() {
    let mut mock_some_trait = MockSomeTrait::default();
    mock_some_trait.expect_foo().once().returning(|x| x + 1);

    let mut mock_other_trait = MockOtherTrait::default();
    mock_other_trait
        .expect_bar()
        .once()
        .returning_dyn(Box::new(|some, x| some.foo(x * 2)));

    assert_eq!(7, mock_other_trait.bar(&mock_some_trait, 3));
}