diff options
author | Christopher Koch <chrisko@google.com> | 2019-04-06 15:50:01 -0700 |
---|---|---|
committer | insomniac <insomniacslk@users.noreply.github.com> | 2019-04-07 23:42:56 +0100 |
commit | b88ad4b870e4c440db756d101fe448f47a487dd8 (patch) | |
tree | a5f6ba6e1faac7ef3507a4596ac5b7b618e3e862 /dhcpv6/server6/server_test.go | |
parent | 73a6b7ffe80ca04462acde3883dbd811f3ab5461 (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.go | 11 |
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 } |