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, 9_223_372_036_854_775_808u64, 18_000_000_000_000_000_000u64,
18_446_744_073_709_551_614u64,
18_446_744_073_709_551_615u64, ];
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, 1048576u64, 1073741824u64, 1099511627776u64, 1125899906842624u64, 1152921504606846976u64, ];
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, 946684800000000000u64, 1640995200000000000u64, 1735689600000000000u64, ];
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);
}