Handle enums in protomap (#1012)

Co-authored-by: Rob Shakir <[email protected]>

Author: DanG100

protomap/integration_tests/integration_test.go

@@ -7,6 +7,7 @@ import (
 	"github.com/google/go-cmp/cmp"
 	"github.com/google/go-cmp/cmp/cmpopts"
 	"github.com/openconfig/gnmi/value"
+	"github.com/openconfig/gribi/v1/proto/gribi_aft/enums"
 	"github.com/openconfig/ygot/protomap"
 	"github.com/openconfig/ygot/testutil"
 	"github.com/openconfig/ygot/ygot"
@@ -91,6 +92,23 @@ func TestGRIBIAFT(t *testing.T) {
 			mustPath("afts/next-hops/next-hop[index=1]/index"):                         uint64(1),
 			mustPath("afts/next-hops/next-hop[index=1]/state/index"):                   uint64(1),
 		},
+	}, {
+		desc: "NH entry with enum",
+		inProto: &aftpb.Afts{
+			NextHop: []*aftpb.Afts_NextHopKey{{
+				Index: 1,
+				NextHop: &aftpb.Afts_NextHop{
+					IpAddress:         &wpb.StringValue{Value: "127.0.0.1"},
+					EncapsulateHeader: enums.OpenconfigAftTypesEncapsulationHeaderType_OPENCONFIGAFTTYPESENCAPSULATIONHEADERTYPE_IPV4,
+				},
+			}},
+		},
+		wantPaths: map[*gpb.Path]interface{}{
+			mustPath("afts/next-hops/next-hop[index=1]/state/ip-address"):         "127.0.0.1",
+			mustPath("afts/next-hops/next-hop[index=1]/index"):                    uint64(1),
+			mustPath("afts/next-hops/next-hop[index=1]/state/index"):              uint64(1),
+			mustPath("afts/next-hops/next-hop[index=1]/state/encapsulate-header"): "IPV4",
+		},
 	}, {
 		desc: "NHG entry",
 		inProto: &aftpb.Afts{

protomap/proto.go

@@ -114,6 +114,13 @@ func parseField(fd protoreflect.FieldDescriptor, v protoreflect.Value, vals map[
 		}
 	}
 
+	// Handle enum values by using the enum value option extension to convert value to string.
+	if fd.Kind() == protoreflect.EnumKind {
+		po := fd.Enum().Values().ByNumber(val.(protoreflect.EnumNumber)).Options().(*descriptorpb.EnumValueOptions)
+		ex := proto.GetExtension(po, yextpb.E_YangName).(string)
+		val = ex
+	}
+
 	// Handle cases where there is >1 path specified for a field based on
 	// path compression.
 	for _, path := range annotatedPath {