import { MongoMemoryServer } from "mongodb-memory-server";
import { MongoClient, Db } from "mongodb";
import {
  setDb,
  setMunicipalDataDb,
  setWebsitalkDb,
  setConfigurationDb,
} from "../index";

let mongoServer: MongoMemoryServer;
let client: MongoClient;

export async function initTestDb(): Promise<Db> {
  mongoServer = await MongoMemoryServer.create({
    instance: { ip: "127.0.0.1" },
  });
  client = new MongoClient(mongoServer.getUri());
  await client.connect();
  const talkpilotDb = client.db();
  const municipalDb = client.db("municipal-data");
  const websitalkDb = client.db("website-talk");
  const configurationDb = client.db("configuration");
  setDb(talkpilotDb);
  setMunicipalDataDb(municipalDb);
  setWebsitalkDb(websitalkDb);
  setConfigurationDb(configurationDb);
  return talkpilotDb;
}

export async function closeTestDb(): Promise<void> {
  if (client) await client.close();
  if (mongoServer) await mongoServer.stop();
}