f_basic_test.go

package tc

import (
	"errors"
	"testing"

	"github.com/google/go-cmp/cmp"
)

func TestBasic(t *testing.T) {
	actions := []*Action{
		{Kind: "csum", CSum: &Csum{Parms: &CsumParms{Index: 4, Capab: 5}}},
	}

	tests := map[string]struct {
		val  Basic
		err1 error
		err2 error
	}{
		"simple": {val: Basic{ClassID: uint32Ptr(2)}},
		"with Action": {
			val: Basic{ClassID: uint32Ptr(73), Actions: &actions},
		},
		"with ematch & police": {
			val: Basic{
				ClassID: uint32Ptr(3),
				Ematch: &Ematch{
					Hdr: &EmatchTreeHdr{NMatches: 1},
					Matches: &[]EmatchMatch{
						{
							Hdr: EmatchHdr{MatchID: 0x0, Kind: EmatchU32, Flags: 0x0, Pad: 0x0},
							// match 'u32(u16 0x1122 0xffff at nexthdr+4)'
							U32Match: &U32Match{
								Mask:    0xffff,
								Value:   0x1122,
								Off:     0x400,
								OffMask: 0xffff,
							},
						},
					},
				},
				Police: &Police{AvRate: uint32Ptr(1337), Result: uint32Ptr(12)},
			},
		},
	}

	for name, testcase := range tests {
		t.Run(name, func(t *testing.T) {
			data, err1 := marshalBasic(&testcase.val)
			if err1 != nil {
				if errors.Is(err1, testcase.err1) {
					return
				}
				t.Fatalf("Unexpected error: %v", err1)
			}
			newData := injectAttribute(t, data, []byte{}, tcaBasicPCNT)
			val := Basic{}
			err2 := unmarshalBasic(newData, &val)
			if err2 != nil {
				if errors.Is(err2, testcase.err2) {
					return
				}
				t.Fatalf("Unexpected error: %v", err2)

			}
			if diff := cmp.Diff(val, testcase.val); diff != "" {
				t.Fatalf("Basic missmatch (want +got):\n%s", diff)
			}
		})
	}
	t.Run("marshal(nil)", func(t *testing.T) {
		_, err := marshalBasic(nil)
		if !errors.Is(err, ErrNoArg) {
			t.Fatalf("unexpected error: %v", err)
		}
	})
	t.Run("unmarshal(0x0)", func(t *testing.T) {
		val := Basic{}
		if err := unmarshalBasic([]byte{0x00}, &val); err == nil {
			t.Fatalf("expected error but got nil")
		}
	})
}