asyavee/medusa-store
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

use truncate in db teardown (#13875)

Browse Source 
* use truncate in db teardown

* fix empty table list condition
This commit is contained in:
Pedro Guzman
2025-10-27 18:39:48 +01:00
committed by GitHub
parent 39f2893738
commit cf1001f11d
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
packages/medusa-test-utils/src/database.ts
View File

@@ -228,7 +228,6 @@ export const dbTestUtilFactory = (): any => ({
const runRawQuery = this.pgConnection_.raw.bind(this.pgConnection_)
schema ??= "public"
await runRawQuery(`SET session_replication_role = 'replica';`)
const { rows: tableNames } = await runRawQuery(`SELECT table_name
FROM information_schema.tables
WHERE table_schema = '${schema}';`)
@@ -237,6 +236,7 @@ export const dbTestUtilFactory = (): any => ({
const mainPartitionTables = ["index_data", "index_relation"]
let hasIndexTables = false
const tablesToTruncate: string[] = []
for (const { table_name } of tableNames) {
if (mainPartitionTables.includes(table_name)) {
hasIndexTables = true
@@ -249,15 +249,17 @@ export const dbTestUtilFactory = (): any => ({
continue
}
await runRawQuery(`DELETE FROM ${schema}."${table_name}";`)
tablesToTruncate.push(`${schema}."${table_name}"`)
}
if (tablesToTruncate.length > 0) {
await runRawQuery(`TRUNCATE ${tablesToTruncate.join(", ")};`)
}
if (hasIndexTables) {
await runRawQuery(`TRUNCATE TABLE ${schema}.index_data;`)
await runRawQuery(`TRUNCATE TABLE ${schema}.index_relation;`)
await runRawQuery(
`TRUNCATE ${schema}.index_data, ${schema}.index_relation;`
)
}
await runRawQuery(`SET session_replication_role = 'origin';`)
} catch (error) {
logger.error("Error during database teardown:", error)
throw error