tests for reading the header

net/src/messages/message_header.rs

@@ -41,3 +41,35 @@ impl Serializable for MessageHeader {
 	}
 }
 
+#[cfg(test)]
+mod tests {
+	use bytes::Bytes;
+	use ser::{serialize, deserialize};
+	use super::MessageHeader;
+
+	#[test]
+	fn test_message_header_serialization() {
+		let expected = "f9beb4d96164647200000000000000001f000000ed52399b".into();
+		let header = MessageHeader {
+			magic: 0xd9b4bef9,
+			command: "addr".into(),
+			len: 0x1f,
+			checksum: [0xed, 0x52, 0x39, 0x9b],
+		};
+
+		assert_eq!(serialize(&header), expected);
+	}
+
+	#[test]
+	fn test_message_header_deserialization() {
+		let raw: Bytes = "f9beb4d96164647200000000000000001f000000ed52399b".into();
+		let expected = MessageHeader {
+			magic: 0xd9b4bef9,
+			command: "addr".into(),
+			len: 0x1f,
+			checksum: [0xed, 0x52, 0x39, 0x9b],
+		};
+
+		assert_eq!(expected, deserialize(&raw).unwrap());
+	}
+}

p2p/src/io/read_header.rs

@@ -29,3 +29,24 @@ impl<A> Future for ReadHeader<A> where A: io::Read {
 		}
 	}
 }
+
+#[cfg(test)]
+mod tests {
+	use futures::Future;
+	use bytes::Bytes;
+	use net::messages::MessageHeader;
+	use super::read_header;
+
+	#[test]
+	fn test_read_header() {
+		let raw: Bytes = "f9beb4d96164647200000000000000001f000000ed52399b".into();
+		let expected = MessageHeader {
+			magic: 0xd9b4bef9,
+			command: "addr".into(),
+			len: 0x1f,
+			checksum: [0xed, 0x52, 0x39, 0x9b],
+		};
+
+		assert_eq!(read_header(raw.as_ref()).wait().unwrap().1, expected);
+	}
+}