aboutsummaryrefslogtreecommitdiffhomepage
path: root/internal/netlink/netlink_test.go
blob: 9db9fca5a078b39ace733e9ffa12d522965bd81b (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
package netlink

import (
	"syscall"
	"testing"
)

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()
			t.Helper()

			c := Conn{port: 1, pos: syscall.NLMSG_HDRLEN}
			tc.f(&c)
			if got := c.pending(); string(got) != string(tc.want) {
				t.Errorf("pending: %#v, want %#v", got, tc.want)
			}
		})
	}
}