fix(ElasticSearchApi): Add body arg to POST, PUT methods

Author: nodkz

src/ElasticApiParser.js

@@ -10,6 +10,7 @@ import {
   GraphQLBoolean,
   GraphQLObjectType,
   GraphQLEnumType,
+  GraphQLNonNull,
 } from 'graphql';
 import { GraphQLJSON, upperFirst, TypeComposer } from 'graphql-compose';
 
@@ -62,6 +63,8 @@ export type ElasticCaSettingsT = {
   },
   url?: ElasticCaSettingsUrlT,
   urls?: ElasticCaSettingsUrlT[],
+  needBody?: true,
+  method?: string,
 };
 
 export default class ElasticApiParser {
@@ -296,7 +299,14 @@ export default class ElasticApiParser {
     settings: ?ElasticCaSettingsT
   ): GraphQLFieldConfigArgumentMap {
     const result = {};
-    const { params, urls, url } = settings || {};
+    const { params, urls, url, method, needBody } = settings || {};
+
+    if (method === 'POST' || method === 'PUT') {
+      result.body = {
+        type: needBody ? new GraphQLNonNull(GraphQLJSON) : GraphQLJSON,
+      };
+    }
+
     if (params) {
       Object.keys(params).forEach(k => {
         const fieldConfig = this.paramToGraphQLArgConfig(params[k], k);

src/__tests__/ElasticApiParser-test.js

@@ -9,6 +9,7 @@ import {
   GraphQLBoolean,
   GraphQLObjectType,
   GraphQLEnumType,
+  GraphQLNonNull,
 } from 'graphql';
 import { GraphQLJSON, TypeComposer } from 'graphql-compose';
 
@@ -272,6 +273,25 @@ describe('ElasticApiParser', () => {
   });
 
   describe('settingsToArgMap()', () => {
+    it ('should create body arg if POST or PUT method', () => {
+      const args = parser.settingsToArgMap({
+        params: {},
+        method: 'POST',
+      });
+      expect(args).toMatchObject({ body: {} });
+      expect(args.body.type).toEqual(GraphQLJSON);
+    });
+
+    it ('should create required body arg if POST or PUT method', () => {
+      const args = parser.settingsToArgMap({
+        params: {},
+        method: 'POST',
+        needBody: true,
+      });
+      expect(args).toMatchObject({ body: {} });
+      expect(args.body.type).toBeInstanceOf(GraphQLNonNull);
+    });
+
     it('should traverse params', () => {
       expect(
         parser.settingsToArgMap({

src/__tests__/__snapshots__/ElasticApiParser-test.js.snap

@@ -32,6 +32,9 @@ Object {
         "description": "The analyzer to use for the query string",
         "type": "String",
       },
+      "body": Object {
+        "type": "JSON",
+      },
       "defaultOperator": Object {
         "defaultValue": "OR",
         "type": "ElasticEnum_DefaultOperator",
@@ -179,6 +182,9 @@ Object {
         "description": "The analyzer to use for the query string",
         "type": "String",
       },
+      "body": Object {
+        "type": "JSON",
+      },
       "conflicts": Object {
         "defaultValue": "abort",
         "type": "ElasticEnum_Conflicts",