use reifydb_codec::row::{
bytes::{RowBuilder, SHAPE_HEADER_SIZE},
shape::{RowFamily, RowShape, RowShapeField},
table::EncodedTableRow,
};
use reifydb_value::value::{datetime::DateTime, value_type::ValueType};
fn shape() -> RowShape {
RowShape::new(
RowFamily::Table,
vec![
RowShapeField::unconstrained("id", ValueType::Int4),
RowShapeField::unconstrained("payload", ValueType::Int4),
],
)
}
#[test]
fn the_table_header_is_the_full_source_header() {
assert_eq!(shape().header_size(), SHAPE_HEADER_SIZE);
assert_eq!(shape().header_size(), 33);
}
#[test]
fn the_typed_view_reads_the_stamps_the_builder_wrote() {
let shape = shape();
let mut row = shape.allocate_table();
row.set_timestamps(DateTime::from_millis(111), DateTime::from_millis(222));
row.set_time(DateTime::from_millis(333));
let frozen = row.freeze_bytes();
let typed = EncodedTableRow::view(&frozen);
assert_eq!(typed.created_at(), DateTime::from_millis(111));
assert_eq!(typed.updated_at(), DateTime::from_millis(222));
assert_eq!(typed.time(), Some(DateTime::from_millis(333)));
assert_eq!(typed.fingerprint(), shape.fingerprint());
}
#[test]
fn an_unstamped_row_reports_no_time() {
let frozen = shape().allocate_table().freeze_bytes();
assert_eq!(EncodedTableRow::view(&frozen).time(), None);
}
#[test]
fn the_typed_definedness_agrees_with_the_shape() {
let shape = shape();
let mut row = shape.allocate_table();
shape.set::<i32>(&mut row, 1, 9);
let frozen = row.freeze_bytes();
let typed = EncodedTableRow::view(&frozen);
for index in 0..shape.field_count() {
assert_eq!(typed.is_defined(index), shape.is_defined(&frozen, index), "field {index} disagrees");
}
assert!(typed.is_defined(1));
assert!(!typed.is_defined(0));
}
#[test]
fn the_body_begins_where_the_bitvec_begins() {
let shape = shape();
let mut row = shape.allocate_table();
shape.set::<i32>(&mut row, 0, 7);
let frozen = row.freeze_bytes();
let typed = EncodedTableRow::view(&frozen);
assert_eq!(typed.body().len(), frozen.len() - SHAPE_HEADER_SIZE);
assert_eq!(typed.body(), &frozen.as_slice()[shape.header_size()..]);
}
#[test]
fn viewing_and_converting_preserve_the_bytes() {
let shape = shape();
let mut row = shape.allocate_table();
shape.set::<i32>(&mut row, 0, 42);
row.set_timestamps(DateTime::from_millis(5), DateTime::from_millis(6));
let frozen = row.freeze_bytes();
let expected = frozen.as_slice().to_vec();
assert_eq!(EncodedTableRow::view(&frozen).as_slice(), expected.as_slice());
assert_eq!(EncodedTableRow::from(frozen.clone()).into_bytes(), frozen);
}