File: address_v4_test.go

package info (click to toggle)
golang-github-kentik-patricia 1.2.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 3,860 kB
  • sloc: makefile: 54
file content (50 lines) | stat: -rw-r--r-- 1,568 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package patricia

import (
	"fmt"
	"net"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestNewIPv4Address(t *testing.T) {
	// 4 byte format
	sut := NewIPv4Address(uint32(0x01234567), 7)

	assert.Equal(t, uint32(0x01234567), sut.Address)
	assert.Equal(t, uint(7), sut.Length)
	assert.Equal(t, "1.35.69.103/7", sut.String())
	assert.Equal(t, "1.35.69.103/7", (&sut).String())
	//nolint
	assert.Equal(t, "1.35.69.103/7", fmt.Sprintf("%s", sut))
	//nolint
	assert.Equal(t, "1.35.69.103/7", fmt.Sprintf("%s", &sut))
}

func TestNewIPv4AddressFromBytes(t *testing.T) {
	// 4-byte format - from bytes
	sut := NewIPv4AddressFromBytes([]byte{3, 4, 5, 6}, 31)
	assert.Equal(t, uint32(0x03040506), sut.Address)
	assert.Equal(t, uint(31), sut.Length)

	// 16-byte format - from net.IPV4
	sut = NewIPv4AddressFromBytes(net.IPv4(3, 4, 5, 6), 31)
	assert.Equal(t, uint32(0x03040506), sut.Address)
	assert.Equal(t, uint(31), sut.Length)

	// 16-byte format - directly from bytes
	sut = NewIPv4AddressFromBytes([]byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 255, 255, 3, 4, 5, 6}, 31)
	assert.Equal(t, uint32(0x03040506), sut.Address)
	assert.Equal(t, uint(31), sut.Length)

	// 3-byte invalid format - directly from bytes
	sut = NewIPv4AddressFromBytes([]byte{4, 5, 6}, 31)
	assert.Equal(t, uint32(0), sut.Address)
	assert.Equal(t, uint(0), sut.Length)

	// 17-byte invalid format - directly from bytes
	sut = NewIPv4AddressFromBytes([]byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 255, 255, 3, 4, 5, 6}, 31)
	assert.Equal(t, uint32(0), sut.Address)
	assert.Equal(t, uint(0), sut.Length)
}