.

Author: mmastrac

gel-protocol/tests/client_messages.rs

@@ -54,6 +54,31 @@ fn client_handshake() -> Result<(), Box<dyn Error>> {
     Ok(())
 }
 
+#[test]
+fn parse2() -> Result<(), Box<dyn Error>> {
+    encoding_eq_ver!(
+        2,
+        0,
+        ClientMessage::Parse(Parse {
+            annotations: None,
+            allowed_capabilities: Capabilities::MODIFICATIONS,
+            compilation_flags: CompilationFlags::INJECT_OUTPUT_TYPE_NAMES,
+            implicit_limit: Some(77),
+            input_language: InputLanguage::EdgeQL,
+            output_format: IoFormat::Binary,
+            expected_cardinality: Cardinality::AtMostOne,
+            command_text: String::from("SELECT 1;"),
+            state: State {
+                typedesc_id: Uuid::from_u128(0),
+                data: Bytes::from(""),
+            },
+        }),
+        b"P\0\0\0A\0\0\0\0\0\0\0\0\0\x01\0\0\0\0\0\0\0\x02\0\0\0\0\0\0\0Mbo\
+          \0\0\0\tSELECT 1;\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0"
+    );
+    Ok(())
+}
+
 #[test]
 fn parse3() -> Result<(), Box<dyn Error>> {
     encoding_eq_ver!(
@@ -79,6 +104,36 @@ fn parse3() -> Result<(), Box<dyn Error>> {
     Ok(())
 }
 
+#[test]
+fn execute2() -> Result<(), Box<dyn Error>> {
+    encoding_eq_ver!(
+        2,
+        0,
+        ClientMessage::Execute1(Execute1 {
+            annotations: None,
+            allowed_capabilities: Capabilities::MODIFICATIONS,
+            compilation_flags: CompilationFlags::INJECT_OUTPUT_TYPE_NAMES,
+            implicit_limit: Some(77),
+            input_language: InputLanguage::EdgeQL,
+            output_format: IoFormat::Binary,
+            expected_cardinality: Cardinality::AtMostOne,
+            command_text: String::from("SELECT 1;"),
+            state: State {
+                typedesc_id: Uuid::from_u128(0),
+                data: Bytes::from(""),
+            },
+            input_typedesc_id: Uuid::from_u128(123),
+            output_typedesc_id: Uuid::from_u128(456),
+            arguments: Bytes::new(),
+        }),
+        b"O\0\0\0e\0\0\0\0\0\0\0\0\0\x01\0\0\0\0\0\0\0\x02\0\0\0\0\0\0\0Mbo\
+          \0\0\0\tSELECT 1;\
+          \0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\
+          \0\0\0{\0\0\0\0\0\0\0\0\0\0\0\0\0\0\x01\xc8\0\0\0\0"
+    );
+    Ok(())
+}
+
 #[test]
 fn execute3() -> Result<(), Box<dyn Error>> {
     encoding_eq_ver!(

gel-protocol/tests/server_messages.rs

@@ -126,6 +126,77 @@ fn parameter_status() -> Result<(), Box<dyn Error>> {
     Ok(())
 }
 
+#[test]
+fn command_complete1() -> Result<(), Box<dyn Error>> {
+    encoding_eq!(
+        ServerMessage::CommandComplete1(CommandComplete1 {
+            annotations: HashMap::new(),
+            capabilities: Capabilities::MODIFICATIONS,
+            status: "okay".to_string(),
+            state: None,
+        }),
+        b"C\0\0\0*\0\0\0\0\0\0\0\0\0\x01\0\0\0\x04okay\
+          \0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\
+          \0\0\0\0"
+    );
+    Ok(())
+}
+
+#[test]
+fn command_data_description2() -> Result<(), Box<dyn Error>> {
+    encoding_eq_ver!(
+        2,
+        0,
+        ServerMessage::CommandDataDescription1(CommandDataDescription1 {
+            annotations: HashMap::new(),
+            capabilities: Capabilities::MODIFICATIONS,
+            result_cardinality: Cardinality::AtMostOne,
+            input: RawTypedesc {
+                proto: ProtocolVersion::new(2, 0),
+                id: Uuid::from_u128(0xFF),
+                data: Bytes::from_static(b"\x04\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\xff\0\0"),
+            },
+            output: RawTypedesc {
+                proto: ProtocolVersion::new(2, 0),
+                id: Uuid::from_u128(0x105),
+                data: Bytes::from_static(b"\x02\0\0\0\0\0\0\0\0\0\0\0\0\0\0\x01\x05"),
+            },
+        }),
+        bconcat!(b"T\0\0\0[\0\0\0\0\0\0\0\0\0\x01o"
+                     b"\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\xff"
+                     b"\0\0\0\x13"
+                     b"\x04\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\xff\0"
+                     b"\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\x01\x05"
+                     b"\0\0\0\x11"
+                     b"\x02\0\0\0\0\0\0\0\0\0\0\0\0\0\0\x01\x05")
+    );
+    encoding_eq_ver!(
+        2,
+        0,
+        ServerMessage::CommandDataDescription1(CommandDataDescription1 {
+            annotations: HashMap::new(),
+            capabilities: Capabilities::MODIFICATIONS,
+            result_cardinality: Cardinality::NoResult,
+            input: RawTypedesc {
+                proto: ProtocolVersion::new(2, 0),
+                id: Uuid::from_u128(0xFF),
+                data: Bytes::from_static(b"\x04\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\xff\0\0"),
+            },
+            output: RawTypedesc {
+                proto: ProtocolVersion::new(2, 0),
+                id: Uuid::from_u128(0),
+                data: Bytes::from_static(b""),
+            },
+        }),
+        bconcat!(b"T\0\0\0J\0\0\0\0\0\0\0\0\0\x01n"
+                     b"\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\xff"
+                     b"\0\0\0\x13"
+                     b"\x04\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\xff\0"
+                     b"\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0")
+    );
+    Ok(())
+}
+
 #[test]
 fn data() -> Result<(), Box<dyn Error>> {
     encoding_eq!(