third_party/rust_crates/vendor/mockall/tests/automock_generic_future.rs

// vim: tw=80
//! A generic mock object that implements Future
//!
//! This is tricky because the Context object has a lifetime parameter, yet the
//! poll method must not be treated as a generic method.
#![deny(warnings)]

use mockall::*;
use std::{
    future::Future,
    pin::Pin,
    task::{Context, Poll},
};

struct Foo<T: 'static>(T);

#[automock]
impl<T: 'static> Future for Foo<T> {
    type Output = ();

    fn poll<'a>(self: Pin<&mut Self>, _cx: &mut Context<'a>)
        -> Poll<Self::Output>
    {
        unimplemented!()
    }
}

#[test]
fn ready() {
    let mut mock = MockFoo::<u32>::new();
    mock.expect_poll()
        .return_const(Poll::Ready(()));
    let _r = async {
        mock.await
    };
}