clean up test model definitions

alt3 · Nov 5, 2019 · 5197b2e · 5197b2e
1 parent aec4926
commit 5197b2e
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 30 deletions.
diff --git a/test/models/document.js b/test/models/document.js
@@ -1,18 +1,11 @@
 /**
  * Sequelize model definition for testing User hasMany, Document belongsTo.
- */
-
-const Sequelize = require('sequelize');
-
-const { DataTypes } = Sequelize;
-
-/**
- * Initialize Profile definition
  *
  * @param sequelize Sequelize Instance
- * @returns {ProfileClass} Returns the Profile model
+ * @param Sequelize Sequelize Class
+ * @returns {DocumentClass} Returns the Document model
  */
-module.exports = sequelize => {
+module.exports = (sequelize, { DataTypes }) => {
   const Model = sequelize.define(
     'document',
     {

diff --git a/test/models/profile.js b/test/models/profile.js
@@ -1,18 +1,11 @@
 /**
  * Sequelize model definition for testing User hasOne, Profile belongsTo.
- */
-
-const Sequelize = require('sequelize');
-
-const { DataTypes } = Sequelize;
-
-/**
- * Initialize Profile definition
  *
  * @param sequelize Sequelize Instance
+ * @param Sequelize Sequelize Class
  * @returns {ProfileClass} Returns the Profile model
  */
-module.exports = sequelize => {
+module.exports = (sequelize, { DataTypes }) => {
   const Model = sequelize.define(
     'profile',
     {

diff --git a/test/models/user.js b/test/models/user.js
@@ -1,3 +1,5 @@
+const supportedDataType = require('../utils/supported-datatype');
+
 /**
  * Sequelize attribute definitions for the `user` model.
  *
@@ -8,21 +10,12 @@
  * be present in the model when testing Sequelize v4.
  *
  * @see https://sequelize.org/master/manual/data-types.html
- */
-
-const Sequelize = require('sequelize');
-
-const { DataTypes } = Sequelize;
-
-const supportedDataType = require('../utils/supported-datatype');
-
-/**
- * Initialize User definition
  *
  * @param sequelize Sequelize Instance
+ * @param Sequelize Sequelize Class
  * @returns {UserClass} Returns the User model
  */
-module.exports = sequelize => {
+module.exports = (sequelize, { DataTypes }) => {
   const Model = sequelize.define(
     'user',
     {