strum/strum_tests/tests/from_str.rs

extern crate strum;
#[macro_use]
extern crate strum_macros;

use std::str::FromStr;

#[derive(Debug,Eq,PartialEq,EnumString)]
enum Color {
    Red,
    Blue { hue: usize },
    #[strum(serialize="y",serialize="yellow")]
    Yellow,
    #[strum(default="true")]
    Green(String),
    #[strum(to_string="purp")]
    Purple,
}

#[test]
fn color_simple() {
    assert_eq!(Color::Red, Color::from_str("Red").unwrap());
}

#[test]
fn color_value() {
    assert_eq!(Color::Blue { hue: 0 }, Color::from_str("Blue").unwrap());
}

#[test]
fn color_serialize() {
    assert_eq!(Color::Yellow, Color::from_str("y").unwrap());
    assert_eq!(Color::Yellow, Color::from_str("yellow").unwrap());
}

#[test]
fn color_to_string() {
    assert_eq!(Color::Purple, Color::from_str("purp").unwrap());
}

#[test]
fn color_default() {
    assert_eq!(Color::Green(String::from("not found")),
               Color::from_str("not found").unwrap());
}

#[derive(Debug,Eq,PartialEq,EnumString)]
enum Week {
    Sunday,
    Monday,
    Tuesday,
    Wednesday,
    Thursday,
    Friday,
    Saturday,
}

#[test]
fn week_not_found() {
    assert_eq!(Result::Err(::strum::ParseError::VariantNotFound),
               Week::from_str("Humpday"));
}

#[test]
fn week_found() {
    assert_eq!(Result::Ok(Week::Sunday), Week::from_str("Sunday"));
    assert_eq!(Result::Ok(Week::Monday), Week::from_str("Monday"));
    assert_eq!(Result::Ok(Week::Tuesday), Week::from_str("Tuesday"));
    assert_eq!(Result::Ok(Week::Wednesday), Week::from_str("Wednesday"));
    assert_eq!(Result::Ok(Week::Thursday), Week::from_str("Thursday"));
    assert_eq!(Result::Ok(Week::Friday), Week::from_str("Friday"));
    assert_eq!(Result::Ok(Week::Saturday), Week::from_str("Saturday"));
}

#[derive(Debug,Eq,PartialEq,EnumString)]
enum Lifetime<'a> {
    Life(&'a str),
    None,
}

#[test]
fn lifetime_test() {
    assert_eq!(Lifetime::Life(""), Lifetime::from_str("Life").unwrap());
}

#[derive(Debug,Eq,PartialEq,EnumString)]
enum Generic<T: Default> {
    Gen(T),
    None,
}

#[test]
fn generic_test() {
    assert_eq!(Generic::Gen(""), Generic::from_str("Gen").unwrap());
}