Export close function to close db

README.md

@@ -45,6 +45,7 @@ import {
   patchPgForTransactions,
   startTransaction,
   rollbackTransaction,
+  close,
 } from 'pg-transactional-tests';
 
 // import instance of your query builder, ORM, something which has `.close` or `.end` or `.destroy` method
@@ -68,7 +69,7 @@ afterAll(async () => {
   // rollback transaction after all tests:
   await rollbackTransaction()
   // end database connection:
-  await db.close()
+  await close()
 });
 ```
 
@@ -81,11 +82,9 @@ import {
   patchPgForTransactions,
   startTransaction,
   rollbackTransaction,
+  close,
 } from 'pg-transactional-tests';
 
-// import instance of your query builder, ORM, something which has `.close` or `.end` or `.destroy` method
-import db from './path-to-your-db'
-
 export const useTestDatabase = () => {
   beforeAll(async () => {
     patchPgForTransactions()
@@ -96,7 +95,7 @@ export const useTestDatabase = () => {
   afterAll(async () => {
     await rollbackTransaction()
     unpatchPgForTransactions()
-    await db.close()
+    await close()
   })
 }
 ```

package.json

@@ -1,6 +1,6 @@
 {
   "name": "pg-transactional-tests",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "description": "Wraps each test in transaction for `pg` package",
   "repository": "https://github.com/romeerez/pg-transactional-tests",
   "main": "dist/index.js",

src/index.ts

@@ -159,3 +159,5 @@ export const rollbackTransaction = async () => {
     await client?.query('ROLLBACK');
   }
 };
+
+export const close = () => client?.end();