fix: support class-based mongoose models too (closes #51)

Author: SleepWalker

src/__tests__/datasource.test.js

@@ -70,6 +70,20 @@ describe('Mongoose', () => {
     expect(isCollectionOrModel(undefined)).toBe(false)
   })
 
+  test('mongoose class-based components', () => {
+    /**
+     * @see https://github.com/GraphQLGuide/apollo-datasource-mongodb/issues/51
+     */
+
+    const ClassModel = mongoose.model(
+      class ClassModel extends mongoose.Model {},
+      new Schema({ name: 'string' })
+    )
+
+    expect(isModel(ClassModel)).toBe(true)
+    expect(isCollectionOrModel(ClassModel)).toBe(true)
+  })
+
   test('getCollectionName', () => {
     expect(getCollection(userCollection).collectionName).toBe('users')
     expect(getCollection(UserModel).collectionName).toBe('users')

src/helpers.js

@@ -1,6 +1,14 @@
 const TYPEOF_COLLECTION = 'object'
 
-export const isModel = x => Boolean(x && x.name === 'model')
+export const isModel = x =>
+  Boolean(
+    typeof x === 'function' &&
+      x.prototype &&
+      /**
+       * @see https://github.com/Automattic/mongoose/blob/b4e0ae52a57b886bc7046d38332ce3b38a2f9acd/lib/model.js#L116
+       */
+      x.prototype.$isMongooseModelPrototype
+  )
 
 export const isCollectionOrModel = x =>
   Boolean(x && (typeof x === TYPEOF_COLLECTION || isModel(x)))