Improved unit tests for message peyload deserialization

datagnition · Jan 27, 2019 · e6ee7e5 · e6ee7e5
1 parent 83acf9a
commit e6ee7e5
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/src/network/message.rs b/src/network/message.rs
@@ -298,7 +298,13 @@ mod test {
         let msg = msg.unwrap();
         assert_eq!(msg.magic, 0xd9b4bef9);
         if let NetworkMessage::Version(version_msg) = msg.payload {
-            // TODO: do other checks
+            assert_eq!(version_msg.version, 70015);
+            assert_eq!(version_msg.services, 1037);
+            assert_eq!(version_msg.timestamp, 1548554224);
+            assert_eq!(version_msg.nonce, 13952548347456104954);
+            assert_eq!(version_msg.user_agent, "/Satoshi:0.17.1/");
+            assert_eq!(version_msg.start_height, 560275);
+            assert_eq!(version_msg.relay, true);
         } else {
             panic!("Wrong message type");
         }
@@ -331,7 +337,13 @@ mod test {
         let msg = msg.unwrap();
         assert_eq!(msg.magic, 0xd9b4bef9);
         if let NetworkMessage::Version(version_msg) = msg.payload {
-            // TODO: do other checks
+            assert_eq!(version_msg.version, 70015);
+            assert_eq!(version_msg.services, 1037);
+            assert_eq!(version_msg.timestamp, 1548554224);
+            assert_eq!(version_msg.nonce, 13952548347456104954);
+            assert_eq!(version_msg.user_agent, "/Satoshi:0.17.1/");
+            assert_eq!(version_msg.start_height, 560275);
+            assert_eq!(version_msg.relay, true);
         } else {
             panic!("Wrong message type");
         }