41 lines
1.5 KiB
TypeScript
41 lines
1.5 KiB
TypeScript
|
import { Logger } from '@nestjs/common';
|
||
|
|
import { Client } from 'pg';
|
||
|
|
const logger = new Logger();
|
||
|
|
|
||
|
|
// This is a very simple helper to make sure that the tests do not overwrite the local development database.
|
||
|
|
export const createTestDatabaseSetup = async () => {
|
||
|
|
const uri: string = process.env[`STORAGE_URI`];
|
||
|
|
|
||
|
|
// This is all very specific to the docker-compose.yml. We should refactor this later. This will break when query parameters are added:
|
||
|
|
const parts = uri.split('/');
|
||
|
|
if (parts.length < 1) {
|
||
|
|
logger.error('No database match found');
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
const databaseSuffix = '-test';
|
||
|
|
const database = parts[parts.length - 1] + databaseSuffix;
|
||
|
|
process.env['STORAGE_URI'] = process.env['STORAGE_URI'] + databaseSuffix;
|
||
|
|
|
||
|
|
// Set up client with uri:
|
||
|
|
const client = new Client({ connectionString: uri });
|
||
|
|
client.connect();
|
||
|
|
|
||
|
|
return client
|
||
|
|
.query(`CREATE DATABASE "${database}"`)
|
||
|
|
.catch((e) => logger.debug('Error executing query:', e)) // this will happen after the first run, so we'll discard it
|
||
|
|
.then(() => client.end());
|
||
|
|
};
|
||
|
|
|
||
|
|
// This helper is required for testing the postgres ttl service. It is designed specificially to clean up after each postgres spec.
|
||
|
|
// Therefore, we don't want to include it in other specs.
|
||
|
|
export const clearDatabase = async () => {
|
||
|
|
const uri: string = process.env[`STORAGE_URI`];
|
||
|
|
// Set up client with uri:
|
||
|
|
const client = new Client({ connectionString: uri });
|
||
|
|
client.connect();
|
||
|
|
return client
|
||
|
|
.query('DELETE FROM keyv;')
|
||
|
|
.catch((e) => console.error(e.stack))
|
||
|
|
.then(() => client.end());
|
||
|
|
};
|