Struct MockDatagramSocket 

pub struct MockDatagramSocket { /* private fields */ }

Available on crate features mock and net only.

A test-only DatagramSocket backed by in-memory queues — exchange datagrams with a MessageDatagram in unit tests, no real socket bound. Enabled by the mock feature (put it in your [dev-dependencies]). Queue inbound datagrams with push_inbound (each is one recv_from) and inspect what was sent with sent.

use bnb::{bin, MessageDatagram, MockDatagramSocket};
#[bin(big)]
#[derive(Debug, PartialEq, Eq)]
struct Ping {
    seq: u16,
}

let mut peer = MessageDatagram::new(MockDatagramSocket::new());
let from = "127.0.0.1:5000".parse().unwrap();
peer.get_ref().push_inbound(&Ping { seq: 7 }.to_bytes().unwrap(), from); // as if it arrived

let (ping, who): (Ping, _) = peer.recv_message().unwrap();
assert_eq!(ping, Ping { seq: 7 });
peer.send_message(&Ping { seq: 8 }, &who).unwrap(); // reply to the sender
assert_eq!(peer.get_ref().sent()[0].0, Ping { seq: 8 }.to_bytes().unwrap());

Implementations

impl MockDatagramSocket

pub fn new() -> Self

An empty mock with no queued datagrams.

pub fn push_inbound(&self, bytes: &[u8], from: SocketAddr)

Queue one datagram (bytes, from from) to be returned by the next recv_from.

pub fn sent(&self) -> Vec<(Vec<u8>, SocketAddr)>

Every datagram sent so far, as (bytes, destination), in send order.

pub fn fail_next_recv(self) -> Self

Make the next recv_from fail with ConnectionReset instead of returning a datagram — to test recv-error handling. One-shot: later recvs behave normally.

Trait Implementations

impl DatagramSocket for MockDatagramSocket

type Addr = SocketAddr

The peer-address type (SocketAddr for UDP; std::os::unix::net::SocketAddr for Unix).

fn recv_from(&self, buf: &mut [u8]) -> Result<(usize, SocketAddr)>

Receive one datagram into buf, returning how many bytes it held and who sent it.

fn send_to(&self, buf: &[u8], addr: &SocketAddr) -> Result<usize>

Send buf as one datagram to addr, returning the bytes sent.

impl Debug for MockDatagramSocket

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for MockDatagramSocket

fn default() -> MockDatagramSocket

Returns the “default value” for a type.