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

#[tokio::test]
#[should_panic(expected = "Missing returning function for `foo`")]
async fn missing_expectation() {
    let mock = MockSomeAsyncTrait::default();
    _ = mock.foo(3).await;
}

#[tokio::test]
#[should_panic(expected = "Failed expectation for `foo`. Called 1 times but expecting 2.")]
async fn call_count_short() {
    let mut mock = MockSomeAsyncTrait::default();
    mock.expect_foo().times(2).returning(|x| x + 1);
    _ = mock.foo(3).await;
}

#[tokio::test]
#[should_panic(expected = "Failed expectation for `foo`. Called 3 times but expecting 1.")]
async fn call_count_high() {
    let mut mock = MockSomeAsyncTrait::default();
    mock.expect_foo().once().returning(|x| x + 1);
    _ = mock.foo(3).await;
    _ = mock.foo(3).await;
    _ = mock.foo(3).await;
}

#[tokio::test]
#[should_panic(expected = "Missing returning function for `foo`")]
async fn no_returning_set() {
    let mut mock = MockSomeAsyncTrait::default();
    mock.expect_foo().once();
    _ = mock.foo(3).await;
}