Fix up tests

This commit is contained in:
2025-06-30 20:11:54 +10:00
parent f2a13df169
commit 81cd901ea6
3 changed files with 10 additions and 14 deletions

View File

@@ -242,7 +242,7 @@ impl lb_tcpstream {
}
fn read_half<'s>(&'s self) -> Result<RefMut<'s, OwnedReadHalf>> {
let read = self.read.borrow_mut();
let read = self.read.try_borrow_mut()?;
match *read {
Some(_) => Ok(RefMut::map(read, |s| s.as_mut().unwrap())),
None => Err(Error::SocketClosed),
@@ -250,7 +250,7 @@ impl lb_tcpstream {
}
fn write_half<'s>(&'s self) -> Result<RefMut<'s, OwnedWriteHalf>> {
let write = self.write.borrow_mut();
let write = self.write.try_borrow_mut()?;
match *write {
Some(_) => Ok(RefMut::map(write, |s| s.as_mut().unwrap())),
None => Err(Error::SocketClosed),