module snappy

import os
import benchmark

const samples_folder = os.join_path(os.dir(@FILE), 'samples')

fn s(fname string) string {
	return os.join_path(samples_folder, fname)
}

fn test_framing_compress_decompress_alice() {
	alice := os.read_bytes(s('alice29.txt'))!

	alice_compressed := encode_stream(alice)
	assert decode_stream(alice_compressed)! == alice, 'compressed alice should be the same as decompressed'
}

fn test_precompressed_framing_alice() {
	alice_compressed := os.read_bytes(s('alice29_framing.snappy'))!
	alice := os.read_bytes(s('alice29.txt'))!

	// Normalize line endings before comparing
	decompressed_str := decode_stream(alice_compressed)!.bytestr().replace('\r\n', '\n')
	alice_str := alice.bytestr().replace('\r\n', '\n')

	assert decompressed_str == alice_str, 'compressed alice should be the same as decompressed'
}

fn test_benchmark_framing_alice() {
	alice := os.read_bytes(s('alice29.txt'))!

	mut b := benchmark.start()

	encode_stream(alice)

	b.measure('framing_alice')
}

fn test_benchmark_block_alice() {
	alice := os.read_bytes(s('alice29.txt'))!

	mut b := benchmark.start()

	compress(alice)

	b.measure('block_alice')
}

fn test_block_compress_decompress_alice() {
	alice := os.read_bytes(s('alice29.txt'))!

	alice_compressed := compress(alice)
	assert decompress(alice_compressed)! == alice, 'compressed alice should be the same as decompressed'
}

fn test_precompressed_block_alice() {
	alice_compressed := os.read_bytes(s('alice29_block.snappy'))!
	alice := os.read_bytes(s('alice29.txt'))!

	decompressed := decompress(alice_compressed)!

	// Normalize line endings before comparing
	decompressed_str := decompressed.bytestr().replace('\r\n', '\n')
	alice_str := alice.bytestr().replace('\r\n', '\n')

	assert decompressed_str == alice_str, 'compressed alice should be the same as decompressed'
}