fixed tests failed

dist/graphql-sequelize-schema-generator.js

@@ -117,6 +117,7 @@ var generateMutationRootType = function generateMutationRootType(models, inputTy
       var _Object$assign2;
 
       var inputType = inputTypes[inputTypeName];
+      var key = models[inputTypeName].primaryKeyAttributes[0];
       var toReturn = Object.assign(fields, (_Object$assign2 = {}, _defineProperty(_Object$assign2, inputTypeName + 'Create', {
         type: outputTypes[inputTypeName], // what is returned by resolve, must be of type GraphQLObjectType
         description: 'Create a ' + inputTypeName,
@@ -129,22 +130,20 @@ var generateMutationRootType = function generateMutationRootType(models, inputTy
         description: 'Update a ' + inputTypeName,
         args: _defineProperty({}, inputTypeName, { type: inputType }),
         resolve: function resolve(source, args, context, info) {
+          var where = _defineProperty({}, key, args[inputTypeName][key]);
           return models[inputTypeName].update(args[inputTypeName], {
-            where: { id: args[inputTypeName].id }
+            where: where
           }).then(function (boolean) {
             // `boolean` equals the number of rows affected (0 or 1)
-            return resolver(models[inputTypeName])(source, { id: args[inputTypeName].id }, context, info);
+            return resolver(models[inputTypeName])(source, where, context, info);
           });
         }
       }), _defineProperty(_Object$assign2, inputTypeName + 'Delete', {
         type: GraphQLInt,
         description: 'Delete a ' + inputTypeName,
-        args: {
-          id: { type: new GraphQLNonNull(GraphQLInt) }
-        },
-        resolve: function resolve(value, _ref) {
-          var id = _ref.id;
-          return models[inputTypeName].destroy({ where: { id: id } });
+        args: _defineProperty({}, key, { type: new GraphQLNonNull(GraphQLInt) }),
+        resolve: function resolve(value, where) {
+          return models[inputTypeName].destroy({ where: where });
         } // Returns the number of rows affected (0 or 1)
       }), _Object$assign2));
       return toReturn;

package.json

@@ -29,10 +29,10 @@
   },
   "homepage": "https://github.com/rpellerin/graphql-sequelize-schema-generator#readme",
   "dependencies": {
-    "graphql": "^0.9.1",
-    "graphql-relay": "^0.5.1",
-    "graphql-sequelize": "^5.3.1",
-    "sequelize": "^3.30.2"
+    "graphql": "^0.10.3",
+    "graphql-relay": "^0.5.2",
+    "graphql-sequelize": "^5.3.3",
+    "sequelize": "^4.2.1"
   },
   "devDependencies": {
     "babel-cli": "^6.24.0",
@@ -50,8 +50,8 @@
     "stringifier": "^1.3.0"
   },
   "peerDependencies": {
-    "graphql": "^0.9.1",
-    "sequelize": "^3.30.2"
+    "graphql": "^0.10.3",
+    "sequelize": "^4.2.1"
   },
   "jest": {
     "testEnvironment": "node",

src/graphql-sequelize-schema-generator.js

@@ -127,6 +127,7 @@ const generateMutationRootType = (models, inputTypes, outputTypes) => {
     fields: Object.keys(inputTypes).reduce(
       (fields, inputTypeName) => {
         const inputType = inputTypes[inputTypeName]
+        const key = models[inputTypeName].primaryKeyAttributes[0]
         const toReturn = Object.assign(fields, {
           [inputTypeName + 'Create']: {
             type: outputTypes[inputTypeName], // what is returned by resolve, must be of type GraphQLObjectType
@@ -145,15 +146,16 @@ const generateMutationRootType = (models, inputTypes, outputTypes) => {
               [inputTypeName]: {type: inputType}
             },
             resolve: (source, args, context, info) => {
+              const where = { [key]: args[inputTypeName][key] }
               return models[inputTypeName]
                 .update(args[inputTypeName], {
-                  where: {id: args[inputTypeName].id}
+                  where
                 })
                 .then(boolean => {
                   // `boolean` equals the number of rows affected (0 or 1)
                   return resolver(models[inputTypeName])(
                     source,
-                    {id: args[inputTypeName].id},
+                    where,
                     context,
                     info
                   )
@@ -164,10 +166,10 @@ const generateMutationRootType = (models, inputTypes, outputTypes) => {
             type: GraphQLInt,
             description: 'Delete a ' + inputTypeName,
             args: {
-              id: {type: new GraphQLNonNull(GraphQLInt)}
+              [key]: {type: new GraphQLNonNull(GraphQLInt)}
             },
-            resolve: (value, {id}) =>
-              models[inputTypeName].destroy({where: {id}}) // Returns the number of rows affected (0 or 1)
+            resolve: (value, where) =>
+              models[inputTypeName].destroy({ where }) // Returns the number of rows affected (0 or 1)
           }
         })
         return toReturn