aboutsummaryrefslogtreecommitdiffhomepage
path: root/internal/netlink/netlink_test.go
blob: 0926e3bdcbbf33251d823c4b550cb10634dfb8ca (plain)
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
package netlink

import (
	"os"
	"syscall"
	"testing"
)

func init() { nlPidOnce.Do(func() {}); nlPid = 1 }

type payloadTestCase struct {
	name string
	f    func(c *conn)
	want []byte
}

// checkPayload runs multiple payloadTestCase against a stub conn and checks
// the outgoing message written to its buffer page.
func checkPayload(t *testing.T, testCases []payloadTestCase) {
	t.Helper()

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()

			c := conn{
				pos: syscall.NLMSG_HDRLEN,
				buf: make([]byte, os.Getpagesize()),
			}
			tc.f(&c)
			if got := c.pending(); string(got) != string(tc.want) {
				t.Errorf("pending: %#v, want %#v", got, tc.want)
			}
		})
	}
}