zecfoundation
/
zebra
mirror of https://github.com/ZcashFoundation/zebra.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add custom Debug impl for DecodeState 

Relates to #63
This commit is contained in:
Deirdre Connolly 2019-10-17 01:43:03 -04:00 committed by Henry de Valence
parent 96ad551681
commit 0db69addec
1 changed files with 42 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
zebra-network/src/protocol/codec.rs
View File

@ -1,5 +1,6 @@
//! A Tokio codec mapping byte streams to Bitcoin message streams.
use std::fmt;
use std::io::{Cursor, Read, Write};
use byteorder::{LittleEndian, ReadBytesExt, WriteBytesExt};
@ -282,7 +283,6 @@ impl Codec {
// ======== Decoding =========
#[derive(Debug)]
enum DecodeState {
Head,
Body {
@ -292,6 +292,24 @@ enum DecodeState {
},
}
impl fmt::Debug for DecodeState {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
match self {
DecodeState::Head => write!(f, "DecodeState::Head"),
DecodeState::Body {
body_len,
command,
checksum,
} => f
.debug_struct("DecodeState::Body")
.field("body_len", &body_len)
.field("command", &String::from_utf8_lossy(command))
.field("checksum", &checksum)
.finish(),
}
}
}
impl Decoder for Codec {
type Item = Message;
type Error = Error;
@ -619,4 +637,27 @@ mod tests {
assert_eq!(v, v_parsed);
}
#[test]
fn decode_state_debug() {
assert_eq!(format!("{:?}", DecodeState::Head), "DecodeState::Head");
let decode_state = DecodeState::Body {
body_len: 43,
command: [118, 101, 114, 115, 105, 111, 110, 0, 0, 0, 0, 0],
checksum: Sha256dChecksum([186, 250, 162, 227]),
};
assert_eq!(format!("{:?}", decode_state),
"DecodeState::Body { body_len: 43, command: \"version\\u{0}\\u{0}\\u{0}\\u{0}\\u{0}\", checksum: Sha256dChecksum(\"bafaa2e3\") }");
let decode_state = DecodeState::Body {
body_len: 43,
command: [118, 240, 144, 128, 105, 111, 110, 0, 0, 0, 0, 0],
checksum: Sha256dChecksum([186, 250, 162, 227]),
};
assert_eq!(format!("{:?}", decode_state),
"DecodeState::Body { body_len: 43, command: \"v<EFBFBD>ion\\u{0}\\u{0}\\u{0}\\u{0}\\u{0}\", checksum: Sha256dChecksum(\"bafaa2e3\") }");
}
}