Function TestGroupE2E

internal/hierarchy/hierarchy_ext_test.go:136–174

(t *testing.T)

}

func TestGroupE2E(t *testing.T) {
	testHierarchy := map[string]map[string][]string{
		"p0": {
			"wt0": {"addr0", "addr1"},
			"wt1": {"addr2", "addr3"},
		},
		"p1": {
			"wt10": {"addr10", "addr11"},
			"wt11": {"addr12", "addr13"},
		},
	}

	var epsWithHierarchy []resolver.Endpoint
	for p, wts := range testHierarchy {
		path1 := []string{p}
		for wt, addrs := range wts {
			path2 := append([]string(nil), path1...)
			path2 = append(path2, wt)
			for _, addr := range addrs {
				a := hierarchy.SetInEndpoint(resolver.Endpoint{Addresses: []resolver.Address{{Addr: addr}}}, path2)
				epsWithHierarchy = append(epsWithHierarchy, a)
			}
		}
	}

	gotHierarchy := make(map[string]map[string][]string)
	for p1, wts := range hierarchy.Group(epsWithHierarchy) {
		gotHierarchy[p1] = make(map[string][]string)
		for p2, eps := range hierarchy.Group(wts) {
			for _, ep := range eps {
				gotHierarchy[p1][p2] = append(gotHierarchy[p1][p2], ep.Addresses[0].Addr)
			}
		}
	}

	if !cmp.Equal(gotHierarchy, testHierarchy) {
		t.Errorf("diff: %v", cmp.Diff(gotHierarchy, testHierarchy))
	}
}