use idmangler_lib::{
    decode_bytes, types::TransformVersion, AnyData, DataEncoder, DecodeError, StartData,
};

#[test]
fn encode_startdata() {
    let mut out = Vec::new();
    let ver = TransformVersion::Version1;

    StartData(ver).encode(ver, &mut out).unwrap();

    assert_eq!(out, Vec::from([0, 0]));
}

#[test]
fn decode_startdata() {
    let bytes: Vec<u8> = Vec::from([0, 0]);

    let ver = decode_bytes(&bytes).unwrap();
    let ver = ver.get(0).unwrap();

    assert_eq!(
        ver,
        &AnyData::StartData(StartData(TransformVersion::Version1))
    );
}

#[test]
fn decode_bad_startdata() {
    let bytes: Vec<u8> = Vec::from([0, 255]);

    let ver = decode_bytes(&bytes);

    match ver {
        Ok(_) => panic!("Expected an error"),
        Err(e) => match e {
            DecodeError::UnknownVersion(d) => assert_eq!(d, 255),
            _ => panic!("Expected an UnknownVersion error, got {:?}", e),
        },
    }
}