136 lines
3.2 KiB
JavaScript
136 lines
3.2 KiB
JavaScript
const path = require("path")
|
|
require("dotenv").config({ path: path.join(__dirname, "../.env") })
|
|
|
|
const { dropDatabase, createDatabase } = require("pg-god")
|
|
const { createConnection } = require("typeorm")
|
|
|
|
const DB_USERNAME = process.env.DB_USERNAME || "postgres"
|
|
const DB_PASSWORD = process.env.DB_PASSWORD || ""
|
|
const DB_URL = `postgres://${DB_USERNAME}:${DB_PASSWORD}@localhost/medusa-integration`
|
|
|
|
const pgGodCredentials = {
|
|
user: DB_USERNAME,
|
|
password: DB_PASSWORD,
|
|
}
|
|
|
|
const keepTables = [
|
|
"staged_job",
|
|
"shipping_profile",
|
|
"fulfillment_provider",
|
|
"payment_provider",
|
|
"country",
|
|
"currency",
|
|
]
|
|
|
|
let connectionType = "postgresql"
|
|
|
|
const DbTestUtil = {
|
|
db_: null,
|
|
|
|
setDb: function (connection) {
|
|
this.db_ = connection
|
|
},
|
|
|
|
clear: async function () {
|
|
this.db_.synchronize(true)
|
|
},
|
|
|
|
teardown: async function () {
|
|
const entities = this.db_.entityMetadatas
|
|
const manager = this.db_.manager
|
|
|
|
if (connectionType === "sqlite") {
|
|
await manager.query(`PRAGMA foreign_keys = OFF`)
|
|
} else {
|
|
await manager.query(`SET session_replication_role = 'replica';`)
|
|
}
|
|
|
|
for (const entity of entities) {
|
|
if (keepTables.includes(entity.tableName)) {
|
|
continue
|
|
}
|
|
|
|
await manager.query(`DELETE FROM "${entity.tableName}";`)
|
|
}
|
|
if (connectionType === "sqlite") {
|
|
await manager.query(`PRAGMA foreign_keys = ON`)
|
|
} else {
|
|
await manager.query(`SET session_replication_role = 'origin';`)
|
|
}
|
|
},
|
|
|
|
shutdown: async function () {
|
|
await this.db_.close()
|
|
const databaseName = "medusa-integration"
|
|
return await dropDatabase({ databaseName }, pgGodCredentials)
|
|
},
|
|
}
|
|
|
|
const instance = DbTestUtil
|
|
|
|
module.exports = {
|
|
initDb: async function ({ cwd }) {
|
|
const configPath = path.resolve(path.join(cwd, `medusa-config.js`))
|
|
|
|
const modelsLoader = require(path.join(
|
|
cwd,
|
|
`node_modules`,
|
|
`@medusajs`,
|
|
`medusa`,
|
|
`dist`,
|
|
`loaders`,
|
|
`models`
|
|
)).default
|
|
const entities = modelsLoader({}, { register: false })
|
|
|
|
const { projectConfig } = require(configPath)
|
|
if (projectConfig.database_type === "sqlite") {
|
|
connectionType = "sqlite"
|
|
const dbConnection = await createConnection({
|
|
type: "sqlite",
|
|
database: projectConfig.database_database,
|
|
synchronize: true,
|
|
entities,
|
|
})
|
|
|
|
instance.setDb(dbConnection)
|
|
return dbConnection
|
|
} else {
|
|
const migrationDir = path.resolve(
|
|
path.join(
|
|
cwd,
|
|
`node_modules`,
|
|
`@medusajs`,
|
|
`medusa`,
|
|
`dist`,
|
|
`migrations`
|
|
)
|
|
)
|
|
|
|
const databaseName = "medusa-integration"
|
|
await createDatabase({ databaseName }, pgGodCredentials)
|
|
|
|
const connection = await createConnection({
|
|
type: "postgres",
|
|
url: DB_URL,
|
|
migrations: [`${migrationDir}/*.js`],
|
|
})
|
|
|
|
await connection.runMigrations()
|
|
await connection.close()
|
|
|
|
const dbConnection = await createConnection({
|
|
type: "postgres",
|
|
url: DB_URL,
|
|
entities,
|
|
})
|
|
|
|
instance.setDb(dbConnection)
|
|
return dbConnection
|
|
}
|
|
},
|
|
useDb: function () {
|
|
return instance
|
|
},
|
|
}
|