rust/tests/ui/generator/issue-57017.rs

// revisions: no_drop_tracking drop_tracking drop_tracking_mir
// [drop_tracking] compile-flags: -Zdrop-tracking
// [drop_tracking_mir] compile-flags: -Zdrop-tracking-mir
// [drop_tracking] build-pass
// [drop_tracking_mir] build-pass

#![feature(generators, negative_impls)]

macro_rules! type_combinations {
    (
        $( $name:ident => { $( $tt:tt )* } );*
    ) => { $(
        mod $name {
            pub mod unsync {
                $( $tt )*

                impl !Sync for Client {}
            }
            pub mod unsend {
                $( $tt )*

                impl !Send for Client {}
            }
        }

        // This is the same bug as issue 57017, but using yield instead of await
        {
            let g = move || match drop(&$name::unsync::Client::default()) {
                _status => yield,
            };
            assert_send(g);
            //[no_drop_tracking]~^ ERROR generator cannot be sent between threads safely
            //[no_drop_tracking]~| ERROR generator cannot be sent between threads safely
            //[no_drop_tracking]~| ERROR generator cannot be sent between threads safely
        }

        // This tests that `Client` is properly considered to be dropped after moving it into the
        // function.
        {
            let g = move || match drop($name::unsend::Client::default()) {
                _status => yield,
            };
            assert_send(g);
            //[no_drop_tracking]~^ ERROR generator cannot be sent between threads safely
            //[no_drop_tracking]~| ERROR generator cannot be sent between threads safely
            //[no_drop_tracking]~| ERROR generator cannot be sent between threads safely
        }
    )* }
}

fn assert_send<T: Send>(_thing: T) {}

fn main() {
    type_combinations!(
        copy => { #[derive(Copy, Clone, Default)] pub struct Client; };
        derived_drop => { #[derive(Default)] pub struct Client { pub nickname: String } };
        significant_drop => {
            #[derive(Default)]
            pub struct Client;
            impl Drop for Client {
                fn drop(&mut self) {}
            }
        }
    );
}