feat(errorsx): add error wrapper for quic (#422)

Part of https://github.com/ooni/probe/issues/1505

internal/netxmocks/conn.go

@@ -32,7 +32,7 @@ func (c *Conn) Close() error {
 	return c.MockClose()
 }
 
-// LocalAddr class MockLocalAddr.
+// LocalAddr calls MockLocalAddr.
 func (c *Conn) LocalAddr() net.Addr {
 	return c.MockLocalAddr()
 }

internal/netxmocks/quic.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"crypto/tls"
 	"net"
+	"time"
 
 	"github.com/lucas-clemente/quic-go"
 	"github.com/ooni/probe-cli/v3/internal/quicx"
@@ -127,3 +128,63 @@ func (s *QUICEarlySession) SendMessage(b []byte) error {
 func (s *QUICEarlySession) ReceiveMessage() ([]byte, error) {
 	return s.MockReceiveMessage()
 }
+
+// QUICUDPConn is an UDP conn used by QUIC.
+type QUICUDPConn struct {
+	MockWriteTo          func(p []byte, addr net.Addr) (int, error)
+	MockReadMsgUDP       func(b, oob []byte) (int, int, int, *net.UDPAddr, error)
+	MockClose            func() error
+	MockLocalAddr        func() net.Addr
+	MockRemoteAddr       func() net.Addr
+	MockSetDeadline      func(t time.Time) error
+	MockSetReadDeadline  func(t time.Time) error
+	MockSetWriteDeadline func(t time.Time) error
+	MockReadFrom         func(p []byte) (n int, addr net.Addr, err error)
+}
+
+var _ net.PacketConn = &QUICUDPConn{}
+
+// WriteTo calls MockWriteTo.
+func (c *QUICUDPConn) WriteTo(p []byte, addr net.Addr) (int, error) {
+	return c.MockWriteTo(p, addr)
+}
+
+// ReadMsgUDP calls MockReadMsgUDP.
+func (c *QUICUDPConn) ReadMsgUDP(b, oob []byte) (int, int, int, *net.UDPAddr, error) {
+	return c.MockReadMsgUDP(b, oob)
+}
+
+// Close calls MockClose.
+func (c *QUICUDPConn) Close() error {
+	return c.MockClose()
+}
+
+// LocalAddr calls MockLocalAddr.
+func (c *QUICUDPConn) LocalAddr() net.Addr {
+	return c.MockLocalAddr()
+}
+
+// RemoteAddr calls MockRemoteAddr.
+func (c *QUICUDPConn) RemoteAddr() net.Addr {
+	return c.MockRemoteAddr()
+}
+
+// SetDeadline calls MockSetDeadline.
+func (c *QUICUDPConn) SetDeadline(t time.Time) error {
+	return c.MockSetDeadline(t)
+}
+
+// SetReadDeadline calls MockSetReadDeadline.
+func (c *QUICUDPConn) SetReadDeadline(t time.Time) error {
+	return c.MockSetReadDeadline(t)
+}
+
+// SetWriteDeadline calls MockSetWriteDeadline.
+func (c *QUICUDPConn) SetWriteDeadline(t time.Time) error {
+	return c.MockSetWriteDeadline(t)
+}
+
+// ReadFrom calls MockReadFrom.
+func (c *QUICUDPConn) ReadFrom(b []byte) (int, net.Addr, error) {
+	return c.MockReadFrom(b)
+}

internal/netxmocks/quic_test.go

@@ -7,7 +7,9 @@ import (
 	"net"
 	"reflect"
 	"testing"
+	"time"
 
+	"github.com/google/go-cmp/cmp"
 	"github.com/lucas-clemente/quic-go"
 	"github.com/ooni/probe-cli/v3/internal/quicx"
 )
@@ -266,3 +268,152 @@ func TestQUICEarlySessionReceiveMessage(t *testing.T) {
 		t.Fatal("expected nil buffer here")
 	}
 }
+
+func TestQUICUDPConnWriteTo(t *testing.T) {
+	expected := errors.New("mocked error")
+	quc := &QUICUDPConn{
+		MockWriteTo: func(p []byte, addr net.Addr) (int, error) {
+			return 0, expected
+		},
+	}
+	pkt := make([]byte, 128)
+	addr := &net.UDPAddr{}
+	cnt, err := quc.WriteTo(pkt, addr)
+	if !errors.Is(err, expected) {
+		t.Fatal("not the error we expected", err)
+	}
+	if cnt != 0 {
+		t.Fatal("expected zero here")
+	}
+}
+
+func TestQUICUDPConnReadMsgUDP(t *testing.T) {
+	expected := errors.New("mocked error")
+	quc := &QUICUDPConn{
+		MockReadMsgUDP: func(b, oob []byte) (int, int, int, *net.UDPAddr, error) {
+			return 0, 0, 0, nil, expected
+		},
+	}
+	b := make([]byte, 128)
+	oob := make([]byte, 128)
+	n, oobn, flags, addr, err := quc.ReadMsgUDP(b, oob)
+	if !errors.Is(err, expected) {
+		t.Fatal("not the error we expected", err)
+	}
+	if n != 0 {
+		t.Fatal("expected zero here")
+	}
+	if oobn != 0 {
+		t.Fatal("expected zero here")
+	}
+	if flags != 0 {
+		t.Fatal("expected zero here")
+	}
+	if addr != nil {
+		t.Fatal("expected nil here")
+	}
+}
+
+func TestQUICUDPConnClose(t *testing.T) {
+	expected := errors.New("mocked error")
+	quc := &QUICUDPConn{
+		MockClose: func() error {
+			return expected
+		},
+	}
+	err := quc.Close()
+	if !errors.Is(err, expected) {
+		t.Fatal("not the error we expected", err)
+	}
+}
+
+func TestQUICUDPConnLocalAddrWorks(t *testing.T) {
+	expected := &net.TCPAddr{
+		IP:   net.IPv6loopback,
+		Port: 1234,
+	}
+	c := &QUICUDPConn{
+		MockLocalAddr: func() net.Addr {
+			return expected
+		},
+	}
+	out := c.LocalAddr()
+	if diff := cmp.Diff(expected, out); diff != "" {
+		t.Fatal(diff)
+	}
+}
+
+func TestQUICUDPConnRemoteAddrWorks(t *testing.T) {
+	expected := &net.TCPAddr{
+		IP:   net.IPv6loopback,
+		Port: 1234,
+	}
+	c := &QUICUDPConn{
+		MockRemoteAddr: func() net.Addr {
+			return expected
+		},
+	}
+	out := c.RemoteAddr()
+	if diff := cmp.Diff(expected, out); diff != "" {
+		t.Fatal(diff)
+	}
+}
+
+func TestQUICUDPConnSetDeadline(t *testing.T) {
+	expected := errors.New("mocked error")
+	c := &QUICUDPConn{
+		MockSetDeadline: func(t time.Time) error {
+			return expected
+		},
+	}
+	err := c.SetDeadline(time.Time{})
+	if !errors.Is(err, expected) {
+		t.Fatal("not the error we expected", err)
+	}
+}
+
+func TestQUICUDPConnSetReadDeadline(t *testing.T) {
+	expected := errors.New("mocked error")
+	c := &QUICUDPConn{
+		MockSetReadDeadline: func(t time.Time) error {
+			return expected
+		},
+	}
+	err := c.SetReadDeadline(time.Time{})
+	if !errors.Is(err, expected) {
+		t.Fatal("not the error we expected", err)
+	}
+}
+
+func TestQUICUDPConnSetWriteDeadline(t *testing.T) {
+	expected := errors.New("mocked error")
+	c := &QUICUDPConn{
+		MockSetWriteDeadline: func(t time.Time) error {
+			return expected
+		},
+	}
+	err := c.SetWriteDeadline(time.Time{})
+	if !errors.Is(err, expected) {
+		t.Fatal("not the error we expected", err)
+	}
+}
+
+func TestQUICUDPConnReadFrom(t *testing.T) {
+	expected := errors.New("mocked error")
+	quc := &QUICUDPConn{
+		MockReadFrom: func(b []byte) (int, net.Addr, error) {
+			return 0, nil, expected
+		},
+	}
+	b := make([]byte, 128)
+	n, addr, err := quc.ReadFrom(b)
+	if !errors.Is(err, expected) {
+		t.Fatal("not the error we expected", err)
+	}
+	if n != 0 {
+		t.Fatal("expected zero here")
+	}
+	if addr != nil {
+		t.Fatal("expected nil here")
+	}
+}