use hacspec_lib::prelude::*;

#[test]
fn test_byte_sequences() {
    let msg = ByteSeq::from_hex("0388dace60b6a392f328c2b971b2fe78");
    let msg_u32 = Seq::<U32>::from_public_slice(&[0x0388dace, 0x60b6a392, 0xf328c2b9, 0x71b2fe78]);
    for i in 0..msg.num_chunks(4) {
        let (l, chunk) = msg.clone().get_chunk(4, i);
        assert_eq!(l, 4);
        assert_eq!(
            U8_from_U32(msg_u32[i] & U32(0xFF)).declassify(),
            U8::declassify(chunk[3])
        );
        assert_eq!(
            U8_from_U32((msg_u32[i] & U32(0xFF00)) >> 8).declassify(),
            U8::declassify(chunk[2])
        );
        assert_eq!(
            U8_from_U32((msg_u32[i] & U32(0xFF0000)) >> 16).declassify(),
            U8::declassify(chunk[1])
        );
        assert_eq!(
            U8_from_U32((msg_u32[i] & U32(0xFF000000)) >> 24).declassify(),
            U8::declassify(chunk[0])
        );
    }
}

#[test]
fn test_sequences_comparison() {
    let a = ByteSeq::from_public_slice(&[1, 2, 3]);
    let b = ByteSeq::from_public_slice(&[1, 2, 3]);
    assert_secret_seq_eq!(a, b, U8);
}

#[test]
fn test_seq_not() {
    let seq = ByteSeq::from_hex("ae125f62bd263edc");
    let expected = ByteSeq::from_hex("51eda09d42d9c123");
    assert_secret_seq_eq!(expected, !seq, U8);
}

#[test]
fn test_seq_or() {
    let seq1 = ByteSeq::from_hex("ae125f62bd263edc");
    let seq2 = ByteSeq::from_hex("51eda09d42d9c123");
    let expected = ByteSeq::from_hex("ffffffffffffffff");
    assert_secret_seq_eq!(expected, seq1 | seq2, U8);
}

#[test]
fn test_seq_xor() {
    let seq1 = ByteSeq::from_hex("296519d609eb2ab2");
    let seq2 = ByteSeq::from_hex("5a37737caae626c7");
    let expected = ByteSeq::from_hex("73526aaaa30d0c75");
    assert_secret_seq_eq!(expected, seq1 ^ seq2, U8);
}

#[test]
fn test_seq_and() {
    let seq1 = ByteSeq::from_hex("ae125f62bd263edc");
    let seq2 = ByteSeq::from_hex("51eda09d42d9c123");
    let expected = ByteSeq::from_hex("0000000000000000");
    assert_secret_seq_eq!(expected, seq1 & seq2, U8);
}