prevent nil wrapping

Ilhasoft · Jan 20, 2020 · 19ef106 · 19ef106
1 parent 8e8b4ba
commit 19ef106
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/search/search_test.go b/search/search_test.go
@@ -55,7 +55,11 @@ func TestElasticSort(t *testing.T) {
 	}
 
 	resolver := func(key string) assets.Field {
-		return registry[key]
+		field, found := registry[key]
+		if !found {
+			return nil
+		}
+		return field
 	}
 
 	for _, tc := range tcs {
@@ -100,7 +104,11 @@ func TestElasticQuery(t *testing.T) {
 	ny, _ := time.LoadLocation("America/New_York")
 
 	resolver := func(key string) assets.Field {
-		return registry[key]
+		field, found := registry[key]
+		if !found {
+			return nil
+		}
+		return field
 	}
 
 	for _, tc := range tcs {