Created
October 1, 2014 09:52
-
-
Save bendrucker/cf298bf515d8b6fdcd05 to your computer and use it in GitHub Desktop.
Truncate data during tests with knex
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
'use strict'; | |
var Promise = require('bluebird'); | |
var knex = require('../../src/db').knex; | |
var tables = [ | |
'organizations', | |
'campaigns', | |
'donors', | |
'pledges', | |
'payments' | |
]; | |
function truncate () { | |
return Promise.each(tables, function (table) { | |
return knex.raw('truncate table ' + table + ' cascade'); | |
}); | |
}; | |
function seed () { | |
return Promise.each(tables, function (table) { | |
return knex(table).insert(require('./seeds/' + table)); | |
}); | |
}; | |
describe('Integration Tests', function () { | |
beforeEach(function () { | |
return truncate().then(seed); | |
}); | |
afterEach(function () { | |
return truncate(); | |
}); | |
require('require-all')(__dirname + '/specs'); | |
}); |
My solution, truncates all tables in schemas
// ./truncateSchemas.js
export async function truncateSchemas(
knexClient,
schemas
) {
if (schemas.length < 1) {
throw new Error(`Expected schemas be non-emtpy array, but got ${schemas}`)
}
await knexClient.raw(
`
DO
$func$
BEGIN
EXECUTE
(
SELECT 'TRUNCATE TABLE ' || string_agg(format('%I.%I', table_schema, table_name), ', ') || ' RESTART IDENTITY CASCADE'
FROM information_schema.tables
WHERE table_schema IN (${schemas.map(x => `'${x}'`).join(', ')})
AND table_type = 'BASE TABLE'
);
END
$func$;
`
)
}
// jest.config.js
module.exports = {
transform: {
'^.+\\.jsx?$': 'babel-jest',
},
setupTestFrameworkScriptFile: '<rootDir>/jest/index.js',
testPathIgnorePatterns: ['/node_modules/'],
}
// jest/index.js
import knex from 'knex'
import { truncateSchemas } from './truncateSchemas'
const knexClient = knex({
client: 'pg',
connection: {
user: process.env.POSTGRES_USER,
password: process.env.POSTGRES_PASSWORD,
host: process.env.POSTGRES_HOST,
port: process.env.POSTGRES_PORT,
database: process.env.POSTGRES_DB,
}
})
beforeEach(async () => {
await truncateSchemas(knexClient, ['app_public', 'app_private', 'app_hidden'])
})
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Thanks for sharing your integration test code!
It is also recommended to update the test database to the latest schema before running any test: