write some tests

amqp_transport/src/classes/parse_helper.rs

@@ -6,7 +6,7 @@ use crate::error::{ConException, ProtocolError, TransError};
 use nom::branch::alt;
 use nom::bytes::complete::{tag, take};
 use nom::error::ErrorKind;
-use nom::multi::count;
+use nom::multi::{count, many0};
 use nom::number::complete::{f32, f64, i16, i32, i64, i8, u16, u32, u64, u8};
 use nom::number::Endianness::Big;
 use nom::Err;
@@ -140,10 +140,20 @@ pub fn timestamp(input: &[u8]) -> IResult<Timestamp> {
 }
 
 pub fn table(input: &[u8]) -> IResult<Table> {
-    let (input, len) = u32(Big)(input)?;
-    let (input, values) = count(table_value_pair, usize::try_from(len).unwrap())(input)?;
+    let (input, size) = u32(Big)(input)?;
+    let (table_input, rest_input) = input.split_at(size.try_into().unwrap());
+
+    let (input, values) = many0(table_value_pair)(table_input)?;
+
+    if input != &[] {
+        fail!(format!(
+            "table longer than expected, expected = {size}, remaining = {}",
+            input.len()
+        ));
+    }
+
     let table = HashMap::from_iter(values.into_iter());
-    Ok((input, table))
+    Ok((rest_input, table))
 }
 
 fn table_value_pair(input: &[u8]) -> IResult<(TableFieldName, FieldValue)> {

amqp_transport/src/classes/tests.rs

@@ -99,7 +99,7 @@ fn pack_many_bits() {
 
 #[test]
 fn random_ser_de() {
-    const ITERATIONS: usize = 100000;
+    const ITERATIONS: usize = 1000;
     let mut rng = rand::rngs::StdRng::from_seed([0; 32]);
 
     for _ in 0..ITERATIONS {

amqp_transport/src/classes/write_helper.rs

@@ -74,14 +74,17 @@ pub fn timestamp<W: Write>(value: Timestamp, writer: &mut W) -> Result<(), Trans
 }
 
 pub fn table<W: Write>(table: Table, writer: &mut W) -> Result<(), TransError> {
-    let len = u32::try_from(table.len()).context("table too big")?;
-    writer.write_all(&len.to_be_bytes())?;
+    let mut table_buf = Vec::new();
 
     for (field_name, value) in table {
-        shortstr(field_name, writer)?;
-        field_value(value, writer)?;
+        shortstr(field_name, &mut table_buf)?;
+        field_value(value, &mut table_buf)?;
     }
 
+    let len = u32::try_from(table_buf.len()).context("table too big")?;
+    writer.write_all(&len.to_be_bytes())?;
+    writer.write_all(&table_buf)?;
+
     Ok(())
 }
 

amqp_transport/src/connection.rs

@@ -63,12 +63,12 @@ impl Connection {
         let mut payload = Vec::with_capacity(64);
         classes::write::write_method(start_method, &mut payload)?;
         frame::write_frame(
-            &mut self.stream,
             &Frame {
                 kind: FrameType::Method,
                 channel: 0,
                 payload,
             },
+            &mut self.stream,
         )
         .await?;
 
@@ -132,12 +132,12 @@ fn server_properties(host: SocketAddr) -> classes::Table {
     };
 
     HashMap::from([
-        ("host".to_string(), host_value),
-        ("product".to_string(), ss("no name yet")),
+        //("host".to_string(), host_value),
+        //("product".to_string(), ss("no name yet")),
         ("version".to_string(), ss("0.1.0")),
-        ("platform".to_string(), ss("microsoft linux")),
-        ("copyright".to_string(), ss("MIT")),
-        ("information".to_string(), ss("hello reader")),
-        ("uwu".to_string(), ss("owo")),
+        //("platform".to_string(), ss("microsoft linux")),
+        //("copyright".to_string(), ss("MIT")),
+        //("information".to_string(), ss("hello reader")),
+        //("uwu".to_string(), ss("owo")),
     ])
 }

amqp_transport/src/frame.rs

@@ -30,7 +30,7 @@ pub enum FrameType {
     Heartbeat = 8,
 }
 
-pub async fn write_frame<W>(mut w: W, frame: &Frame) -> Result<()>
+pub async fn write_frame<W>(frame: &Frame, mut w: W, ) -> Result<()>
 where
     W: AsyncWriteExt + Unpin,
 {

amqp_transport/src/lib.rs

@@ -6,6 +6,8 @@ mod classes;
 mod connection;
 mod error;
 mod frame;
+#[cfg(test)]
+mod tests;
 
 use crate::connection::Connection;
 use anyhow::Result;

amqp_transport/src/tests.rs

@@ -0,0 +1,140 @@
+use crate::classes::FieldValue;
+use crate::frame::FrameType;
+use crate::{classes, frame};
+use std::collections::HashMap;
+
+#[tokio::test]
+async fn write_start_frame() {
+    let mut payload = Vec::new();
+    let method = classes::Class::Connection(classes::Connection::Start {
+        version_major: 0,
+        version_minor: 9,
+        server_properties: HashMap::from([(
+            "version".to_string(),
+            FieldValue::ShortString("0.1.0".to_string()),
+        )]),
+        mechanisms: vec![],
+        locales: vec![],
+    });
+
+    classes::write::write_method(method, &mut payload).unwrap();
+
+    let frame = frame::Frame {
+        kind: FrameType::Method,
+        channel: 0,
+        payload,
+    };
+
+    let mut output = Vec::new();
+
+    frame::write_frame(&frame, &mut output).await.unwrap();
+
+    #[rustfmt::skip]
+    let expected = [
+        /* type, octet */
+        1u8, // = method
+        /* channel, short */
+        0, 0,
+        /* size, long */
+        /* count all the bytes in the payload, 33 here */
+        0, 0, 0, 33,
+        /* payload */
+
+        /* class-id, short */
+        0, 10, // connection
+        /* method-id, short */
+        0, 10, // start
+        /* version-major, octet */
+        0,
+        /* version-minor, octet */
+        9,
+        /* server-properties, table */
+          /* table-size, long (actual byte size) */
+          0, 0, 0, 15,
+          /* table-items */
+          /* name ("version"), shortstr */
+          /* len (7) ; bytes */
+          7, b'v', b'e', b'r', b's', b'i', b'o', b'n',
+          /* value, a shortstr ("0.1.0") here */
+             /* tag (s) ; len (5) ; data */
+             b's', 5, b'0', b'.', b'1', b'.', b'0',
+        /* mechanisms, longstr */
+        /* str-len, long ; data (none here) */
+        0, 0, 0, 0,  
+        /* locales, longstr */
+        /* str-len, long ; data (none here) */
+        0, 0, 0, 0,
+
+        /* frame-end */
+        0xCE,
+    ];
+
+    assert_eq!(expected.as_slice(), output.as_slice());
+}
+
+#[tokio::test]
+async fn read_start_ok_payload() {
+    // comes from a python pika amqp client - can assumed to be valid
+    // annotated manually
+    #[rustfmt::skip]
+    let raw_data = [
+        /* Connection.Start-Ok */
+        0, 10, 0, 11,
+        /* field client-properties */
+        /* byte size of the table */
+        0, 0, 0, 254,
+          /* first key of len 7, "product"*/
+          7, 112, 114, 111, 100, 117, 99, 116,
+          /* value is of type 83 ("S"), long-string */
+          /* has length 26 "Pika Python Client Library" */
+          83, 0, 0, 0, 26,
+          80, 105, 107, 97, 32, 80, 121, 116, 104, 111, 110, 32, 67, 108, 105, 101, 110, 116, 32, 76, 105, 98,
+          114, 97, 114, 121,
+          /* second key of len 8, "platform" */
+          8, 112, 108, 97, 116, 102, 111, 114, 109,
+          /* value is of type 83("S"), long-string */
+          /* has length 13, "Python 3.8.10" */
+          83, 0, 0, 0, 13,
+          80, 121, 116, 104, 111, 110, 32, 51, 46, 56, 46, 49, 48,
+          /* third key has len 12 "capabilities" */
+          12, 99, 97, 112, 97, 98, 105, 108, 105, 116, 105, 101, 115,
+          /* type is 70 F (table), with byte-len of 111 */
+          70, 0, 0, 0, 111,
+            /* first key has length 28, "authentication_failure_close" */
+            28, 97, 117, 116, 104, 101, 110, 116, 105, 99, 97, 116, 105, 111, 110, 95, 102, 97, 105, 108, 117, 114, 101, 95, 99, 108, 111, 115, 101,
+            /* value of type 116, "t", boolean, true */
+            116, 1,
+            /* second key has length 10, "basic.nack" */
+            10, 98, 97, 115, 105, 99, 46, 110, 97, 99, 107,
+            /* value of type 116, "t", boolean, true */
+            116, 1,
+            /* third key has length 18 "connection.blocked" */
+            18, 99, 111, 110, 110, 101, 99, 116, 105, 111, 110, 46, 98, 108, 111, 99, 107, 101, 100,
+            /* value of type 116, "t", boolean, true */
+            116, 1,
+            /* fourth key has length 22 "consumer_cancel_notify" */
+            22, 99, 111, 110, 115, 117, 109, 101, 114, 95, 99, 97, 110, 99, 101, 108, 95, 110, 111, 116, 105, 102, 121,
+            /* value of type 116, "t", boolean, true */
+            116, 1,
+            /* fifth key has length 18 "publisher_confirms" */
+            18, 112, 117, 98, 108, 105, 115, 104, 101, 114, 95, 99, 111, 110, 102, 105, 114, 109, 115,
+            /* value of type 116, "t", boolean, true */
+            116, 1,
+          /* unsure after this */
+          /* sixth key has length 11 "information" */
+          11, 105, 110, 102, 111, 114, 109, 97, 116, 105, 111, 110,
+          /* value of type 83, "S" long-str ; len 24 */
+          83, 0, 0, 0, 24,
+/* it gets very very confusing and possibly wrong on my side here */
+          /* data "See http://pika.rtf\n\x00.or" */
+          83, 101, 101, 32, 104, 116, 116, 112, 58, 47, 47, 112, 105, 107, 97, 46, 114, 116, 102, 10, 0, 46, 111, 114, 103, 7, 118, 101, 114, 115, 105, 111, 110, 83, 0, 0, 0, 5, 49, 46, 49, 46,
+
+        /* table should only end here */
+
+        48, 5, 80, 76, 65, 73, 78, 0, 0, 0, 7, 0, 97, 100, 109, 105, 110, 0,
+        /* locale, shortstr, len 5 "en_US" */
+        5, 101, 110, 95, 85, 83,
+    ];
+
+    classes::parse_method(&raw_data).unwrap();
+}