Browse Source

Add invalid code test for decoding ServerRequests.

wip
Titouan Rigoudy 7 years ago
parent
commit
3b055fa9e0
2 changed files with 16 additions and 2 deletions
  1. +4
    -2
      src/proto/peer/message.rs
  2. +12
    -0
      src/proto/server/request.rs

+ 4
- 2
src/proto/peer/message.rs View File

@ -169,13 +169,15 @@ mod tests {
}
#[test]
#[should_panic]
fn invalid_code() {
let mut bytes = BytesMut::new();
ProtoEncoder::new(&mut bytes).encode_u32(1337).unwrap();
let mut cursor = io::Cursor::new(bytes);
Message::decode(&mut ProtoDecoder::new(&mut cursor)).unwrap();
match Message::decode(&mut ProtoDecoder::new(&mut cursor)) {
Err(DecodeError::UnknownCodeError(1337)) => {},
result => panic!(result),
}
}
#[test]


+ 12
- 0
src/proto/server/request.rs View File

@ -603,6 +603,18 @@ mod tests {
assert_eq!(output, input);
}
#[test]
fn invalid_code() {
let mut bytes = BytesMut::new();
ProtoEncoder::new(&mut bytes).encode_u32(1337).unwrap();
let mut cursor = io::Cursor::new(bytes);
match ServerRequest::decode(&mut ProtoDecoder::new(&mut cursor)) {
Err(DecodeError::UnknownCodeError(1337)) => {},
result => panic!(result),
}
}
#[test]
fn roundtrip_cannot_connect_request() {
roundtrip(ServerRequest::CannotConnectRequest(CannotConnectRequest {


Loading…
Cancel
Save