mirror of
https://github.com/danog/strum.git
synced 2024-12-11 17:09:36 +01:00
e558e913f4
* Add get_documentation() to EnumMessage. * Address 1st review comments. * Remove usage of strip_prefix to support older Rust versions. Co-authored-by: Peter Glotfelty <glotfelty.2@osu.edu>
131 lines
3.2 KiB
Rust
131 lines
3.2 KiB
Rust
use strum::EnumMessage;
|
|
|
|
#[derive(Debug, Eq, PartialEq, EnumMessage)]
|
|
enum Pets {
|
|
// This comment is not collected since it starts with "//" instead of "///".
|
|
#[strum(message = "I'm a dog")]
|
|
Dog,
|
|
/// I eat birds.
|
|
///
|
|
/// And fish.
|
|
#[strum(message = "I'm a cat")]
|
|
#[strum(detailed_message = "I'm a very exquisite striped cat")]
|
|
Cat,
|
|
/// I'm a fish.
|
|
#[strum(detailed_message = "My fish is named Charles McFish")]
|
|
Fish,
|
|
/// I'm a bird.
|
|
Bird,
|
|
/// This comment is not collected because it is explicitly disabled.
|
|
#[strum(disabled)]
|
|
Hamster,
|
|
}
|
|
|
|
#[test]
|
|
fn simple_message() {
|
|
assert_eq!("I'm a dog", (Pets::Dog).get_message().unwrap());
|
|
assert_eq!("I'm a dog", (Pets::Dog).get_detailed_message().unwrap());
|
|
}
|
|
|
|
#[test]
|
|
fn detailed_message() {
|
|
assert_eq!("I'm a cat", (Pets::Cat).get_message().unwrap());
|
|
assert_eq!(
|
|
"I'm a very exquisite striped cat",
|
|
(Pets::Cat).get_detailed_message().unwrap()
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn only_detailed_message() {
|
|
assert_eq!(None, (Pets::Fish).get_message());
|
|
assert_eq!(
|
|
"My fish is named Charles McFish",
|
|
(Pets::Fish).get_detailed_message().unwrap()
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn documentation() {
|
|
assert_eq!("I eat birds.\n\nAnd fish.\n", (Pets::Cat).get_documentation().unwrap());
|
|
assert_eq!("I'm a fish.", (Pets::Fish).get_documentation().unwrap());
|
|
assert_eq!("I'm a bird.", (Pets::Bird).get_documentation().unwrap());
|
|
}
|
|
|
|
#[test]
|
|
fn no_documentation() {
|
|
assert_eq!(None, (Pets::Dog).get_documentation());
|
|
}
|
|
|
|
#[test]
|
|
fn disabled_documentation() {
|
|
assert_eq!(None, (Pets::Hamster).get_documentation());
|
|
}
|
|
|
|
#[test]
|
|
fn no_message() {
|
|
assert_eq!(None, (Pets::Bird).get_message());
|
|
assert_eq!(None, (Pets::Bird).get_detailed_message());
|
|
}
|
|
|
|
#[test]
|
|
fn disabled_message() {
|
|
assert_eq!(None, (Pets::Hamster).get_message());
|
|
assert_eq!(None, (Pets::Hamster).get_detailed_message());
|
|
}
|
|
|
|
#[derive(Debug, Eq, PartialEq, EnumMessage)]
|
|
#[strum(serialize_all = "kebab_case")]
|
|
enum Brightness {
|
|
DarkBlack,
|
|
Dim {
|
|
glow: usize,
|
|
},
|
|
#[strum(serialize = "bright")]
|
|
BrightWhite,
|
|
}
|
|
|
|
#[test]
|
|
fn get_serializations() {
|
|
assert_eq!(
|
|
vec!["dark-black"],
|
|
(Brightness::DarkBlack).get_serializations()
|
|
);
|
|
assert_eq!(
|
|
vec!["dim"],
|
|
(Brightness::Dim { glow: 1 }).get_serializations()
|
|
);
|
|
assert_eq!(
|
|
vec!["bright"],
|
|
(Brightness::BrightWhite).get_serializations()
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn crate_module_path_test() {
|
|
pub mod nested {
|
|
pub mod module {
|
|
pub use strum;
|
|
}
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
#[derive(Debug, Eq, PartialEq, EnumMessage)]
|
|
#[strum(crate = "nested::module::strum")]
|
|
enum Pets {
|
|
#[strum(message = "I'm a dog")]
|
|
Dog,
|
|
#[strum(message = "I'm a cat")]
|
|
#[strum(detailed_message = "I'm a very exquisite striped cat")]
|
|
Cat,
|
|
#[strum(detailed_message = "My fish is named Charles McFish")]
|
|
Fish,
|
|
Bird,
|
|
#[strum(disabled)]
|
|
Hamster,
|
|
}
|
|
|
|
assert_eq!("I'm a dog", (Pets::Dog).get_message().unwrap());
|
|
assert_eq!("I'm a dog", (Pets::Dog).get_detailed_message().unwrap());
|
|
}
|