golang__protobuf/internal/mapsort/mapsort_test.go

// Copyright 2019 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package mapsort_test

import (
	"strconv"
	"testing"

	"google.golang.org/protobuf/internal/mapsort"
	testpb "google.golang.org/protobuf/internal/testprotos/test"
	pref "google.golang.org/protobuf/reflect/protoreflect"
)

func TestRange(t *testing.T) {
	m := (&testpb.TestAllTypes{
		MapBoolBool: map[bool]bool{
			false: false,
			true:  true,
		},
		MapInt32Int32: map[int32]int32{
			0: 0,
			1: 1,
			2: 2,
		},
		MapUint64Uint64: map[uint64]uint64{
			0: 0,
			1: 1,
			2: 2,
		},
		MapStringString: map[string]string{
			"0": "0",
			"1": "1",
			"2": "2",
		},
	}).ProtoReflect()
	m.Range(func(fd pref.FieldDescriptor, v pref.Value) bool {
		mapv := v.Map()
		var got []pref.MapKey
		mapsort.Range(mapv, fd.MapKey().Kind(), func(key pref.MapKey, _ pref.Value) bool {
			got = append(got, key)
			return true
		})
		for wanti, key := range got {
			var goti int
			switch x := mapv.Get(key).Interface().(type) {
			case bool:
				if x {
					goti = 1
				}
			case int32:
				goti = int(x)
			case uint64:
				goti = int(x)
			case string:
				goti, _ = strconv.Atoi(x)
			default:
				t.Fatalf("unhandled map value type %T", x)
			}
			if wanti != goti {
				t.Errorf("out of order range over map field %v: %v", fd.FullName(), got)
				break
			}
		}
		return true
	})
}