use enum_extract_macro::EnumExtract; pub mod name_collisions { #![allow(dead_code, missing_copy_implementations, missing_docs)] pub struct Option; pub struct Some; pub struct None; pub struct Result; pub struct Ok; pub struct Err; } #[allow(unused_imports)] use name_collisions::*; #[derive(Debug, PartialEq, Clone, Copy, EnumExtract)] enum ManyVariants { One(u32), Two(u32, i32), Three(bool, u32, i64), } #[test] fn test_one_unnamed() { let mut many = ManyVariants::One(1); assert!(many.is_one()); assert!(!many.is_two()); assert!(!many.is_three()); assert!(many.as_one().is_ok()); assert!(many.as_two().is_err()); assert!(many.as_three().is_err()); assert!(many.as_one_mut().is_ok()); assert!(many.as_two_mut().is_err()); assert!(many.as_three_mut().is_err()); assert_eq!(*many.as_one().unwrap(), 1_u32); assert_eq!(*many.as_one_mut().unwrap(), 1_u32); assert_eq!(many.into_one().unwrap(), 1_u32); } #[test] fn test_two_unnamed() { let mut many = ManyVariants::Two(1, 2); assert!(!many.is_one()); assert!(many.is_two()); assert!(!many.is_three()); assert!(many.as_one().is_err()); assert!(many.as_two().is_ok()); assert!(many.as_three().is_err()); assert!(many.as_one_mut().is_err()); assert!(many.as_two_mut().is_ok()); assert!(many.as_three_mut().is_err()); assert_eq!(many.as_two().unwrap(), (&1_u32, &2_i32)); assert_eq!(many.as_two_mut().unwrap(), (&mut 1_u32, &mut 2_i32)); assert_eq!(many.into_two().unwrap(), (1_u32, 2_i32)); } #[test] fn test_three_unnamed() { let mut many = ManyVariants::Three(true, 1, 2); assert!(!many.is_one()); assert!(!many.is_two()); assert!(many.is_three()); assert!(many.as_one().is_err()); assert!(many.as_two().is_err()); assert!(many.as_three().is_ok()); assert!(many.as_one_mut().is_err()); assert!(many.as_two_mut().is_err()); assert!(many.as_three_mut().is_ok()); assert_eq!(many.as_three().unwrap(), (&true, &1_u32, &2_i64)); assert_eq!( many.as_three_mut().unwrap(), (&mut true, &mut 1_u32, &mut 2_i64) ); assert_eq!(many.into_three().unwrap(), (true, 1_u32, 2_i64)); } #[test] fn error_should_contain_expected_and_actual() { let many = ManyVariants::Three(true, 1, 2); let error = many.into_one().unwrap_err(); assert_eq!(error.source.expected, "One"); assert_eq!(error.source.actual, "Three"); assert_eq!(error.value, many); assert_eq!(error.to_string(), "expected One, got Three"); } #[test] fn extract_as_should_not_panic_when_expected_variant_matches_actual() { let many = ManyVariants::Three(true, 1, 2); assert_eq!(many.extract_as_three(), (&true, &1_u32, &2_i64)); } #[test] #[should_panic(expected = "expected One, got Three")] fn extract_as_should_panic_when_expected_variant_does_not_match_actual() { let many = ManyVariants::Three(true, 1, 2); many.extract_as_one(); } #[test] fn extract_as_mut_should_not_panic_when_expected_variant_matches_actual() { let mut many = ManyVariants::Three(true, 1, 2); assert_eq!( many.extract_as_three_mut(), (&mut true, &mut 1_u32, &mut 2_i64) ); } #[test] #[should_panic(expected = "expected One, got Three")] fn extract_as_mut_should_panic_when_expected_variant_does_not_match_actual() { let mut many = ManyVariants::Three(true, 1, 2); many.extract_as_one_mut(); } #[test] fn extract_into_should_not_panic_when_expected_variant_matches_actual() { let many = ManyVariants::Three(true, 1, 2); assert_eq!(many.extract_into_three(), (true, 1_u32, 2_i64)); } #[test] #[should_panic(expected = "expected One, got Three")] fn extract_into_should_panic_when_expected_variant_does_not_match_actual() { let many = ManyVariants::Three(true, 1, 2); many.extract_into_one(); }