tests/ui/consts/const_discriminant.rs

   1 // run-pass
   2 #![feature(const_discriminant)]
   3 #![allow(dead_code)]
   4
   5 use std::mem::{discriminant, Discriminant};
   6 use std::hint::black_box;
   7
   8 enum Test {
   9     A(u8),
  10     B,
  11     C { a: u8, b: u8 },
  12 }
  13
  14 const TEST_A: Discriminant<Test> = discriminant(&Test::A(5));
  15 const TEST_A_OTHER: Discriminant<Test> = discriminant(&Test::A(17));
  16 const TEST_B: Discriminant<Test> = discriminant(&Test::B);
  17
  18 enum Void {}
  19
  20 enum SingleVariant {
  21     V,
  22     Never(Void),
  23 }
  24
  25 const TEST_V: Discriminant<SingleVariant> = discriminant(&SingleVariant::V);
  26
  27 pub const TEST_VOID: () = {
  28     // This is UB, but CTFE does not check validity so it does not detect this.
  29     // This is a regression test for https://github.com/rust-lang/rust/issues/89765.
  30     unsafe { std::mem::discriminant(&*(&() as *const () as *const Void)); };
  31 };
  32
  33
  34 fn main() {
  35     assert_eq!(TEST_A, TEST_A_OTHER);
  36     assert_eq!(TEST_A, discriminant(black_box(&Test::A(17))));
  37     assert_eq!(TEST_B, discriminant(black_box(&Test::B)));
  38     assert_ne!(TEST_A, TEST_B);
  39     assert_ne!(TEST_B, discriminant(black_box(&Test::C { a: 42, b: 7 })));
  40
  41     assert_eq!(TEST_V, discriminant(black_box(&SingleVariant::V)));
  42 }