use rust_async_tuyapi::{
    mesparse::{CommandType, Message, MessageParser, TuyaVersion},
    Payload,
};
use serde_json::json;
use std::collections::HashMap;

fn create_test_payload() -> Payload {
    let mut dps = HashMap::new();
    dps.insert("1".to_string(), json!(true));
    Payload::new(
        "002004265ccf7fb1b659".to_string(),
        None,
        None,
        None,
        None,
        Some(serde_json::to_value(&dps).unwrap()),
    )
}
#[test]
fn encode_and_decode_message() {
    let payload = create_test_payload();
    let parser = MessageParser::create(TuyaVersion::ThreeOne, None).unwrap();
    let mut message_to_encode = Message::new(payload, CommandType::DpQuery);
    message_to_encode.seq_nr = Some(0);
    let encoded = parser.encode(&message_to_encode, false).unwrap();

    let decoded = parser.parse(&encoded).unwrap();

    assert_eq!(message_to_encode, decoded[0]);
}

#[test]
fn encode_and_decode_get_message_version_three_three() {
    let payload = create_test_payload();
    let parser = MessageParser::create(
        TuyaVersion::ThreeThree,
        Some("bbe88b3f4106d354".to_string()),
    )
    .unwrap();
    let mut message_to_encode = Message::new(payload, CommandType::DpQuery);
    message_to_encode.seq_nr = Some(0);

    let encoded = parser.encode(&message_to_encode, false).unwrap();
    let decoded = parser.parse(&encoded).unwrap();

    assert_eq!(message_to_encode, decoded[0]);
}

#[test]
fn encode_and_decode_set_message_version_three_three() {
    let payload = create_test_payload();
    let parser = MessageParser::create(
        TuyaVersion::ThreeThree,
        Some("bbe88b3f4106d354".to_string()),
    )
    .unwrap();
    let mut message_to_encode = Message::new(payload, CommandType::Control);
    message_to_encode.seq_nr = Some(0);

    let encoded = parser.encode(&message_to_encode, false).unwrap();
    let decoded = parser.parse(&encoded).unwrap();

    assert_eq!(message_to_encode, decoded[0]);
}

#[test]
fn decode_empty_message() {
    let payload = Payload::String("".to_string());

    let parser = MessageParser::create(TuyaVersion::ThreeOne, None).unwrap();
    let mut message_to_encode = Message::new(payload, CommandType::DpQuery);
    message_to_encode.seq_nr = Some(0);

    let encoded = parser.encode(&message_to_encode, false).unwrap();
    let decoded = parser.parse(&encoded).unwrap();

    assert_eq!(message_to_encode, decoded[0]);
}

#[test]
fn decode_corrupt_shortened_message() {
    let payload = create_test_payload();
    let parser = MessageParser::create(TuyaVersion::ThreeOne, None).unwrap();
    let message_to_encode = Message::new(payload, CommandType::DpQuery);

    let encoded = parser.encode(&message_to_encode, false).unwrap();

    assert!(parser.parse(&encoded[40..]).is_err());
}

#[test]
fn decode_corrupt_shorter_than_possible_message() {
    let payload = create_test_payload();
    let parser = MessageParser::create(TuyaVersion::ThreeOne, None).unwrap();
    let message_to_encode = Message::new(payload, CommandType::DpQuery);
    let encoded = parser.encode(&message_to_encode, false).unwrap();

    assert!(parser.parse(&encoded[0..23]).is_err());
}

#[test]
fn decode_corrupt_crc_mismatch_message() {
    let payload = create_test_payload();
    let parser = MessageParser::create(TuyaVersion::ThreeOne, None).unwrap();
    let message_to_encode = Message::new(payload, CommandType::DpQuery);
    let encoded = parser.encode(&message_to_encode, false).unwrap();
    // mess up the crc code
    let mut messedup_encoded: Vec<u8> = vec![];
    messedup_encoded.extend(encoded[0..encoded.len() - 8].iter());
    messedup_encoded.extend(hex::decode("DEADBEEF").unwrap());
    messedup_encoded.extend(hex::decode("0000AA55").unwrap());
    assert!(parser.parse(&messedup_encoded).is_err());
}