module deflate

fn test_zlib_roundtrip() {
    data := 'Hello world!'.bytes()
    compressed := compress(data)!
    assert compressed[0] == 0x78 && compressed[1] == 0x9c // zlib header
    assert decompress(compressed)! == data
}

fn test_gzip_roundtrip() {
    data := 'Hello gzip!'.repeat(10).bytes()
    compressed := compress(data, format: .gzip)!
    assert compressed[0] == 0x1f && compressed[1] == 0x8b // gzip magic
    assert decompress(compressed)! == data
}

fn test_raw_deflate_roundtrip() {
    data := 'raw deflate'.repeat(20).bytes()
    raw := compress(data, format: .raw_deflate)!
    decoded := decompress(raw)! // auto-detected as raw
    assert decoded == data
}

fn test_decompress_auto_detects_all_formats() {
    data := 'multi-format detection test'.repeat(5).bytes()
    assert decompress(compress(data)!)! == data
    assert decompress(compress(data, format: .gzip)!)! == data
    assert decompress(compress(data, format: .raw_deflate)!)! == data
}

fn test_wrapper_helpers_match_unified_api() {
    data := 'wrapper compatibility'.repeat(8).bytes()
    assert compress(data)! == compress(data, format: .zlib)!
    assert compress_gzip(data)! == compress(data, format: .gzip)!
    assert compress_raw(data)! == compress(data, format: .raw_deflate)!
}

fn test_roundtrip_repeated() {
    data := 'abcabc'.repeat(100).bytes()
    compressed := compress(data)!
    assert compressed.len < data.len
    assert decompress(compressed)! == data
}

fn test_bad_compression_method_fails() {
    bad := [u8(0x79), 0x18, 0x00, 0x00, 0x00, 0x00]
    decompress(bad) or {
        assert err.msg().len > 0
        return
    }
    assert false
}

fn test_corrupt_checksum_fails() {
    mut enc := compress(('hello world').repeat(10).bytes())!
    // flip a byte in the adler32 footer
    enc[enc.len - 1] ^= 0xff
    decompress(enc) or {
        assert err.msg().contains('adler32')
        return
    }
    assert false
}