medusa-store/packages/modules/workflow-engine-redis/integration-tests/utils/database.ts

import * as process from "process"

const DB_HOST = process.env.DB_HOST ?? "localhost"
const DB_USERNAME = process.env.DB_USERNAME ?? ""
const DB_PASSWORD = process.env.DB_PASSWORD
const DB_NAME = process.env.DB_TEMP_NAME

export const DB_URL = `postgres://${DB_USERNAME}${
  DB_PASSWORD ? `:${DB_PASSWORD}` : ""
}@${DB_HOST}/${DB_NAME}`

const Redis = require("ioredis")

const redisUrl = process.env.REDIS_URL || "redis://localhost:6379"
const redis = new Redis(redisUrl)

interface TestDatabase {
  clearTables(): Promise<void>
}

export const TestDatabase: TestDatabase = {
  clearTables: async () => {
    await cleanRedis()
  },
}

async function deleteKeysByPattern(pattern) {
  const stream = redis.scanStream({
    match: pattern,
    count: 100,
  })

  const pipeline = redis.pipeline()
  for await (const keys of stream) {
    if (keys.length) {
      keys.forEach((key) => pipeline.unlink(key))
    }
  }
  await pipeline.exec()
}

async function cleanRedis() {
  await deleteKeysByPattern("bull:*")
  await deleteKeysByPattern("dtrx:*")
}