handle results on delete mutations in tests differently then regular …

…mutations
chriskalmar · Jan 30, 2021 · c1bc3f7 · c1bc3f7
1 parent 13c6f1e
commit c1bc3f7
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/test/db.ts b/test/db.ts
@@ -113,17 +113,30 @@ export const mutate = async (
 ) => {
   const typeName = entity.name;
 
+  const entityMutation = entity.getMutationByName(mutationName);
+
   const resolver = getMutationResolver({
     entity,
-    entityMutation: entity.getMutationByName(mutationName),
+    entityMutation,
     typeName,
     nested: false,
     idResolver: () => {
       return id;
     },
   });
 
-  return resolver({}, { input: { [typeName]: payload } }, context, {});
+  const result = await resolver(
+    {},
+    { input: { [typeName]: payload } },
+    context,
+    {},
+  );
+
+  if (entityMutation.isTypeDelete) {
+    return result;
+  }
+
+  return result[typeName];
 };
 
 export const findOne = async (entity, id, payload, context) => {