nocobase/packages/database/src/__tests__/index.ts

import Database from '../database';
import { Dialect } from 'sequelize';
import path from 'path';
import { uid } from '../utils';

function getTestKey() {
  const { id } = require.main;
  const key = id
    // .replace(process.env.PWD, '')
    .replace(`${process.env.PWD}/packages`, '')
    .replace(/src\/__tests__/g, '')
    .replace('.test.ts', '')
    .replace(/[^\w]/g, '_');
  return key;
}

let config: any = {
  username: process.env.DB_USER,
  password: process.env.DB_PASSWORD,
  database: process.env.DB_DATABASE,
  host: process.env.DB_HOST,
  port: Number.parseInt(process.env.DB_PORT, 10),
  dialect: process.env.DB_DIALECT as Dialect,
  logging: process.env.DB_LOG_SQL === 'on' ? console.log : false,
};

if (process.env.DB_DIALECT === 'sqlite') {
  config = {
    dialect: process.env.DB_DIALECT as Dialect,
    storage: path.resolve(__dirname, `./.db/${uid()}.sqlite`),
    logging: process.env.DB_LOG_SQL === 'on' ? console.log : false,
  };
}

export function getDatabase() {
  return new Database({
    ...config,
    sync: {
      force: true,
    },
    hooks: {
      beforeDefine(model, options) {
        // @ts-ignore
        options.tableNamePrefix = `${getTestKey()}_`;
        // @ts-ignore
        options.hookModelName = options.tableName;
        options.tableName = `${getTestKey()}_${
          options.tableName || options.name.plural
        }`;
      },
    },
  });
}