TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

network/payload: add missing tests for AddressAndTime

Browse source
This commit is contained in:
Evgenii Stratonikov 2020-12-04 15:25:55 +03:00
parent 6451a4bed3
commit 19a8ccbdb8
1 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
pkg/network/payload/address_test.go
View file

@ -72,3 +72,22 @@ func TestEncodeDecodeBadAddressList(t *testing.T) {
err = testserdes.DecodeBinary(bin, newAL)
require.Error(t, err)
}
func TestGetTCPAddress(t *testing.T) {
t.Run("bad, no capability", func(t *testing.T) {
p := &AddressAndTime{}
copy(p.IP[:], net.IPv4(1, 1, 1, 1))
p.Capabilities = append(p.Capabilities, capability.Capability{
Type: capability.TCPServer,
Data: &capability.Server{Port: 123},
})
s, err := p.GetTCPAddress()
require.NoError(t, err)
require.Equal(t, "1.1.1.1:123", s)
})
t.Run("bad, no capability", func(t *testing.T) {
p := &AddressAndTime{}
s, err := p.GetTCPAddress()
fmt.Println(s, err)
})
}