MockPayloadGenerator support for @no_inline with args

Summary: MockPayloadGenerator had partial support for `no_inline` (which appears as FragmentSpread nodes in normalization ASTs). However, we missed adding support for `arguments` to `no_inline` fragments, this diff adds support.

Reviewed By: captbaritone

Differential Revision: D46122331

fbshipit-source-id: 24b5e8c2e86184a0bf63f60400a62a35104a8c42

packages/relay-runtime/index.js

@@ -356,6 +356,7 @@ module.exports = {
     OperationTracker: RelayOperationTracker,
     createRelayContext: createRelayContext,
     getOperationVariables: RelayConcreteVariables.getOperationVariables,
+    getLocalVariables: RelayConcreteVariables.getLocalVariables,
     fetchQuery: fetchQueryInternal.fetchQuery,
     fetchQueryDeduped: fetchQueryInternal.fetchQueryDeduped,
     getPromiseForActiveRequest: fetchQueryInternal.getPromiseForActiveRequest,

packages/relay-test-utils/RelayMockPayloadGenerator.js

@@ -26,6 +26,7 @@ import type {
 
 const invariant = require('invariant');
 const {
+  __internal,
   RelayConcreteNode,
   TYPENAME_KEY,
   getModuleComponentKey,
@@ -314,8 +315,24 @@ class RelayMockPayloadGenerator {
           break;
         }
 
-        case CLIENT_COMPONENT:
+        case CLIENT_COMPONENT: {
+          mockData = this._traverseSelections(
+            selection.fragment.selections,
+            typeName,
+            isAbstractType,
+            path,
+            mockData,
+            defaultValues,
+          );
+          break;
+        }
         case FRAGMENT_SPREAD: {
+          const prevVariables = this._variables;
+          this._variables = __internal.getLocalVariables(
+            this._variables,
+            selection.fragment.argumentDefinitions,
+            selection.args,
+          );
           mockData = this._traverseSelections(
             selection.fragment.selections,
             typeName,
@@ -324,6 +341,7 @@ class RelayMockPayloadGenerator {
             mockData,
             defaultValues,
           );
+          this._variables = prevVariables;
           break;
         }
 

packages/relay-test-utils/__tests__/RelayMockPayloadGenerator-test.js

@@ -22,8 +22,9 @@ function testGeneratedData<TVariables: Variables, TData, TRawResponse>(
   query: Query<TVariables, TData, TRawResponse>,
   mockResolvers: ?MockResolvers,
   options: ?{mockClientData?: boolean},
+  variables: Variables = {},
 ): void {
-  const operation = createOperationDescriptor(query, {});
+  const operation = createOperationDescriptor(query, variables);
   const payload = RelayMockPayloadGenerator.generate(
     operation,
     mockResolvers,
@@ -1689,3 +1690,52 @@ describe('with @relay_test_operation', () => {
     });
   });
 });
+
+test('Query with @no_inline fragment spread with literal argument', () => {
+  const query = graphql`
+    query RelayMockPayloadGeneratorTest58Query {
+      node(id: "4") {
+        ...RelayMockPayloadGeneratorTest_fragment59 @arguments(cond: true)
+      }
+    }
+  `;
+  graphql`
+    fragment RelayMockPayloadGeneratorTest_fragment59 on User
+    @argumentDefinitions(cond: {type: "Boolean", defaultValue: false})
+    @no_inline(raw_response_type: true) {
+      id
+      name @include(if: $cond)
+    }
+  `;
+  testGeneratedData(query, undefined, {
+    mockClientData: false,
+  });
+});
+
+test('Query with @no_inline fragment spread with variable argument', () => {
+  const query = graphql`
+    query RelayMockPayloadGeneratorTest60Query($cond: Boolean!) {
+      node(id: "4") {
+        ...RelayMockPayloadGeneratorTest_fragment61 @arguments(cond: $cond)
+      }
+    }
+  `;
+  graphql`
+    fragment RelayMockPayloadGeneratorTest_fragment61 on User
+    @argumentDefinitions(cond: {type: "Boolean", defaultValue: false})
+    @no_inline(raw_response_type: true) {
+      id
+      name @include(if: $cond)
+    }
+  `;
+  testGeneratedData(
+    query,
+    undefined,
+    {
+      mockClientData: false,
+    },
+    {
+      cond: true,
+    },
+  );
+});