From e4f04283e6a268eaf10d133dcc3beaabaca82342 Mon Sep 17 00:00:00 2001
From: Daniel Akhterov <daniel@launchbadge.com>
Date: Mon, 17 Jun 2019 18:06:34 -0700
Subject: [PATCH] Add errpacket test

---
 mason-mariadb/src/protocol/server.rs | 38 ++++++++++++++++++----------
 1 file changed, 25 insertions(+), 13 deletions(-)

diff --git a/mason-mariadb/src/protocol/server.rs b/mason-mariadb/src/protocol/server.rs
index 11348e66..5cf28f60 100644
--- a/mason-mariadb/src/protocol/server.rs
+++ b/mason-mariadb/src/protocol/server.rs
@@ -313,7 +313,21 @@ impl Deserialize for OkPacket {
 
 impl Deserialize for ErrPacket {
     fn deserialize(buf: &mut Vec<u8>) -> Result<Self, Error> {
-        let mut index = 1;
+        let mut index = 0;
+
+        let length = deserialize_int_3(&buf, &mut index);
+
+        if buf.len() != length as usize {
+            return Err(err_msg("Lengths to do not match"));
+        }
+
+        let _sequence_number = deserialize_int_1(&buf, &mut index);
+
+        let packet_header = deserialize_int_1(&buf, &mut index);
+        if packet_header != 0xFF {
+            panic!("Packet header is not 0xFF for ErrPacket");
+        }
+
         let error_code = deserialize_int_2(&buf, &mut index);
 
         let mut stage = None;
@@ -430,26 +444,24 @@ mod test {
     }
 
     #[test]
-    fn it_decodes_okpacket() -> Result<(), Error> {
+    fn it_decodes_errpacket() -> Result<(), Error> {
         let mut buf = b"\
         \x0F\x00\x00\
         \x01\
         \xFF\
-        \xFB\
-        \xFB\
-        \x01\x01\
-        \x00\x00\
-        info\
+        \xEA\x03\
+        #\
+        HY000\
+        NO\
         "
         .to_vec();
 
-        let message = OkPacket::deserialize(&mut buf)?;
+        let message = ErrPacket::deserialize(&mut buf)?;
 
-        assert_eq!(message.affected_rows, None);
-        assert_eq!(message.last_insert_id, None);
-        assert!(!(message.server_status & ServerStatusFlag::SERVER_STATUS_IN_TRANS).is_empty());
-        assert_eq!(message.warning_count, 0);
-        assert_eq!(message.info, b"info".to_vec());
+        assert_eq!(message.error_code, 1002);
+        assert_eq!(message.sql_state_marker, Some(Bytes::from(b"#".to_vec())));
+        assert_eq!(message.sql_state, Some(Bytes::from(b"HY000".to_vec())));
+        assert_eq!(message.error_message, Some(Bytes::from(b"NO".to_vec())));
 
         Ok(())
     }