Fix Prost codec tests and cfg (#584)

This commit is contained in:
Greg Melton
2021-03-20 18:07:01 +01:00
committed by GitHub
parent 74148549ff
commit 2788628916
+3 -5
View File
@@ -75,11 +75,9 @@ fn from_decode_error(error: prost1::DecodeError) -> crate::Status {
Status::new(Code::Internal, error.to_string()) Status::new(Code::Internal, error.to_string())
} }
#[cfg(tests)] #[cfg(test)]
mod tests { mod tests {
use super::{encode_server, Decoder, Encoder, Streaming}; use crate::codec::{encode_server, DecodeBuf, Decoder, EncodeBuf, Encoder, Streaming};
use crate::codec::buffer::DecodeBuf;
use crate::codec::EncodeBuf;
use crate::Status; use crate::Status;
use bytes::{Buf, BufMut, BytesMut}; use bytes::{Buf, BufMut, BytesMut};
use http_body::Body; use http_body::Body;
@@ -151,7 +149,7 @@ mod tests {
type Error = Status; type Error = Status;
fn decode(&mut self, buf: &mut DecodeBuf<'_>) -> Result<Option<Self::Item>, Self::Error> { fn decode(&mut self, buf: &mut DecodeBuf<'_>) -> Result<Option<Self::Item>, Self::Error> {
let out = Vec::from(buf.bytes()); let out = Vec::from(buf.chunk());
buf.advance(LEN); buf.advance(LEN);
Ok(Some(out)) Ok(Some(out))
} }