Handle nil results for entity queries

DivvyPayHQ · Jun 4, 2024 · b0d6da2 · b0d6da2
1 parent 477aa09
commit b0d6da2
Show file tree

Hide file tree

Showing 3 changed files with 69 additions and 3 deletions.
diff --git a/lib/absinthe/federation/schema/entities_field.ex b/lib/absinthe/federation/schema/entities_field.ex
@@ -242,9 +242,9 @@ defmodule Absinthe.Federation.Schema.EntitiesField do
     }
 
   defp reduce_resolution(%{state: :resolved} = res) do
-    case res.value do
-      nil -> {res.arguments.representation, {:error, res.errors |> List.first()}}
-      _ -> {res.arguments.representation, res.value}
+    case res.errors do
+      [] -> {res.arguments.representation, res.value}
+      errors -> {res.arguments.representation, {:error, errors |> List.first()}}
     end
   end
 

diff --git a/test/absinthe/federation/schema/entities_field_test.exs b/test/absinthe/federation/schema/entities_field_test.exs
@@ -63,6 +63,7 @@ defmodule Absinthe.Federation.Schema.EntitiesFieldTest do
               case upc do
                 "123" -> {:ok, args}
                 "456" -> {:ok, args}
+                "nil" <> _ -> {:ok, nil}
                 _ -> {:error, "Couldn't find product with upc #{upc}"}
               end
             end)
@@ -145,6 +146,35 @@ defmodule Absinthe.Federation.Schema.EntitiesFieldTest do
              } = resp
     end
 
+    test "Handles missing data" do
+      query = """
+        query {
+          _entities(representations: [
+            {
+              __typename: "Product",
+              upc: "nil1"
+            },
+            {
+              __typename: "Product",
+              upc: "nil22"
+            }
+          ]) {
+            __typename
+            ...on Product {
+              upc
+              foo
+            }
+          }
+        }
+      """
+
+      {:ok, resp} = Absinthe.run(query, ResolverSchema, variables: %{})
+
+      assert %{
+               data: %{"_entities" => [nil, nil]}
+             } = resp
+    end
+
     test "falls back to default _resolve_reference implementation" do
       query = """
         query {

diff --git a/test/absinthe/federation/schema/persistent_term_schema_entity_field_test.exs b/test/absinthe/federation/schema/persistent_term_schema_entity_field_test.exs
@@ -26,6 +26,7 @@ defmodule Absinthe.Federation.Schema.PersistentTermSchemaEntityFieldTest do
         resolve(fn
           _, %{name: "John"}, _ -> {:ok, %{__typename: "Person", name: "John", age: 20}}
           _, %{name: "Acme"}, _ -> {:ok, %{__typename: "Business", name: "Acme", employee_count: 10}}
+          _, %{name: "nil" <> _}, _ -> {:ok, nil}
         end)
       end
 
@@ -83,4 +84,39 @@ defmodule Absinthe.Federation.Schema.PersistentTermSchemaEntityFieldTest do
              }
            } = resp
   end
+
+  test "Handles missing data" do
+    query = """
+      query {
+        _entities(representations: [
+          { __typename: "NamedEntity", name: "John" },
+          { __typename: "NamedEntity", name: "nilJohn" },
+          { __typename: "NamedEntity", name: "nilAcme" }
+        ]) {
+          ...on NamedEntity {
+            __typename
+            name
+            ... on Person {
+              age
+            }
+            ... on Business {
+              employeeCount
+            }
+          }
+        }
+      }
+    """
+
+    {:ok, resp} = Absinthe.run(query, EntitySchemaWithPersistentTermProvider, variables: %{})
+
+    assert %{
+             data: %{
+               "_entities" => [
+                 %{"__typename" => "Person", "name" => "John", "age" => 20},
+                 nil,
+                 nil
+               ]
+             }
+           } = resp
+  end
 end