fix: return extended class definitions

json-schema-faker · Jan 14, 2021 · 72b1181 · 72b1181
1 parent e4a4ac2
commit 72b1181
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -24,6 +24,10 @@ const AUTOLOAD = {
   'instance.js': 'instanceMethods',
 };
 
+function makeModel() {
+  return class extends Sequelize.Model {};
+}
+
 function JSONSchemaSequelizer(settings, refs, cwd) {
   if (typeof settings === 'string') {
     settings = {
@@ -145,16 +149,20 @@ function JSONSchemaSequelizer(settings, refs, cwd) {
   });
 
   // append model from settings
-  this.add = model => {
+  this.add = (model, _class) => {
     /* istanbul ignore else */
     if (!(model.$schema && model.$schema.id)) {
       throw new Error(`Missing $schema and/or id, given '${_util.inspect(model)}'`);
     }
 
     if (!_defns[model.$schema.id]) {
       _defns[model.$schema.id] = model;
-      _defns[model.$schema.id].$class = class extends Sequelize.Model {};
+      _defns[model.$schema.id].$class = makeModel();
     } else {
+      if (!_defns[model.$schema.id].$class) {
+        _defns[model.$schema.id].$class = makeModel();
+      }
+
       Object.keys(model).forEach(key => {
         /* istanbul ignore else */
         if (!_defns[model.$schema.id][key]) {
@@ -165,7 +173,7 @@ function JSONSchemaSequelizer(settings, refs, cwd) {
       });
     }
 
-    return this;
+    return _class ? _defns[model.$schema.id].$class : this;
   };
 
   // append model from filesystem