Support mutations (#22)

README.MD

@@ -162,6 +162,12 @@ You can select, if you want to save a `json` file with all the queries that wher
 to do it, on the options pass `queryFile: true`, if you don't pass anything it is not going
 to be saved.
 
+#### Mutations
+You can use [`easygraphql-load-tester`](https://github.com/EasyGraphQL/easygraphql-load-tester) to test
+your mutations as well; to do it, on the options pass `withMutations: true`, if you don't pass anything it is only
+going to test the queries.
+*If you set `withMutations: true`, don't forget to add the input values on the args*
+
 #### Options
 *This is optional, you can leave the first argument empty, if you don't want to pass any options*
 The artillery method will receive some options that will help to create custom queries,
@@ -171,7 +177,8 @@ and test specific queries.
 const options = {
   selectedQueries,
   customQueries: queries,
-  queryFile: true
+  queryFile: true,
+  withMutations: true
 }
 ```
 
@@ -198,7 +205,8 @@ const queries = [
 
 const options = {
   selectedQueries: ['getFamilyInfo', 'searchUser'],
-  customQueries: queries
+  customQueries: queries,
+  withMutations: true
 }
 
 const testCases = easyGraphQLLoadTester.artillery(options)
@@ -336,6 +344,12 @@ You can select, if you want to save a `json` file with all the queries that wher
 to do it, on the options pass `queryFile: true`, if you don't pass anything it is not going
 to be saved.
 
+#### Mutations
+You can use [`easygraphql-load-tester`](https://github.com/EasyGraphQL/easygraphql-load-tester) to test
+your mutations as well; to do it, on the options pass `withMutations: true`, if you don't pass anything it is only
+going to test the queries.
+*If you set `withMutations: true`, don't forget to add the input values on the args*
+
 #### Virtual users
 You can select how many virtual users do you want for your tests, just pass to the options
 `vus: <NUMBER_OF_VUS>`.
@@ -351,6 +365,7 @@ const options = {
   selectedQueries,
   customQueries: queries,
   queryFile: true,
+  withMutations: true,
   vus: 10,
   duration: '10s'
 }
@@ -379,6 +394,7 @@ const queries = [
 const options = {
   selectedQueries: ['getFamilyInfo', 'searchUser'],
   queryFile: true,
+  withMutations: true,
   vus: 10,
   duration: '10s'
 }

examples/artillery/artillery.yml

@@ -14,7 +14,7 @@ scenarios:
               json:
                 query: "{{ $loopElement.query }}"
           - log: "----------------------------------"
-          - log: "Sent a request to the Query: {{ $loopElement.name }}"
+          - log: "Sent a request to the {{ $loopElement.operation }}: {{ $loopElement.name }}"
         over: cases
 
 

examples/artillery/index.js

@@ -15,6 +15,15 @@ const args = {
   },
   searchUser: {
     name: 'demo'
+  },
+  createUser: {
+    name: 'demo'
+  },
+  createCity: {
+    input: {
+      name: 'demo',
+      country: 'Demo'
+    }
   }
 }
 
@@ -35,7 +44,7 @@ const queries = [
 
 const testCases = easyGraphQLLoadTester.artillery({
   customQueries: queries,
-  queryFile: true
+  withMutations: true
 })
 
 module.exports = {

examples/artillery/schema.gql

@@ -16,8 +16,19 @@ type Test {
   email: String!
 }
 
+input CityInput {
+  name: String!
+  country: String!
+}
+
 type Query {
   getFamilyInfo: FamilyInfo!
   searchUser(name: String!): User!
   getFamilyInfoByIsLocal(isLocal: Boolean!, test: [String]!, age: Int!, name: String!): FamilyInfo!
+  getMe: Boolean!
+}
+
+type Mutation {
+  createUser(name: String!): User!
+  createCity(input: CityInput!): User!
 }

lib/loadTesting.js

@@ -3,7 +3,8 @@
 const fs = require('fs')
 const schemaParser = require('easygraphql-parser')
 const { spawn } = require('child_process')
-const { getField, createQueryArguments, createUnionQuery, createQueryToTest } = require('../util')
+const create = require('../utils')
+const { queryField, mutationField } = require('../utils/schemaDefinition')
 
 class LoadTesting {
   constructor (schema, args = {}) {
@@ -15,57 +16,35 @@ class LoadTesting {
     this.arguments = args
   }
 
-  createQuery (queries, selectedQueries) {
+  createQuery (queries, selectedQueries, withMutations) {
     queries = queries || []
     selectedQueries = selectedQueries || []
 
-    this.schema.Query.fields.forEach(query => {
-      if (selectedQueries.length) {
-        const isQueryToTest = selectedQueries.indexOf(query.name)
-        if (isQueryToTest < 0) return
-      }
-
-      const nestedType = this.schema[query.type]
-      const fields = []
-
-      let queryHeader
-      if (query.arguments.length) {
-        const createdArgs = createQueryArguments(query.arguments, this.arguments[query.name])
-        queryHeader = `${query.name}(${createdArgs})`
-      } else {
-        queryHeader = query.name
-      }
-
-      if (nestedType.type === 'UnionTypeDefinition') {
-        const unionQueries = []
-        nestedType.types.forEach(type => {
-          const newNestedType = this.schema[type]
-
-          unionQueries.push(createUnionQuery(newNestedType, this.schema, type))
-        })
-
-        const queryToTest = createQueryToTest(unionQueries, queryHeader)
-
-        queries.push(queryToTest)
-      } else {
-        nestedType.fields.forEach(field => {
-          const createdField = getField(field, this.schema)
-          fields.push(createdField)
-
-          const queryToTest = createQueryToTest(fields, queryHeader)
-
-          queries.push(queryToTest)
-        })
+    const Query = queryField(this.schema)
+    this.schema[Query].fields.forEach(query => {
+      const createdQueries = create(this.schema, query, selectedQueries, this.arguments)
+      if (createdQueries && createdQueries.length > 0) {
+        queries.push(...createdQueries)
       }
     })
 
+    if (withMutations) {
+      const Mutation = mutationField(this.schema)
+      this.schema[Mutation].fields.forEach(mutation => {
+        const createdQueries = create(this.schema, mutation, selectedQueries, this.arguments, true)
+        if (createdQueries && createdQueries.length > 0) {
+          queries.push(...createdQueries)
+        }
+      })
+    }
+
     return queries
   }
 
   artillery (options = {}) {
-    const { customQueries, selectedQueries, queryFile } = options
+    const { customQueries, selectedQueries, queryFile, withMutations } = options
 
-    const queries = this.createQuery(customQueries, selectedQueries)
+    const queries = this.createQuery(customQueries, selectedQueries, withMutations)
 
     if (queryFile) {
       const queryFileName = 'easygraphql-load-tester-queries.json'
@@ -84,9 +63,9 @@ class LoadTesting {
     }
     const queryFileName = 'easygraphql-load-tester-queries.json'
 
-    const { customQueries, selectedQueries, vus, duration, queryFile } = options
+    const { customQueries, selectedQueries, vus, duration, queryFile, withMutations } = options
 
-    const queries = this.createQuery(customQueries, selectedQueries)
+    const queries = this.createQuery(customQueries, selectedQueries, withMutations)
     const selectedVus = vus || ''
 
     const selectedDuration = duration || ''

test/queryGenerator.js

@@ -22,6 +22,9 @@ describe('Query generator', () => {
           id: '1',
           name: 'demo'
         }
+      },
+      isAdmin: {
+        username: 'test'
       }
     }
     const loadTest = new EasyGraphQLTester(schema, args)
@@ -46,6 +49,15 @@ describe('Query generator', () => {
           id: '1',
           name: 'demo'
         }
+      },
+      isAdmin: {
+        username: 'test'
+      },
+      createUser: {
+        input: {
+          name: 'test',
+          email: 'test@test.com'
+        }
       }
     }
     const loadTest = new EasyGraphQLTester(schema, args)
@@ -60,21 +72,23 @@ describe('Query generator', () => {
         }`
       }
     ]
-    const queries = loadTest.createQuery(myQueries)
+    const queries = loadTest.createQuery(myQueries, null, true)
 
     expect(queries).to.exist
     expect(queries).to.be.a('array')
     expect(queries[0].name).to.includes('myNewQuery')
     expect(queries[1].name).to.includes('getMe')
-    expect(queries[queries.length - 1].name).to.includes('getUser')
-    expect(queries[queries.length - 1].name).to.includes('where: {')
+    expect(queries[queries.length - 1].name).to.includes('createUser')
   })
 
   it('Should initialize constructor with selectedQueries queries', () => {
     const args = {
       getUserByUsername: {
         username: 'Test',
         id: 1
+      },
+      isAdmin: {
+        username: 'test'
       }
     }
     const loadTest = new EasyGraphQLTester(schema, args)
@@ -176,6 +190,9 @@ describe('Query generator', () => {
           id: '1',
           name: 'demo'
         }
+      },
+      isAdmin: {
+        username: 'test'
       }
     }
 

test/schema/schema.gql

@@ -1,3 +1,8 @@
+enum Roles {
+  ADMIN
+  USER
+}
+
 type Family {
   name: String!
   ages: [Int]!
@@ -14,6 +19,7 @@ type Me {
   apiKey: String!
   users: [User]!
   verified: Boolean!
+  role: Roles!
 }
 
 type User {
@@ -29,8 +35,18 @@ input UserInfoInput {
   name: String!
 }
 
+type UserInput {
+  name: String!
+  email: String!
+}
+
 type Query {
   getMe: Me
   getUserByUsername(username: String!, id: Int!): User
+  isAdmin(username: String!): Boolean!
   getUser(where: UserInfoInput!): User
 }
+
+type Mutation {
+  createUser(input: UserInput!): User!
+}

utils/index.js

@@ -0,0 +1,54 @@
+'use strict'
+
+const { getField, createQueryArguments, createUnionQuery, createQueryToTest } = require('./util')
+
+function create (schema, query, selectedQueries, args, isMutation) {
+  if (selectedQueries.length) {
+    const isQueryToTest = selectedQueries.indexOf(query.name)
+    if (isQueryToTest < 0) return
+  }
+
+  let queryHeader
+  if (query.arguments.length) {
+    const createdArgs = createQueryArguments(query.arguments, args[query.name])
+    queryHeader = `${query.name}(${createdArgs})`
+  } else {
+    queryHeader = query.name
+  }
+
+  const queriesToTest = []
+  if (['ID', 'String', 'Int', 'Float', 'Boolean'].indexOf(query.type) >= 0) {
+    const queryToTest = createQueryToTest([], queryHeader, isMutation)
+    queriesToTest.push(queryToTest)
+    return queriesToTest
+  }
+
+  const nestedType = schema[query.type]
+  const fields = []
+
+  if (nestedType.type === 'UnionTypeDefinition') {
+    const unionQueries = []
+    nestedType.types.forEach(type => {
+      const newNestedType = schema[type]
+
+      unionQueries.push(createUnionQuery(newNestedType, schema, type))
+    })
+
+    const queryToTest = createQueryToTest(unionQueries, queryHeader, isMutation)
+
+    queriesToTest.push(queryToTest)
+  } else {
+    nestedType.fields.forEach(field => {
+      const createdField = getField(field, schema)
+      fields.push(createdField)
+
+      const queryToTest = createQueryToTest(fields, queryHeader, isMutation)
+
+      queriesToTest.push(queryToTest)
+    })
+  }
+
+  return queriesToTest
+}
+
+module.exports = create