Function TestMapCollectionWithLookups

pkg/util/util_test.go:65–378

(t *testing.T)

}

func TestMapCollectionWithLookups(t *testing.T) {
	// model: A person works in the Sales Department in the Country of New Zealand
	type Person struct {
		ID           string
		FirstName    string
		LastName     string
		DepartmentID string
		CountryID    string
	}

	type Department struct {
		ID   string
		Name string
	}

	departments := map[string]Department{
		"S": {ID: "S", Name: "Sales"},
		"M": {ID: "M", Name: "Marketing"},
	}

	type Country struct {
		ID   string
		Name string
	}

	countries := map[string]Country{
		"NZ": {ID: "NZ", Name: "New Zealand"},
		"AU": {ID: "AU", Name: "Australia"},
	}

	people := []Person{
		{ID: "1", FirstName: "John", LastName: "Smith", DepartmentID: "S", CountryID: "NZ"},
		{ID: "2", FirstName: "Jane", LastName: "Doe", DepartmentID: "M", CountryID: "NZ"},
		{ID: "3", FirstName: "Alan", LastName: "Walker", DepartmentID: "S", CountryID: "AU"},
	}

	type PersonWithCountryDepartment struct {
		PersonID       string
		Name           string
		DepartmentName string
		CountryName    string
	}

	t.Run("typical with two lookups", func(t *testing.T) {
		caches := util.MapCollectionCacheContainer{}

		results, err := util.MapCollectionWithLookups(
			&caches,
			people,
			func(p Person) []string { return []string{p.DepartmentID, p.CountryID} },
			func(p Person, lookup []string) PersonWithCountryDepartment {
				return PersonWithCountryDepartment{
					PersonID:       p.ID,
					Name:           fmt.Sprintf("%s %s", p.FirstName, p.LastName),
					DepartmentName: lookup[0],
					CountryName:    lookup[1],
				}
			},