File: match_addrtype_test.go

package info (click to toggle)
golang-github-google-nftables 0.2.0-3
  • links: PTS, VCS
  • area: main
  • in suites: experimental, forky, sid, trixie
  • size: 756 kB
  • sloc: makefile: 8
file content (71 lines) | stat: -rw-r--r-- 1,380 bytes parent folder | download | duplicates (3)
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
package xt

import (
	"reflect"
	"testing"
)

func TestTargetAddrType(t *testing.T) {
	t.Parallel()
	tests := []struct {
		name  string
		fam   byte
		rev   uint32
		info  InfoAny
		empty InfoAny
	}{
		{
			name: "un/marshal AddrType Rev 0 round-trip",
			fam:  0,
			rev:  0,
			info: &AddrType{
				Source:       0x1234,
				Dest:         0x5678,
				InvertSource: true,
				InvertDest:   false,
			},
			empty: &AddrType{},
		},
		{
			name: "un/marshal AddrType Rev 0 round-trip",
			fam:  0,
			rev:  0,
			info: &AddrType{
				Source:       0x1234,
				Dest:         0x5678,
				InvertSource: false,
				InvertDest:   true,
			},
			empty: &AddrType{},
		},
		{
			name: "un/marshal AddrType Rev 1 round-trip",
			fam:  0,
			rev:  0,
			info: &AddrTypeV1{
				Source: 0x1234,
				Dest:   0x5678,
				Flags:  0xb00f,
			},
			empty: &AddrTypeV1{},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			data, err := tt.info.marshal(TableFamily(tt.fam), tt.rev)
			if err != nil {
				t.Fatalf("marshal error: %+v", err)

			}
			var recoveredInfo InfoAny = tt.empty
			err = recoveredInfo.unmarshal(TableFamily(tt.fam), tt.rev, data)
			if err != nil {
				t.Fatalf("unmarshal error: %+v", err)
			}
			if !reflect.DeepEqual(tt.info, recoveredInfo) {
				t.Fatalf("original %+v and recovered %+v are different", tt.info, recoveredInfo)
			}
		})
	}
}