1*1c12ee1eSDan Willemsen// Copyright 2020 The Go Authors. All rights reserved. 2*1c12ee1eSDan Willemsen// Use of this source code is governed by a BSD-style 3*1c12ee1eSDan Willemsen// license that can be found in the LICENSE file. 4*1c12ee1eSDan Willemsen 5*1c12ee1eSDan Willemsenpackage prototest 6*1c12ee1eSDan Willemsen 7*1c12ee1eSDan Willemsenimport ( 8*1c12ee1eSDan Willemsen "testing" 9*1c12ee1eSDan Willemsen 10*1c12ee1eSDan Willemsen "google.golang.org/protobuf/reflect/protoreflect" 11*1c12ee1eSDan Willemsen) 12*1c12ee1eSDan Willemsen 13*1c12ee1eSDan Willemsen// Enum tests an EnumType implementation. 14*1c12ee1eSDan Willemsentype Enum struct{} 15*1c12ee1eSDan Willemsen 16*1c12ee1eSDan Willemsenfunc (test Enum) Test(t testing.TB, et protoreflect.EnumType) { 17*1c12ee1eSDan Willemsen ed := et.Descriptor() 18*1c12ee1eSDan Willemsen values := ed.Values() 19*1c12ee1eSDan Willemsen for i := 0; i < values.Len(); i++ { 20*1c12ee1eSDan Willemsen evd := values.Get(i) 21*1c12ee1eSDan Willemsen num := evd.Number() 22*1c12ee1eSDan Willemsen e := et.New(num) 23*1c12ee1eSDan Willemsen if e.Descriptor() != ed { 24*1c12ee1eSDan Willemsen t.Errorf("enumType.New(%v).Descriptor() != enumType.Descriptor(), should match", num) 25*1c12ee1eSDan Willemsen } 26*1c12ee1eSDan Willemsen if e.Type() != et { 27*1c12ee1eSDan Willemsen t.Errorf("enumType.New(%v).Type() != enumType, should match", num) 28*1c12ee1eSDan Willemsen } 29*1c12ee1eSDan Willemsen if got, want := e.Number(), num; got != want { 30*1c12ee1eSDan Willemsen t.Errorf("enumType.New(%v).Number() = %v, want %v", num, got, want) 31*1c12ee1eSDan Willemsen } 32*1c12ee1eSDan Willemsen } 33*1c12ee1eSDan Willemsen} 34