#![feature(get_type_id)]

use std::io::Error;

pub trait State {
    fn handle(self) -> Result<Box<State>, Error>;
}

struct A {}
impl State for A {
    fn handle(self) -> Result<Box<State>, Error> {
        println!("Running handle of A");
        Ok(Box::new(B {}))
    }
}

struct B {}
impl State for B {
    fn handle(self) -> Result<Box<State>, Error> {
        println!("Running handle of B");
        Ok(Box::new(A {}))
    }
}

#[test]
fn assert_works() {
    macro_rules! assert_state {
        ($machine:ident, $state:ident) => {
            use std::any::Any;
            let any: Box<Any> = Box::new($machine);
            assert!(
                any.is::<$state>(),
                "Failed to get to {} state.",
                stringify!($state)
            );
        };
    }

    let boxb: Box<Any> = Box::new(B {});

    println!("id do boxb: {:?}", boxb.get_type_id());

    let to_b = Box::new(A {}).handle().unwrap();

    let box_tob: Box<Any> = Box::new(to_b);

    println!("id do box_tob: {:?}", box_tob.get_type_id());

    println!("id do B puro: {:?}", std::any::TypeId::of::<B>());

    let to_b = Box::new(A {}).handle().unwrap();

    assert_state!(to_b, B);
}