use a net.PacketConn instead of a net.UDPConn in Server and Session

conn_test.go

@@ -0,0 +1,65 @@
+package quic
+
+import (
+	"bytes"
+	"net"
+	"time"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+type mockPacketConn struct {
+	dataWritten   bytes.Buffer
+	dataWrittenTo net.Addr
+}
+
+func (c *mockPacketConn) ReadFrom(b []byte) (n int, addr net.Addr, err error) {
+	panic("not implemented")
+}
+func (c *mockPacketConn) WriteTo(b []byte, addr net.Addr) (n int, err error) {
+	c.dataWrittenTo = addr
+	return c.dataWritten.Write(b)
+}
+func (c *mockPacketConn) Close() error                       { panic("not implemented") }
+func (c *mockPacketConn) LocalAddr() net.Addr                { panic("not implemented") }
+func (c *mockPacketConn) SetDeadline(t time.Time) error      { panic("not implemented") }
+func (c *mockPacketConn) SetReadDeadline(t time.Time) error  { panic("not implemented") }
+func (c *mockPacketConn) SetWriteDeadline(t time.Time) error { panic("not implemented") }
+
+var _ net.PacketConn = &mockPacketConn{}
+
+var _ = Describe("Connection", func() {
+	var c *conn
+
+	BeforeEach(func() {
+		addr := &net.UDPAddr{
+			IP:   net.IPv4(192, 168, 100, 200),
+			Port: 1337,
+		}
+		c = &conn{
+			currentAddr: addr,
+			pconn:       &mockPacketConn{},
+		}
+	})
+
+	It("writes", func() {
+		err := c.write([]byte("foobar"))
+		Expect(err).ToNot(HaveOccurred())
+		Expect(c.pconn.(*mockPacketConn).dataWritten.Bytes()).To(Equal([]byte("foobar")))
+		Expect(c.pconn.(*mockPacketConn).dataWrittenTo.String()).To(Equal("192.168.100.200:1337"))
+	})
+
+	It("gets the remote address", func() {
+		Expect(c.RemoteAddr().String()).To(Equal("192.168.100.200:1337"))
+	})
+
+	It("changes the remote address", func() {
+		addr := &net.UDPAddr{
+			IP:   net.IPv4(127, 0, 0, 1),
+			Port: 7331,
+		}
+		c.setCurrentRemoteAddr(addr)
+		Expect(c.RemoteAddr().String()).To(Equal(addr.String()))
+	})
+})