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(); 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(), DateTime::from_epoch_secs(946684800).unwrap(), DateTime::from_epoch_secs(1640995200).unwrap(), DateTime::from_epoch_secs(1735689600).unwrap(), ];
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();
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();
let post_2038 = DateTime::from_epoch_secs(2147483648).unwrap(); 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(); 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);
}