rust-protocol: Implement a simple unit test.

common/rust-protocol/src/lib.rs

@@ -2,12 +2,15 @@
 //! base station and all sensor/display nodes.
 #![no_std]
 
+#[cfg(test)]
+extern crate std;
+
 use core::convert::TryInto;
 
 use crc16::{State, KERMIT};
 use xxtea_nostd::{decrypt, encrypt};
 
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, PartialEq)]
 pub enum Packet {
     GetSalt,
     Salt(u64),
@@ -83,7 +86,7 @@ impl Packet {
     }
 }
 
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, PartialEq)]
 pub struct Report {
     pub count: u8,
     pub values: [Value; 8],
@@ -116,7 +119,7 @@ impl Report {
     }
 }
 
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, PartialEq)]
 pub struct GetValues {
     pub count: u8,
     pub location: Location,
@@ -135,7 +138,7 @@ impl GetValues {
     }
 }
 
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, PartialEq)]
 pub struct Values {
     pub count: u8,
     pub location: Location,
@@ -154,7 +157,7 @@ impl Values {
     }
 }
 
-#[derive(Debug, Clone, Copy)]
+#[derive(Debug, Clone, Copy, PartialEq)]
 pub enum Location {
     Livingroom,
     Bathroom,
@@ -163,7 +166,7 @@ pub enum Location {
     Balcony,
 }
 
-#[derive(Debug, Clone, Copy)]
+#[derive(Debug, Clone, Copy, PartialEq)]
 pub enum Value {
     Invalid,
     Time(u64),
@@ -230,7 +233,7 @@ impl Value {
     }
 }
 
-#[derive(Debug, Clone, Copy)]
+#[derive(Debug, Clone, Copy, PartialEq)]
 pub enum ValueType {
     Time,
     Temperature,
@@ -272,3 +275,28 @@ pub enum Error {
     InvalidLocation,
     TooShort,
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use std::println;
+
+    #[test]
+    fn test_identity() {
+        let packets = [Packet::GetSalt, Packet::Salt(0x12345678abcdef12)];
+
+        let key = [
+            0x12, 0x34, 0x56, 0x78, 0xab, 0xcd, 0xef, 0x12, 0x12, 0x34, 0x56, 0x78, 0xab, 0xcd,
+            0xef, 0x12,
+        ];
+
+        for packet in packets.iter() {
+            let mut encoded = [0u8; 32];
+            packet.encode_and_encrypt(&key, 0x1234123412341234, &mut encoded);
+            assert_eq!(encoded[0], 0x34);
+            assert_eq!(encoded[7], 0x12);
+            let decoded = Packet::decrypt_and_decode(&key, &mut encoded).unwrap();
+            // TODO: Compare encoded and decoded.
+        }
+    }
+}