Skip to content

Commit fe8b6f5

Browse files
committed
backend_req::tests: Extract test setup to a function
Extract creating a backend and a frontend from a socketpair to a utility function. Signed-off-by: Matej Hrica <[email protected]>
1 parent e4d9e23 commit fe8b6f5

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

vhost/src/vhost_user/backend_req.rs

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -191,10 +191,16 @@ mod tests {
191191

192192
use super::*;
193193

194+
fn frontend_backend_pair() -> (Endpoint<VhostUserMsgHeader<BackendReq>>, Backend) {
195+
let (p1, p2) = UnixStream::pair().unwrap();
196+
let backend = Backend::from_stream(p1);
197+
let frontend = Endpoint::<VhostUserMsgHeader<BackendReq>>::from_stream(p2);
198+
(frontend, backend)
199+
}
200+
194201
#[test]
195202
fn test_backend_req_set_failed() {
196-
let (p1, _p2) = UnixStream::pair().unwrap();
197-
let backend = Backend::from_stream(p1);
203+
let (_, backend) = frontend_backend_pair();
198204

199205
assert!(backend.node().error.is_none());
200206
backend.set_failed(libc::EAGAIN);
@@ -203,8 +209,7 @@ mod tests {
203209

204210
#[test]
205211
fn test_backend_req_send_failure() {
206-
let (p1, _) = UnixStream::pair().unwrap();
207-
let backend = Backend::from_stream(p1);
212+
let (_, backend) = frontend_backend_pair();
208213

209214
backend.set_failed(libc::ECONNRESET);
210215
backend
@@ -218,9 +223,7 @@ mod tests {
218223

219224
#[test]
220225
fn test_backend_req_recv_negative() {
221-
let (p1, p2) = UnixStream::pair().unwrap();
222-
let backend = Backend::from_stream(p1);
223-
let mut frontend = Endpoint::<VhostUserMsgHeader<BackendReq>>::from_stream(p2);
226+
let (mut frontend, backend) = frontend_backend_pair();
224227

225228
let len = mem::size_of::<VhostUserSharedMsg>();
226229
let mut hdr = VhostUserMsgHeader::new(

0 commit comments

Comments
 (0)