fix(test): wrong message size

hyperium · Feb 16, 2023 · e4ccacb · e4ccacb
1 parent d667950
commit e4ccacb
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/tonic/src/codec/encode.rs b/tonic/src/codec/encode.rs
@@ -141,7 +141,6 @@ fn finish_encoding(
     buf: &mut BytesMut,
 ) -> Result<Bytes, Status> {
     let len = buf.len() - HEADER_SIZE;
-
     let limit = max_message_size.unwrap_or(DEFAULT_MAX_MESSAGE_SIZE);
     if len > limit {
         return Err(Status::new(

diff --git a/tonic/src/codec/prost.rs b/tonic/src/codec/prost.rs
@@ -129,7 +129,7 @@ mod tests {
 
         buf.put(&msg[..]);
 
-        let body = body::MockBody::new(&buf[..], 10005, 0);
+        let body = body::MockBody::new(&buf[..], MAX_MESSAGE_SIZE + HEADER_SIZE + 1, 0);
 
         let mut stream = Streaming::new_request(decoder, body, None, Some(MAX_MESSAGE_SIZE));