module sha3

fn test_shake256_streaming_matches_oneshot() {
    data := 'hello world'.bytes()
    // oneshot
    expected := shake256(data, 64)

    // streaming
    mut s := new_shake256()
    s.write(data)
    result := s.read(64)

    assert result == expected, 'streaming SHAKE-256 output differs from one-shot'
}

fn test_shake128_streaming_matches_oneshot() {
    data := 'hello world'.bytes()
    expected := shake128(data, 64)

    mut s := new_shake128()
    s.write(data)
    result := s.read(64)

    assert result == expected, 'streaming SHAKE-128 output differs from one-shot'
}

fn test_shake256_incremental_write() {
    data := 'the quick brown fox jumps over the lazy dog'.bytes()
    expected := shake256(data, 128)

    mut s := new_shake256()
    s.write(data[..10])
    s.write(data[10..25])
    s.write(data[25..])
    result := s.read(128)

    assert result == expected, 'incremental write produced different output'
}

fn test_shake256_incremental_read() {
    data := 'test data for incremental reads'.bytes()

    // all at once
    mut s1 := new_shake256()
    s1.write(data)
    all_at_once := s1.read(200)

    // in chunks
    mut s2 := new_shake256()
    s2.write(data)
    mut chunked := []u8{}
    chunked << s2.read(50)
    chunked << s2.read(80)
    chunked << s2.read(70)

    assert chunked == all_at_once, 'incremental read produced different output'
}

fn test_shake128_large_output() {
    data := 'large output test'.bytes()
    mut s := new_shake128()
    s.write(data)
    // more than one block (168 bytes in shake128)
    result := s.read(500)
    assert result.len == 500
}

fn test_shake_reset() {
    data := 'reset test'.bytes()

    mut s := new_shake256()
    s.write(data)
    first := s.read(32)

    s.reset()
    s.write(data)
    second := s.read(32)

    assert first == second, 'reset did not restore initial state'
}