@nocobase/database 0.5.0-alpha.37 → 0.7.0-alpha.1

package/lib/update-guard.js

@@ -0,0 +1,129 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UpdateGuard = void 0;
+const lodash_1 = __importDefault(require("lodash"));
+const model_1 = require("./model");
+class UpdateGuard {
+    setAction(action) {
+        this.action = action;
+    }
+    setModel(model) {
+        this.model = model;
+    }
+    setAssociationKeysToBeUpdate(associationKeysToBeUpdate) {
+        if (this.action == 'create') {
+            this.associationKeysToBeUpdate = Object.keys(this.model.associations);
+        }
+        else {
+            this.associationKeysToBeUpdate = associationKeysToBeUpdate;
+        }
+    }
+    setWhiteList(whiteList) {
+        this.whiteList = whiteList;
+    }
+    setBlackList(blackList) {
+        this.blackList = blackList;
+    }
+    /**
+     * Sanitize values by whitelist blacklist
+     * @param values
+     */
+    sanitize(values) {
+        values = lodash_1.default.clone(values);
+        if (!this.model) {
+            throw new Error('please set model first');
+        }
+        const associations = this.model.associations;
+        const associationsValues = lodash_1.default.pick(values, Object.keys(associations));
+        // build params of association update guard
+        const listOfAssociation = (list, association) => {
+            if (list) {
+                list = list
+                    .filter((whiteListKey) => whiteListKey.startsWith(`${association}.`))
+                    .map((whiteListKey) => whiteListKey.replace(`${association}.`, ''));
+                if (list.length == 0) {
+                    return undefined;
+                }
+                return list;
+            }
+            return undefined;
+        };
+        // sanitize association values
+        Object.keys(associationsValues).forEach((association) => {
+            let associationValues = associationsValues[association];
+            const filterAssociationToBeUpdate = (value) => {
+                const associationKeysToBeUpdate = this.associationKeysToBeUpdate || [];
+                if (associationKeysToBeUpdate.includes(association)) {
+                    return value;
+                }
+                const associationObj = associations[association];
+                const associationKeyName = associationObj.associationType == 'BelongsTo' || associationObj.associationType == 'HasOne'
+                    ? associationObj.targetKey
+                    : associationObj.target.primaryKeyAttribute;
+                if (value[associationKeyName]) {
+                    return lodash_1.default.pick(value, [associationKeyName, ...Object.keys(associationObj.target.associations)]);
+                }
+                return value;
+            };
+            const sanitizeValue = (value) => {
+                const associationUpdateGuard = new UpdateGuard();
+                associationUpdateGuard.setModel(associations[association].target);
+                ['whiteList', 'blackList', 'associationKeysToBeUpdate'].forEach((optionKey) => {
+                    associationUpdateGuard[`set${lodash_1.default.upperFirst(optionKey)}`](listOfAssociation(this[optionKey], association));
+                });
+                return associationUpdateGuard.sanitize(filterAssociationToBeUpdate(value));
+            };
+            if (Array.isArray(associationValues)) {
+                associationValues = associationValues.map((value) => {
+                    if (typeof value == 'string' || typeof value == 'number') {
+                        return value;
+                    }
+                    else {
+                        return sanitizeValue(value);
+                    }
+                });
+            }
+            else if (typeof associationValues === 'object' && associationValues !== null) {
+                associationValues = sanitizeValue(associationValues);
+            }
+            // set association values to sanitized value
+            values[association] = associationValues;
+        });
+        if (values instanceof model_1.Model) {
+            return values;
+        }
+        let valuesKeys = Object.keys(values || {});
+        // handle whitelist
+        if (this.whiteList) {
+            valuesKeys = valuesKeys.filter((valueKey) => {
+                return (this.whiteList.findIndex((whiteKey) => {
+                    const keyPaths = whiteKey.split('.');
+                    return keyPaths[0] === valueKey;
+                }) !== -1);
+            });
+        }
+        // handle blacklist
+        if (this.blackList) {
+            valuesKeys = valuesKeys.filter((valueKey) => !this.blackList.includes(valueKey));
+        }
+        const result = valuesKeys.reduce((obj, key) => {
+            lodash_1.default.set(obj, key, values[key]);
+            return obj;
+        }, {});
+        return result;
+    }
+    static fromOptions(model, options) {
+        const guard = new UpdateGuard();
+        guard.setModel(model);
+        guard.setWhiteList(options.whitelist);
+        guard.setBlackList(options.blacklist);
+        guard.setAction(lodash_1.default.get(options, 'action', 'update'));
+        guard.setAssociationKeysToBeUpdate(options.updateAssociationValues);
+        return guard;
+    }
+}
+exports.UpdateGuard = UpdateGuard;
+//# sourceMappingURL=update-guard.js.map

package/lib/update-guard.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"update-guard.js","sourceRoot":"","sources":["../src/update-guard.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAE5B,mCAAgC;AAUhC,MAAa,WAAW;IAOtB,SAAS,CAAC,MAAoB;QAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,QAAQ,CAAC,KAAqB;QAC5B,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,4BAA4B,CAAC,yBAAoD;QAC/E,IAAI,IAAI,CAAC,MAAM,IAAI,QAAQ,EAAE;YAC3B,IAAI,CAAC,yBAAyB,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;SACvE;aAAM;YACL,IAAI,CAAC,yBAAyB,GAAG,yBAAyB,CAAC;SAC5D;IACH,CAAC;IAED,YAAY,CAAC,SAAoB;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,YAAY,CAAC,SAAoB;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED;;;OAGG;IACH,QAAQ,CAAC,MAAoB;QAC3B,MAAM,GAAG,gBAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAE9B,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE;YACf,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;SAC3C;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC;QAC7C,MAAM,kBAAkB,GAAG,gBAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;QAE1E,2CAA2C;QAC3C,MAAM,iBAAiB,GAAG,CAAC,IAAI,EAAE,WAAW,EAAE,EAAE;YAC9C,IAAI,IAAI,EAAE;gBACR,IAAI,GAAG,IAAI;qBACR,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,UAAU,CAAC,GAAG,WAAW,GAAG,CAAC,CAAC;qBACpE,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,WAAW,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;gBAEtE,IAAI,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE;oBACpB,OAAO,SAAS,CAAC;iBAClB;gBAED,OAAO,IAAI,CAAC;aACb;YAED,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC;QAEF,8BAA8B;QAC9B,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE;YACtD,IAAI,iBAAiB,GAAG,kBAAkB,CAAC,WAAW,CAAC,CAAC;YAExD,MAAM,2BAA2B,GAAG,CAAC,KAAK,EAAE,EAAE;gBAC5C,MAAM,yBAAyB,GAAG,IAAI,CAAC,yBAAyB,IAAI,EAAE,CAAC;gBAEvE,IAAI,yBAAyB,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE;oBACnD,OAAO,KAAK,CAAC;iBACd;gBAED,MAAM,cAAc,GAAG,YAAY,CAAC,WAAW,CAAC,CAAC;gBAEjD,MAAM,kBAAkB,GACtB,cAAc,CAAC,eAAe,IAAI,WAAW,IAAI,cAAc,CAAC,eAAe,IAAI,QAAQ;oBACzF,CAAC,CAAO,cAAe,CAAC,SAAS;oBACjC,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,mBAAmB,CAAC;gBAEhD,IAAI,KAAK,CAAC,kBAAkB,CAAC,EAAE;oBAC7B,OAAO,gBAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,kBAAkB,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;iBACrG;gBAED,OAAO,KAAK,CAAC;YACf,CAAC,CAAC;YAEF,MAAM,aAAa,GAAG,CAAC,KAAK,EAAE,EAAE;gBAC9B,MAAM,sBAAsB,GAAG,IAAI,WAAW,EAAE,CAAC;gBACjD,sBAAsB,CAAC,QAAQ,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,CAAC;gBAElE,CAAC,WAAW,EAAE,WAAW,EAAE,2BAA2B,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;oBAC5E,sBAAsB,CAAC,MAAM,gBAAM,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC;gBAChH,CAAC,CAAC,CAAC;gBAEH,OAAO,sBAAsB,CAAC,QAAQ,CAAC,2BAA2B,CAAC,KAAK,CAAC,CAAC,CAAC;YAC7E,CAAC,CAAC;YAEF,IAAI,KAAK,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE;gBACpC,iBAAiB,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;oBAClD,IAAI,OAAO,KAAK,IAAI,QAAQ,IAAI,OAAO,KAAK,IAAI,QAAQ,EAAE;wBACxD,OAAO,KAAK,CAAC;qBACd;yBAAM;wBACL,OAAO,aAAa,CAAC,KAAK,CAAC,CAAC;qBAC7B;gBACH,CAAC,CAAC,CAAC;aACJ;iBAAM,IAAI,OAAO,iBAAiB,KAAK,QAAQ,IAAI,iBAAiB,KAAK,IAAI,EAAE;gBAC9E,iBAAiB,GAAG,aAAa,CAAC,iBAAiB,CAAC,CAAC;aACtD;YAED,4CAA4C;YAC5C,MAAM,CAAC,WAAW,CAAC,GAAG,iBAAiB,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,IAAI,MAAM,YAAY,aAAK,EAAE;YAC3B,OAAO,MAAM,CAAC;SACf;QAED,IAAI,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QAE3C,mBAAmB;QACnB,IAAI,IAAI,CAAC,SAAS,EAAE;YAClB,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAC1C,OAAO,CACL,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,QAAQ,EAAE,EAAE;oBACpC,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;oBACrC,OAAO,QAAQ,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC;gBAClC,CAAC,CAAC,KAAK,CAAC,CAAC,CACV,CAAC;YACJ,CAAC,CAAC,CAAC;SACJ;QAED,mBAAmB;QACnB,IAAI,IAAI,CAAC,SAAS,EAAE;YAClB,UAAU,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;SAClF;QAED,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YAC5C,gBAAM,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YAClC,OAAO,GAAG,CAAC;QACb,CAAC,EAAE,EAAE,CAAC,CAAC;QAEP,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,MAAM,CAAC,WAAW,CAAC,KAAK,EAAE,OAAO;QAC/B,MAAM,KAAK,GAAG,IAAI,WAAW,EAAE,CAAC;QAChC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACtB,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACtC,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACtC,KAAK,CAAC,SAAS,CAAC,gBAAM,CAAC,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;QACzD,KAAK,CAAC,4BAA4B,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAC;QACpE,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AA1JD,kCA0JC","sourcesContent":["import lodash from 'lodash';\nimport { ModelCtor } from 'sequelize';\nimport { Model } from './model';\nimport { AssociationKeysToBeUpdate, BlackList, WhiteList } from './repository';\n\ntype UpdateValueItem = string | number | UpdateValues;\n\ntype UpdateValues = {\n  [key: string]: UpdateValueItem | Array<UpdateValueItem>;\n};\n\ntype UpdateAction = 'create' | 'update';\nexport class UpdateGuard {\n  model: ModelCtor<any>;\n  action: UpdateAction;\n  private associationKeysToBeUpdate: AssociationKeysToBeUpdate;\n  private blackList: BlackList;\n  private whiteList: WhiteList;\n\n  setAction(action: UpdateAction) {\n    this.action = action;\n  }\n\n  setModel(model: ModelCtor<any>) {\n    this.model = model;\n  }\n\n  setAssociationKeysToBeUpdate(associationKeysToBeUpdate: AssociationKeysToBeUpdate) {\n    if (this.action == 'create') {\n      this.associationKeysToBeUpdate = Object.keys(this.model.associations);\n    } else {\n      this.associationKeysToBeUpdate = associationKeysToBeUpdate;\n    }\n  }\n\n  setWhiteList(whiteList: WhiteList) {\n    this.whiteList = whiteList;\n  }\n\n  setBlackList(blackList: BlackList) {\n    this.blackList = blackList;\n  }\n\n  /**\n   * Sanitize values by whitelist blacklist\n   * @param values\n   */\n  sanitize(values: UpdateValues) {\n    values = lodash.clone(values);\n\n    if (!this.model) {\n      throw new Error('please set model first');\n    }\n\n    const associations = this.model.associations;\n    const associationsValues = lodash.pick(values, Object.keys(associations));\n\n    // build params of association update guard\n    const listOfAssociation = (list, association) => {\n      if (list) {\n        list = list\n          .filter((whiteListKey) => whiteListKey.startsWith(`${association}.`))\n          .map((whiteListKey) => whiteListKey.replace(`${association}.`, ''));\n\n        if (list.length == 0) {\n          return undefined;\n        }\n\n        return list;\n      }\n\n      return undefined;\n    };\n\n    // sanitize association values\n    Object.keys(associationsValues).forEach((association) => {\n      let associationValues = associationsValues[association];\n\n      const filterAssociationToBeUpdate = (value) => {\n        const associationKeysToBeUpdate = this.associationKeysToBeUpdate || [];\n\n        if (associationKeysToBeUpdate.includes(association)) {\n          return value;\n        }\n\n        const associationObj = associations[association];\n\n        const associationKeyName =\n          associationObj.associationType == 'BelongsTo' || associationObj.associationType == 'HasOne'\n            ? (<any>associationObj).targetKey\n            : associationObj.target.primaryKeyAttribute;\n\n        if (value[associationKeyName]) {\n          return lodash.pick(value, [associationKeyName, ...Object.keys(associationObj.target.associations)]);\n        }\n\n        return value;\n      };\n\n      const sanitizeValue = (value) => {\n        const associationUpdateGuard = new UpdateGuard();\n        associationUpdateGuard.setModel(associations[association].target);\n\n        ['whiteList', 'blackList', 'associationKeysToBeUpdate'].forEach((optionKey) => {\n          associationUpdateGuard[`set${lodash.upperFirst(optionKey)}`](listOfAssociation(this[optionKey], association));\n        });\n\n        return associationUpdateGuard.sanitize(filterAssociationToBeUpdate(value));\n      };\n\n      if (Array.isArray(associationValues)) {\n        associationValues = associationValues.map((value) => {\n          if (typeof value == 'string' || typeof value == 'number') {\n            return value;\n          } else {\n            return sanitizeValue(value);\n          }\n        });\n      } else if (typeof associationValues === 'object' && associationValues !== null) {\n        associationValues = sanitizeValue(associationValues);\n      }\n\n      // set association values to sanitized value\n      values[association] = associationValues;\n    });\n\n    if (values instanceof Model) {\n      return values;\n    }\n\n    let valuesKeys = Object.keys(values || {});\n\n    // handle whitelist\n    if (this.whiteList) {\n      valuesKeys = valuesKeys.filter((valueKey) => {\n        return (\n          this.whiteList.findIndex((whiteKey) => {\n            const keyPaths = whiteKey.split('.');\n            return keyPaths[0] === valueKey;\n          }) !== -1\n        );\n      });\n    }\n\n    // handle blacklist\n    if (this.blackList) {\n      valuesKeys = valuesKeys.filter((valueKey) => !this.blackList.includes(valueKey));\n    }\n\n    const result = valuesKeys.reduce((obj, key) => {\n      lodash.set(obj, key, values[key]);\n      return obj;\n    }, {});\n\n    return result;\n  }\n\n  static fromOptions(model, options) {\n    const guard = new UpdateGuard();\n    guard.setModel(model);\n    guard.setWhiteList(options.whitelist);\n    guard.setBlackList(options.blacklist);\n    guard.setAction(lodash.get(options, 'action', 'update'));\n    guard.setAssociationKeysToBeUpdate(options.updateAssociationValues);\n    return guard;\n  }\n}\n"]}

package/package.json

@@ -1,14 +1,26 @@
 {
   "name": "@nocobase/database",
-  "version": "0.5.0-alpha.37",
+  "version": "0.7.0-alpha.1",
   "description": "",
   "main": "./lib/index.js",
   "types": "./lib/index.d.ts",
-  "license": "MIT",
-  "scripts": {},
+  "license": "Apache-2.0",
+  "licenses": [
+    {
+      "type": "Apache-2.0",
+      "url": "http://www.apache.org/licenses/LICENSE-2.0"
+    }
+  ],
+  "scripts": {
+    "build": "rimraf -rf lib esm dist && npm run build:cjs && npm run build:esm",
+    "build:cjs": "tsc --project tsconfig.build.json",
+    "build:esm": "tsc --project tsconfig.build.json --module es2015 --outDir esm"
+  },
   "dependencies": {
-    "bcrypt": "^5.0.0",
+    "@nocobase/utils": "0.7.0-alpha.1",
+    "async-mutex": "^0.3.2",
     "deepmerge": "^4.2.2",
+    "flat": "^5.0.2",
     "glob": "^7.1.6",
     "sequelize": "^6.9.0"
   },
@@ -17,5 +29,5 @@
     "url": "git+https://github.com/nocobase/nocobase.git",
     "directory": "packages/database"
   },
-  "gitHead": "a7b2ffbb76389144de1b54b32e1fb3e2f7b91350"
+  "gitHead": "e7293ad7aadbdf2084042f7800a232af6e0b7a8a"
 }

package/src/__tests__/collection-importer.test.ts

@@ -0,0 +1,21 @@
+import { ImporterReader } from '../collection-importer';
+import * as path from 'path';
+import { extend } from '../database';
+
+describe('collection importer', () => {
+  test('import reader', async () => {
+    const reader = new ImporterReader(path.resolve(__dirname, './fixtures/collections'));
+
+    const modules = await reader.read();
+    expect(modules).toBeDefined();
+  });
+
+  test('extend', async () => {
+    const extendObject = extend({
+      name: 'tags',
+      fields: [{ type: 'string', name: 'title' }],
+    });
+
+    expect(extendObject).toHaveProperty('extend');
+  });
+});

package/src/__tests__/collection.sortable.test.ts

@@ -0,0 +1,65 @@
+import { mockDatabase } from './index';
+import { Database } from '../database';
+
+describe('collection sortable options', () => {
+  let db: Database;
+
+  beforeEach(async () => {
+    db = mockDatabase();
+  });
+
+  afterEach(async () => {
+    await db.close();
+  });
+
+  test('sortable=true', async () => {
+    const Test = db.collection({
+      name: 'test',
+      sortable: true,
+    });
+
+    const model = Test.model;
+
+    await db.sync();
+    const instance = await model.create();
+    expect(model.rawAttributes['sort']).toBeDefined();
+    expect(instance.get('sort')).toBe(1);
+  });
+
+  test('sortable=string', async () => {
+    const Test = db.collection({
+      name: 'test',
+      sortable: 'order',
+    });
+
+    const model = Test.model;
+
+    await db.sync();
+    const instance = await model.create();
+    expect(model.rawAttributes['order']).toBeDefined();
+    expect(instance.get('order')).toBe(1);
+  });
+
+  test('sortable=object', async () => {
+    const Test = db.collection({
+      name: 'test',
+      sortable: {
+        name: 'sort',
+        scopeKey: 'status',
+      },
+      fields: [{ type: 'string', name: 'status' }],
+    });
+
+    await db.sync();
+
+    const t1 = await Test.model.create({ status: 'publish' });
+    const t2 = await Test.model.create({ status: 'publish' });
+    const t3 = await Test.model.create({ status: 'draft' });
+    const t4 = await Test.model.create({ status: 'draft' });
+
+    expect(t1.get('sort')).toBe(1);
+    expect(t2.get('sort')).toBe(2);
+    expect(t3.get('sort')).toBe(1);
+    expect(t4.get('sort')).toBe(2);
+  });
+});

package/src/__tests__/collection.test.ts

@@ -0,0 +1,218 @@
+import { Collection } from '../collection';
+import { Database } from '../database';
+import { mockDatabase } from './index';
+
+describe('collection', () => {
+  let db: Database;
+
+  beforeEach(async () => {
+    db = mockDatabase();
+  });
+
+  afterEach(async () => {
+    await db.close();
+  });
+
+  test('collection disable authGenId', async () => {
+    const Test = db.collection({
+      name: 'test',
+      autoGenId: false,
+      fields: [{ type: 'string', name: 'uid', primaryKey: true }],
+    });
+
+    const model = Test.model;
+
+    await db.sync();
+    expect(model.rawAttributes['id']).toBeUndefined();
+  });
+
+  test('new collection', async () => {
+    const collection = new Collection(
+      {
+        name: 'test',
+      },
+      { database: db },
+    );
+
+    expect(collection.name).toEqual('test');
+  });
+
+  test('collection create field', async () => {
+    const collection = new Collection(
+      {
+        name: 'user',
+      },
+      { database: db },
+    );
+
+    collection.addField('age', {
+      type: 'integer',
+    });
+
+    const ageField = collection.getField('age');
+    expect(ageField).toBeDefined();
+    expect(collection.hasField('age')).toBeTruthy();
+    expect(collection.hasField('test')).toBeFalsy();
+
+    collection.removeField('age');
+    expect(collection.hasField('age')).toBeFalsy();
+  });
+
+  test('collection set fields', () => {
+    const collection = new Collection(
+      {
+        name: 'user',
+      },
+      { database: db },
+    );
+
+    collection.setFields([{ type: 'string', name: 'firstName' }]);
+    expect(collection.hasField('firstName')).toBeTruthy();
+  });
+
+  test('update collection field', async () => {
+    const collection = new Collection(
+      {
+        name: 'posts',
+        fields: [{ type: 'string', name: 'title' }],
+      },
+      {
+        database: db,
+      },
+    );
+    expect(collection.hasField('title')).toBeTruthy();
+
+    collection.updateField('title', {
+      type: 'string',
+      name: 'content',
+    });
+
+    expect(collection.hasField('title')).toBeFalsy();
+    expect(collection.hasField('content')).toBeTruthy();
+  });
+
+  test('collection with association', async () => {
+    const User = db.collection({
+      name: 'users',
+      fields: [
+        { type: 'string', name: 'name' },
+        { type: 'integer', name: 'age' },
+        { type: 'hasMany', name: 'posts' },
+      ],
+    });
+
+    const Post = db.collection({
+      name: 'posts',
+      fields: [
+        { type: 'string', name: 'title' },
+        { type: 'string', name: 'content' },
+        {
+          type: 'belongsTo',
+          name: 'user',
+        },
+        {
+          type: 'hasMany',
+          name: 'comments',
+        },
+      ],
+    });
+
+    const Comment = db.collection({
+      name: 'comments',
+      fields: [
+        { type: 'string', name: 'content' },
+        { type: 'string', name: 'comment_as' },
+        { type: 'belongsTo', name: 'post' },
+      ],
+    });
+
+    expect(User.model.associations['posts']).toBeDefined();
+    expect(Post.model.associations['comments']).toBeDefined();
+
+    expect(User.model.associations['posts'].target.associations['comments']).toBeDefined();
+  });
+});
+
+describe('collection sync', () => {
+  let db: Database;
+
+  beforeEach(async () => {
+    db = mockDatabase();
+  });
+
+  afterEach(async () => {
+    await db.close();
+  });
+
+  test('sync fields', async () => {
+    const collection = new Collection(
+      {
+        name: 'users',
+      },
+      { database: db },
+    );
+
+    collection.setFields([
+      { type: 'string', name: 'firstName' },
+      { type: 'string', name: 'lastName' },
+      { type: 'integer', name: 'age' },
+    ]);
+
+    await collection.sync();
+    const tableFields = await (<any>collection.model).queryInterface.describeTable(`${db.getTablePrefix()}users`);
+
+    expect(tableFields).toHaveProperty('firstName');
+    expect(tableFields).toHaveProperty('lastName');
+    expect(tableFields).toHaveProperty('age');
+  });
+
+  test('sync with association not exists', async () => {
+    const collection = new Collection(
+      {
+        name: 'posts',
+        fields: [
+          { type: 'string', name: 'title' },
+          { type: 'belongsTo', name: 'users' },
+        ],
+      },
+      { database: db },
+    );
+
+    await collection.sync();
+
+    const model = collection.model;
+
+    const tableFields = await (<any>model).queryInterface.describeTable(`${db.getTablePrefix()}posts`);
+
+    expect(tableFields['userId']).toBeUndefined();
+  });
+
+  test('sync with association', async () => {
+    new Collection(
+      {
+        name: 'tags',
+        fields: [{ type: 'string', name: 'name' }],
+      },
+      { database: db },
+    );
+
+    const collection = new Collection(
+      {
+        name: 'posts',
+        fields: [
+          { type: 'string', name: 'title' },
+          { type: 'belongsToMany', name: 'tags' },
+        ],
+      },
+      {
+        database: db,
+      },
+    );
+
+    const model = collection.model;
+    await collection.sync();
+    const tableFields = await (<any>model).queryInterface.describeTable(`${db.getTablePrefix()}postsTags`);
+    expect(tableFields['postId']).toBeDefined();
+    expect(tableFields['tagId']).toBeDefined();
+  });
+});

package/src/__tests__/database.import.test.ts

@@ -0,0 +1,33 @@
+import { mockDatabase } from './index';
+import path from 'path';
+import Database from '../database';
+
+describe('database', () => {
+  let db: Database;
+
+  beforeEach(() => {
+    db = mockDatabase();
+  });
+
+  afterEach(async () => {
+    await db.close();
+  });
+
+  test('import', async () => {
+    await db.import({
+      directory: path.resolve(__dirname, './fixtures/c0'),
+    });
+    await db.import({
+      directory: path.resolve(__dirname, './fixtures/c1'),
+    });
+    await db.import({
+      directory: path.resolve(__dirname, './fixtures/c2'),
+    });
+
+    const test = db.getCollection('tests');
+
+    expect(test.getField('n0')).toBeDefined();
+    expect(test.getField('n1')).toBeDefined();
+    expect(test.getField('n2')).toBeDefined();
+  });
+});