Browse Source

add IPv4 packet addr encoding test

Shelikhoo 4 years ago
parent
commit
16bff33858
1 changed files with 21 additions and 0 deletions
  1. 21 0
      common/net/packetaddr/packetaddr_test.go

+ 21 - 0
common/net/packetaddr/packetaddr_test.go

@@ -0,0 +1,21 @@
+package packetaddr
+
+import (
+	"github.com/stretchr/testify/assert"
+	sysnet "net"
+	"testing"
+)
+
+func TestPacketEncodingIPv4(t *testing.T) {
+	packetAddress := &sysnet.UDPAddr{
+		IP:   sysnet.IPv4(1, 2, 3, 4).To4(),
+		Port: 1234,
+	}
+	var packetData [256]byte
+	wrapped := AttachAddressToPacket(packetData[:], packetAddress)
+
+	packetPayload, decodedAddress := ExtractAddressFromPacket(wrapped)
+
+	assert.Equal(t, packetPayload, packetData[:])
+	assert.Equal(t, packetAddress, decodedAddress)
+}