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::{datetime::DateTime, value_type::ValueType};

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

	let value = DateTime::new(2024, 9, 9, 08, 17, 0, 1234).unwrap();
	shape.set::<DateTime>(&mut row, 0, value.clone());
	assert_eq!(shape.get::<DateTime>(&row, 0), value);
}

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

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

	let test_datetime = DateTime::from_epoch_secs(1642694400).unwrap();
	shape.set::<DateTime>(&mut row, 0, test_datetime.clone());
	assert_eq!(shape.try_get::<DateTime>(&row, 0), Some(test_datetime));
}

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

	let epoch = DateTime::default(); // Unix epoch
	shape.set::<DateTime>(&mut row, 0, epoch.clone());
	assert_eq!(shape.get::<DateTime>(&row, 0), epoch);
}

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

	let precise_datetime = DateTime::new(2024, 12, 25, 15, 30, 45, 123456789).unwrap();
	shape.set::<DateTime>(&mut row, 0, precise_datetime.clone());
	assert_eq!(shape.get::<DateTime>(&row, 0), precise_datetime);
}

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

	let test_datetimes = [
		DateTime::from_epoch_secs(0).unwrap(),          // Unix epoch
		DateTime::from_epoch_secs(946684800).unwrap(),  // 2000-01-01
		DateTime::from_epoch_secs(1640995200).unwrap(), // 2022-01-01
		DateTime::from_epoch_secs(1735689600).unwrap(), // 2025-01-01
	];

	for datetime in test_datetimes {
		let mut row = shape.allocate_pod();
		shape.set::<DateTime>(&mut row, 0, datetime.clone());
		assert_eq!(shape.get::<DateTime>(&row, 0), datetime);
	}
}

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

	let datetime1 = DateTime::new(2025, 6, 15, 12, 0, 0, 0).unwrap();
	let datetime2 = DateTime::new(1995, 3, 22, 18, 30, 45, 500000000).unwrap();

	shape.set::<DateTime>(&mut row, 0, datetime1.clone());
	shape.set::<bool>(&mut row, 1, true);
	shape.set::<DateTime>(&mut row, 2, datetime2.clone());
	shape.set::<i64>(&mut row, 3, 1234567890i64);

	assert_eq!(shape.get::<DateTime>(&row, 0), datetime1);
	assert_eq!(shape.get::<bool>(&row, 1), true);
	assert_eq!(shape.get::<DateTime>(&row, 2), datetime2);
	assert_eq!(shape.get::<i64>(&row, 3), 1234567890);
}

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

	let datetime = DateTime::new(2025, 7, 4, 16, 20, 15, 750000000).unwrap();
	shape.set::<DateTime>(&mut row, 0, datetime.clone());

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

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

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

	// The slot holds u64 nanos, so no sub-second digit may be rounded away.
	let high_precision = DateTime::new(2024, 1, 1, 0, 0, 0, 999999999).unwrap();
	shape.set::<DateTime>(&mut row, 0, high_precision.clone());

	let retrieved = shape.get::<DateTime>(&row, 0);
	assert_eq!(retrieved, high_precision);

	let orig_nanos = high_precision.to_nanos();
	let ret_nanos = retrieved.to_nanos();
	assert_eq!(orig_nanos, ret_nanos);
}

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

	// Past i32 seconds since the epoch, which is where a 32-bit timestamp would wrap.
	let post_2038 = DateTime::from_epoch_secs(2147483648).unwrap(); // 2038-01-19
	shape.set::<DateTime>(&mut row, 0, post_2038.clone());
	assert_eq!(shape.get::<DateTime>(&row, 0), post_2038);
}

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

	let far_future = DateTime::from_epoch_secs(4102444800).unwrap(); // 2100-01-01
	shape.set::<DateTime>(&mut row, 0, far_future.clone());
	assert_eq!(shape.get::<DateTime>(&row, 0), far_future);
}

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

	let microsecond_precision = DateTime::new(2024, 6, 15, 14, 30, 25, 123456000).unwrap();
	shape.set::<DateTime>(&mut row, 0, microsecond_precision.clone());
	assert_eq!(shape.get::<DateTime>(&row, 0), microsecond_precision);
}

#[test]
fn test_try_get_datetime_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::<DateTime>(&row, 0), None);
}