dis_rs/common/data_query/
mod.rs

1pub mod builder;
2pub mod model;
3pub mod parser;
4pub mod writer;
5
6#[cfg(test)]
7mod tests {
8    use crate::common::data_query::model::DataQuery;
9    use crate::common::model::DisTimeStamp;
10    use crate::common::model::{EntityId, Pdu, PduHeader};
11    use crate::common::parser::parse_pdu;
12    use crate::enumerations::{PduType, VariableRecordType};
13    use bytes::BytesMut;
14
15    #[test]
16    fn data_query_internal_consistency() {
17        let header = PduHeader::new_v6(1, PduType::DataQuery);
18
19        let body = DataQuery::builder()
20            .with_origination_id(EntityId::new(10, 10, 10))
21            .with_receiving_id(EntityId::new(20, 20, 20))
22            .with_request_id(5)
23            .with_fixed_datums(vec![VariableRecordType::Azimuth_52340])
24            .with_variable_datums(vec![
25                VariableRecordType::Age_34100,
26                VariableRecordType::Acceleration_37000,
27            ])
28            .build()
29            .into_pdu_body();
30        let original_pdu =
31            Pdu::finalize_from_parts(header, body, DisTimeStamp::new_absolute_from_secs(100));
32        let pdu_length = original_pdu.header.pdu_length;
33        let original_length = original_pdu.pdu_length();
34
35        let mut buf = BytesMut::with_capacity(pdu_length as usize);
36
37        let serialized_length = original_pdu.serialize(&mut buf).unwrap();
38
39        assert_eq!(original_length, serialized_length);
40
41        let parsed = parse_pdu(&buf);
42        match parsed {
43            Ok(ref pdu) => {
44                assert_eq!(&original_pdu, pdu);
45            }
46            Err(ref err) => {
47                panic!("Parse error: {err}");
48            }
49        }
50    }
51}