fix: Fix compilation issue

tsedio · Jul 1, 2020 · e1134d8 · e1134d8
1 parent 32d2d72
commit e1134d8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/packages/mongoose/package.json b/packages/mongoose/package.json
@@ -22,7 +22,6 @@
   },
   "devDependencies": {
     "@tsed/common": "5.59.3",
-    "@tsed/core": "5.59.3",
-    "@tsed/testing-mongoose": "5.59.3"
+    "@tsed/core": "5.59.3"
   }
-}
+}
diff --git a/packages/mongoose/src/utils/createModel.ts b/packages/mongoose/src/utils/createModel.ts
@@ -1,5 +1,6 @@
 import {nameOf} from "@tsed/core";
 import * as mongoose from "mongoose";
+import {Connection} from "mongoose";
 
 /**
  * Create an instance of mongoose.model from a class.
@@ -18,7 +19,12 @@ export function createModel<T>(
   name: string = nameOf(target),
   collection?: string,
   skipInit?: boolean,
-  connection = mongoose
+  connection?: Connection
 ) {
-  return connection.model(name, schema, collection, skipInit);
+  /* istanbul ignore else */
+  if (connection) {
+    return connection.model(name, schema, collection);
+  }
+
+  return mongoose.model(name, schema, collection, skipInit);
 }