reifydb-codec 0.9.0

Unified encoding and decoding for all ReifyDB values, frames, rows, keys, and boundaries
Documentation
// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2026 ReifyDB

use reifydb_codec::row::shape::{RowFamily, RowShape};
use reifydb_value::value::value_type::ValueType;

#[test]
fn test_set_get_u64() {
	let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint8]);
	let mut row = shape.allocate_pod();
	shape.set::<u64>(&mut row, 0, 18446744073709551615u64);
	assert_eq!(shape.get::<u64>(&row, 0), 18446744073709551615u64);
}

#[test]
fn test_try_get_u64() {
	let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint8]);
	let mut row = shape.allocate_pod();

	assert_eq!(shape.try_get::<u64>(&row, 0), None);

	shape.set::<u64>(&mut row, 0, 18446744073709551615u64);
	assert_eq!(shape.try_get::<u64>(&row, 0), Some(18446744073709551615u64));
}

#[test]
fn test_extremes() {
	let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint8]);
	let mut row = shape.allocate_pod();

	shape.set::<u64>(&mut row, 0, u64::MAX);
	assert_eq!(shape.get::<u64>(&row, 0), u64::MAX);

	let mut row2 = shape.allocate_pod();
	shape.set::<u64>(&mut row2, 0, u64::MIN);
	assert_eq!(shape.get::<u64>(&row2, 0), u64::MIN);

	let mut row3 = shape.allocate_pod();
	shape.set::<u64>(&mut row3, 0, 0u64);
	assert_eq!(shape.get::<u64>(&row3, 0), 0u64);
}

#[test]
fn test_large_values() {
	let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint8]);

	let test_values = [
		0u64,
		1u64,
		1_000_000_000u64,
		1_000_000_000_000_000_000u64,
		9_223_372_036_854_775_807u64, // i64::MAX
		9_223_372_036_854_775_808u64, // i64::MAX + 1
		18_000_000_000_000_000_000u64,
		18_446_744_073_709_551_614u64,
		18_446_744_073_709_551_615u64, // u64::MAX
	];

	for value in test_values {
		let mut row = shape.allocate_pod();
		shape.set::<u64>(&mut row, 0, value);
		assert_eq!(shape.get::<u64>(&row, 0), value);
	}
}

#[test]
fn test_memory_sizes() {
	let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint8]);

	let memory_sizes = [
		1024u64,                // 1 KB
		1048576u64,             // 1 MB
		1073741824u64,          // 1 GB
		1099511627776u64,       // 1 TB
		1125899906842624u64,    // 1 PB
		1152921504606846976u64, // 1 EB
	];

	for size in memory_sizes {
		let mut row = shape.allocate_pod();
		shape.set::<u64>(&mut row, 0, size);
		assert_eq!(shape.get::<u64>(&row, 0), size);
	}
}

#[test]
fn test_nanosecond_timestamps() {
	let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint8]);

	let ns_timestamps = [
		0u64,                   // Unix epoch in ns
		946684800000000000u64,  // 2000-01-01 in ns
		1640995200000000000u64, // 2022-01-01 in ns
		1735689600000000000u64, // 2025-01-01 in ns
	];

	for timestamp in ns_timestamps {
		let mut row = shape.allocate_pod();
		shape.set::<u64>(&mut row, 0, timestamp);
		assert_eq!(shape.get::<u64>(&row, 0), timestamp);
	}
}

#[test]
fn test_mixed_with_other_types() {
	let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint8, ValueType::Float8, ValueType::Uint8]);
	let mut row = shape.allocate_pod();

	shape.set::<u64>(&mut row, 0, 15_000_000_000_000_000_000u64);
	shape.set::<f64>(&mut row, 1, 3.14159265359f64);
	shape.set::<u64>(&mut row, 2, 12_000_000_000_000_000_000u64);

	assert_eq!(shape.get::<u64>(&row, 0), 15_000_000_000_000_000_000u64);
	assert_eq!(shape.get::<f64>(&row, 1), 3.14159265359);
	assert_eq!(shape.get::<u64>(&row, 2), 12_000_000_000_000_000_000u64);
}

#[test]
fn test_undefined_handling() {
	let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint8, ValueType::Uint8]);
	let mut row = shape.allocate_pod();

	shape.set::<u64>(&mut row, 0, 1234567890123456789u64);

	assert_eq!(shape.try_get::<u64>(&row, 0), Some(1234567890123456789));
	assert_eq!(shape.try_get::<u64>(&row, 1), None);

	shape.set_none(&mut row, 0);
	assert_eq!(shape.try_get::<u64>(&row, 0), None);
}

#[test]
fn test_try_get_u64_wrong_type() {
	let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Boolean]);
	let mut row = shape.allocate_pod();

	shape.set::<bool>(&mut row, 0, true);

	assert_eq!(shape.try_get::<u64>(&row, 0), None);
}