tests/ui/consts/fn_trait_refs.rs

   1 // check-pass
   2
   3 #![feature(const_fn_trait_ref_impls)]
   4 #![feature(fn_traits)]
   5 #![feature(unboxed_closures)]
   6 #![feature(const_trait_impl)]
   7 #![feature(const_mut_refs)]
   8 #![feature(const_cmp)]
   9 #![feature(const_refs_to_cell)]
  10
  11 use std::marker::Destruct;
  12
  13 const fn tester_fn<T>(f: T) -> T::Output
  14 where
  15     T: ~const Fn<()> + ~const Destruct,
  16 {
  17     f()
  18 }
  19
  20 const fn tester_fn_mut<T>(mut f: T) -> T::Output
  21 where
  22     T: ~const FnMut<()> + ~const Destruct,
  23 {
  24     f()
  25 }
  26
  27 const fn tester_fn_once<T>(f: T) -> T::Output
  28 where
  29     T: ~const FnOnce<()>,
  30 {
  31     f()
  32 }
  33
  34 const fn test_fn<T>(mut f: T) -> (T::Output, T::Output, T::Output)
  35 where
  36     T: ~const Fn<()> + ~const Destruct,
  37 {
  38     (
  39         // impl<A: Tuple, F: ~const Fn + ?Sized> const Fn<A> for &F
  40         tester_fn(&f),
  41         // impl<A: Tuple, F: ~const Fn + ?Sized> const FnMut<A> for &F
  42         tester_fn_mut(&f),
  43         // impl<A: Tuple, F: ~const Fn + ?Sized> const FnOnce<A> for &F
  44         tester_fn_once(&f),
  45     )
  46 }
  47
  48 const fn test_fn_mut<T>(mut f: T) -> (T::Output, T::Output)
  49 where
  50     T: ~const FnMut<()> + ~const Destruct,
  51 {
  52     (
  53         // impl<A: Tuple, F: ~const FnMut + ?Sized> const FnMut<A> for &mut F
  54         tester_fn_mut(&mut f),
  55         // impl<A: Tuple, F: ~const FnMut + ?Sized> const FnOnce<A> for &mut F
  56         tester_fn_once(&mut f),
  57     )
  58 }
  59 const fn test(i: i32) -> i32 {
  60     i + 1
  61 }
  62
  63 fn main() {
  64     const fn one() -> i32 {
  65         1
  66     };
  67     const fn two() -> i32 {
  68         2
  69     };
  70     const _: () = {
  71         let test_one = test_fn(one);
  72         assert!(test_one == (1, 1, 1));
  73
  74         let test_two = test_fn_mut(two);
  75         assert!(test_two == (2, 2));
  76     };
  77 }