use reifydb_codec::row::shape::{RowFamily, RowShape};
use reifydb_value::value::value_type::ValueType;
#[test]
fn test_set_get_u8() {
let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint1]);
let mut row = shape.allocate_pod();
shape.set::<u8>(&mut row, 0, 255u8);
assert_eq!(shape.get::<u8>(&row, 0), 255u8);
}
#[test]
fn test_try_get_u8() {
let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint1]);
let mut row = shape.allocate_pod();
assert_eq!(shape.try_get::<u8>(&row, 0), None);
shape.set::<u8>(&mut row, 0, 255u8);
assert_eq!(shape.try_get::<u8>(&row, 0), Some(255u8));
}
#[test]
fn test_extremes() {
let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint1]);
let mut row = shape.allocate_pod();
shape.set::<u8>(&mut row, 0, u8::MAX);
assert_eq!(shape.get::<u8>(&row, 0), u8::MAX);
let mut row2 = shape.allocate_pod();
shape.set::<u8>(&mut row2, 0, u8::MIN);
assert_eq!(shape.get::<u8>(&row2, 0), u8::MIN);
let mut row3 = shape.allocate_pod();
shape.set::<u8>(&mut row3, 0, 0u8);
assert_eq!(shape.get::<u8>(&row3, 0), 0u8);
}
#[test]
fn test_various_values() {
let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint1]);
let test_values = [0u8, 1u8, 127u8, 128u8, 254u8, 255u8];
for value in test_values {
let mut row = shape.allocate_pod();
shape.set::<u8>(&mut row, 0, value);
assert_eq!(shape.get::<u8>(&row, 0), value);
}
}
#[test]
fn test_mixed_with_other_types() {
let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint1, ValueType::Boolean, ValueType::Uint1]);
let mut row = shape.allocate_pod();
shape.set::<u8>(&mut row, 0, 200u8);
shape.set::<bool>(&mut row, 1, true);
shape.set::<u8>(&mut row, 2, 100u8);
assert_eq!(shape.get::<u8>(&row, 0), 200u8);
assert_eq!(shape.get::<bool>(&row, 1), true);
assert_eq!(shape.get::<u8>(&row, 2), 100u8);
}
#[test]
fn test_undefined_handling() {
let shape = RowShape::testing(RowFamily::Pod, &[ValueType::Uint1, ValueType::Uint1]);
let mut row = shape.allocate_pod();
shape.set::<u8>(&mut row, 0, 42u8);
assert_eq!(shape.try_get::<u8>(&row, 0), Some(42));
assert_eq!(shape.try_get::<u8>(&row, 1), None);
shape.set_none(&mut row, 0);
assert_eq!(shape.try_get::<u8>(&row, 0), None);
}
#[test]
fn test_try_get_u8_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::<u8>(&row, 0), None);
}