Fixed test for errors

99designs · Nov 12, 2021 · 02ea591 · 02ea591
1 parent 590bbd4
commit 02ea591
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/plugin/federation/federation_entityresolver_test.go b/plugin/federation/federation_entityresolver_test.go
@@ -88,8 +88,13 @@ func TestEntityResolver(t *testing.T) {
 		entityErrors, err := getEntityErrors(err)
 		require.NoError(t, err)
 		require.Len(t, entityErrors, 2)
-		require.Equal(t, entityErrors[0].Message, "error resolving HelloWithErrorsByName. empty key")
-		require.Equal(t, entityErrors[1].Message, "error resolving HelloWithErrorsByName")
+		errMessages := []string{
+			entityErrors[0].Message,
+			entityErrors[1].Message,
+		}
+
+		require.Contains(t, errMessages, "error (empty key) resolving HelloWithErrorsByName")
+		require.Contains(t, errMessages, "error resolving HelloWithErrorsByName")
 
 		require.Len(t, resp.Entities, 5)
 		require.Equal(t, resp.Entities[0].Name, "first name - 1")

diff --git a/plugin/federation/testdata/entityresolver/entity.resolvers.go b/plugin/federation/testdata/entityresolver/entity.resolvers.go
@@ -20,7 +20,7 @@ func (r *entityResolver) FindHelloWithErrorsByName(ctx context.Context, name str
 	if name == "inject error" {
 		return nil, fmt.Errorf("error resolving HelloWithErrorsByName")
 	} else if name == "" {
-		return nil, fmt.Errorf("error resolving HelloWithErrorsByName. empty key")
+		return nil, fmt.Errorf("error (empty key) resolving HelloWithErrorsByName")
 	}
 
 	return &generated.HelloWithErrors{