@creator.co/wapi 1.3.19 → 1.4.0-alpha1

package/tests/Database/integrations/pg/PostgresTransaction.test.ts

@@ -1,10 +1,10 @@
-import { PoolClient } from 'pg'
+import { Pool, PoolClient } from 'pg'
 
 import type { DbConfig } from '../../../../src/Database'
 import { PostgresDatabase } from '../../../../src/Database/integrations/pgsql/PostgresDatabase'
 import { PostgresTransactionImpl } from '../../../../src/Database/integrations/pgsql/PostgresTransaction'
 
-const testResources = (config: Partial<DbConfig<'pg'>>) => {
+const testResources = async (config: Partial<DbConfig<'pg'>>) => {
   const database = {
     config: {
       autoCommit: true,
@@ -12,7 +12,7 @@ const testResources = (config: Partial<DbConfig<'pg'>>) => {
     } as DbConfig<'pg'>,
   } as PostgresDatabase
 
-  const delegate = jest.fn().mockImplementation(() => {
+  const transaction = jest.fn().mockImplementation(() => {
     return {
       commit: jest.fn(),
       rollback: jest.fn(),
@@ -20,18 +20,35 @@ const testResources = (config: Partial<DbConfig<'pg'>>) => {
     }
   })() as any as PoolClient
 
+  const writer = jest.fn().mockImplementation(() => {
+    return {
+      connect: jest.fn(() => transaction),
+    }
+  })() as any as Pool
+
   return {
-    delegate,
-    underTest: PostgresTransactionImpl.wrapDelegate(delegate, database),
+    transaction,
+    writer,
+    underTest: await PostgresTransactionImpl.newTransaction(writer, database),
   }
 }
 
 describe('PostgresTransaction', () => {
+  test('Transaction starts opened', async () => {
+    const { transaction, underTest } = await testResources({})
+    expect(underTest.isOpen()).toBe(true)
+    expect(transaction.query).toHaveBeenNthCalledWith(1, 'BEGIN')
+
+    await expect(underTest.begin).rejects.toThrowError(
+      'Cannot begin, transaction is already opened!'
+    )
+  })
+
   test('Can only commit once', async () => {
-    const { delegate, underTest } = testResources({})
+    const { transaction, underTest } = await testResources({})
     await underTest.commit()
     expect(underTest.isOpen()).toBe(false)
-    expect(delegate.query).toBeCalledWith('COMMIT')
+    expect(transaction.query).toBeCalledWith('COMMIT')
 
     await expect(underTest.commit).rejects.toThrowError(
       'Cannot commit, transaction is already closed!'
@@ -39,13 +56,62 @@ describe('PostgresTransaction', () => {
   })
 
   test('Can only rollback once', async () => {
-    const { delegate, underTest } = testResources({})
+    const { transaction, underTest } = await testResources({})
     await underTest.rollback()
     expect(underTest.isOpen()).toBe(false)
-    expect(delegate.query).toBeCalledWith('ROLLBACK')
+    expect(transaction.query).toBeCalledWith('ROLLBACK')
 
     await expect(underTest.rollback).rejects.toThrowError(
       'Cannot rollback, transaction is already closed!'
     )
   })
+
+  test("Can't open transaction twice", async () => {
+    const { transaction, underTest } = await testResources({})
+    expect(underTest.isOpen()).toBe(true)
+    expect(transaction.query).toHaveBeenNthCalledWith(1, 'BEGIN')
+
+    await expect(underTest.begin).rejects.toThrowError(
+      'Cannot begin, transaction is already opened!'
+    )
+  })
+
+  test('Can commit, begin and commit again ', async () => {
+    const { transaction, underTest } = await testResources({})
+    await underTest.commit()
+    expect(underTest.isOpen()).toBe(false)
+    expect(transaction.query).toBeCalledWith('COMMIT')
+    await underTest.begin()
+    expect(underTest.isOpen()).toBe(true)
+    expect(transaction.query).toHaveBeenNthCalledWith(3, 'BEGIN')
+    await underTest.commit()
+    expect(underTest.isOpen()).toBe(false)
+    expect(transaction.query).toHaveBeenNthCalledWith(4, 'COMMIT')
+  })
+
+  test('Can commit, begin and rollback ', async () => {
+    const { transaction, underTest } = await testResources({})
+    await underTest.commit()
+    expect(underTest.isOpen()).toBe(false)
+    expect(transaction.query).toBeCalledWith('COMMIT')
+    await underTest.begin()
+    expect(underTest.isOpen()).toBe(true)
+    expect(transaction.query).toHaveBeenNthCalledWith(3, 'BEGIN')
+    await underTest.rollback()
+    expect(underTest.isOpen()).toBe(false)
+    expect(transaction.query).toHaveBeenNthCalledWith(4, 'ROLLBACK')
+  })
+
+  test('Can rollback, begin and commit ', async () => {
+    const { transaction, underTest } = await testResources({})
+    await underTest.rollback()
+    expect(underTest.isOpen()).toBe(false)
+    expect(transaction.query).toBeCalledWith('ROLLBACK')
+    await underTest.begin()
+    expect(underTest.isOpen()).toBe(true)
+    expect(transaction.query).toHaveBeenNthCalledWith(3, 'BEGIN')
+    await underTest.commit()
+    expect(underTest.isOpen()).toBe(false)
+    expect(transaction.query).toHaveBeenNthCalledWith(4, 'COMMIT')
+  })
 })

package/tsconfig.json

@@ -15,6 +15,7 @@
     "target": "es5",
     "outDir": "dist",
     "declaration": true,
+    "skipLibCheck": true,
     "strictNullChecks": true
   }
 }