|
此文章由 DDD888 原创或转贴,不代表本站立场和观点,版权归 oursteps.com.au 和作者 DDD888 所有!转贴必须注明作者、出处和本声明,并保持内容完整
本帖最后由 DDD888 于 2021-8-2 14:01 编辑
我自己来回答自己的问题
use actix_web::body::{Body, ResponseBody};
pub trait BodyTest {
fn as_str(&self) -> &str;
}
impl BodyTest for ResponseBody<Body> {
fn as_str(&self) -> &str {
match self {
ResponseBody::Body(ref b) => match b {
Body::Bytes(ref by) => std::str::from_utf8(&by).unwrap(),
_ => panic!(),
},
ResponseBody::Other(ref b) => match b {
Body::Bytes(ref by) => std::str::from_utf8(&by).unwrap(),
_ => panic!(),
},
}
}
}
#[cfg(test)]
mod tests {
use super::*;
use actix_web::body::Body;
#[test]
fn body_empty_expect_empty() {
let response_body_empty = ResponseBody::Body(Body::from(""));
assert_eq!(response_body_empty.as_str(), "");
}
#[test]
fn body_not_empty_expect_hello() {
let response_body = ResponseBody::Body(Body::from("hello"));
assert_eq!(response_body.as_str(), "hello");
}
#[test]
fn body_other_empty_expect_empty() {
let response_body = ResponseBody::Other(Body::from(""));
assert_eq!(response_body.as_str(), "");
}
#[test]
fn body_other_not_empty_expect_hello() {
let response_body = ResponseBody::Other(Body::from("hello"));
assert_eq!(response_body.as_str(), "hello");
}
} |
|