#![feature(async_closure)]
#[cfg(test)]
use async_mock::async_mock;
use async_trait::async_trait;

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

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

#[tokio::test]
async 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).await);
}