Function TestDecode_CustomMonitorDecoder

pkg/hubble/parser/threefour/parser_test.go:2464–2656

(t *testing.T)

}

func TestDecode_CustomMonitorDecoder(t *testing.T) {
	getTemplate := func() *flowpb.Flow {
		template := &flowpb.Flow{
			EventType:   &flowpb.CiliumEventType{Type: 5},
			Summary:     flowpb.IPVersion_IPv4.String(),
			Type:        flowpb.FlowType_L3_L4,
			Source:      &flowpb.Endpoint{},
			Destination: &flowpb.Endpoint{},
			Ethernet: &flowpb.Ethernet{
				Source:      srcMAC.String(),
				Destination: dstMAC.String(),
			},
			IP: &flowpb.IP{
				IpVersion:   flowpb.IPVersion_IPv4,
				Source:      localIP.String(),
				Destination: remoteIP.String(),
			},
		}
		return template
	}

	testCases := []struct {
		name    string
		opts    []options.Option
		event   any
		ipTuple ipTuple
		want    *flowpb.Flow
	}{
		{
			name: "policy",
			event: monitor.PolicyVerdictNotify{
				Type:        byte(monitorAPI.MessageTypePolicyVerdict),
				Source:      localEP,
				Flags:       monitorAPI.PolicyEgress,
				RemoteLabel: identity.NumericIdentity(remoteID),
			},
			opts: []options.Option{
				options.WithPolicyVerdictNotifyDecoder(func(data []byte, decoded *flowpb.Flow) (*monitor.PolicyVerdictNotify, error) {
					decoded.Uuid = "foobar"
					return &monitor.PolicyVerdictNotify{
						Type:        byte(monitorAPI.MessageTypePolicyVerdict),
						Source:      1226,
						Flags:       monitorAPI.PolicyEgress,
						RemoteLabel: identity.NumericIdentity(1337),
					}, nil
				}),
			},
			ipTuple: egressTuple,
			want: &flowpb.Flow{
				Source:           &flowpb.Endpoint{ID: 1234},
				Destination:      &flowpb.Endpoint{Identity: 1337},
				TrafficDirection: flowpb.TrafficDirection_EGRESS,
				Uuid:             "foobar",
				IsReply:          wrapperspb.Bool(false),
				Verdict:          flowpb.Verdict_FORWARDED,
			},
		},
		{