summaryrefslogtreecommitdiffhomepage
path: root/dhcpv6/server6/server_test.go
diff options
context:
space:
mode:
authorChristopher Koch <chrisko@google.com>2019-04-06 15:50:01 -0700
committerinsomniac <insomniacslk@users.noreply.github.com>2019-04-07 23:42:56 +0100
commitb88ad4b870e4c440db756d101fe448f47a487dd8 (patch)
treea5f6ba6e1faac7ef3507a4596ac5b7b618e3e862 /dhcpv6/server6/server_test.go
parent73a6b7ffe80ca04462acde3883dbd811f3ab5461 (diff)
nclient6: small fixes
- RapidCommit solicits wait for Reply messages, not Advertise. - Default recipient should be all relay agents and servers, not just servers. - Make New() and NewWithConn() interface same as in nclient4. Signed-off-by: Christopher Koch <chrisko@google.com>
Diffstat (limited to 'dhcpv6/server6/server_test.go')
-rw-r--r--dhcpv6/server6/server_test.go11
1 files changed, 5 insertions, 6 deletions
diff --git a/dhcpv6/server6/server_test.go b/dhcpv6/server6/server_test.go
index 05d62cb..3b204f1 100644
--- a/dhcpv6/server6/server_test.go
+++ b/dhcpv6/server6/server_test.go
@@ -12,15 +12,16 @@ import (
"github.com/stretchr/testify/require"
)
-type fakeUnconnectedConn struct {
+// Turns a connected UDP conn into an "unconnected" UDP conn.
+type unconnectedConn struct {
*net.UDPConn
}
-func (f fakeUnconnectedConn) WriteTo(b []byte, _ net.Addr) (int, error) {
+func (f unconnectedConn) WriteTo(b []byte, _ net.Addr) (int, error) {
return f.UDPConn.Write(b)
}
-func (f fakeUnconnectedConn) ReadFrom(b []byte) (int, net.Addr, error) {
+func (f unconnectedConn) ReadFrom(b []byte) (int, net.Addr, error) {
n, err := f.Read(b)
return n, nil, err
}
@@ -43,12 +44,10 @@ func setUpClientAndServer(handler Handler) (*nclient6.Client, *Server) {
panic(err)
}
- c, err := nclient6.New(net.HardwareAddr{1, 2, 3, 4, 5, 6},
- nclient6.WithConn(fakeUnconnectedConn{clientConn}))
+ c, err := nclient6.NewWithConn(unconnectedConn{clientConn}, net.HardwareAddr{1, 2, 3, 4, 5, 6})
if err != nil {
panic(err)
}
-
return c, s
}