refactor: restructure test db setup

Author: chrispader

example/src/tests/db.ts

@@ -4,21 +4,33 @@ import type {
   BatchQueryCommand,
 } from 'react-native-nitro-sqlite'
 import { open } from 'react-native-nitro-sqlite'
+import {
+  getDatabaseQueue,
+  type DatabaseQueue,
+} from '../../../package/src/DatabaseQueue'
 
 const chance = new Chance()
 
 export const TEST_DB_NAME = 'test'
 
-export let testDb: NitroSQLiteConnection | undefined
+export let testDb: NitroSQLiteConnection
+export let testDbQueue: DatabaseQueue
 export function resetTestDb() {
   try {
     if (testDb != null) {
       testDb.close()
       testDb.delete()
     }
+
     testDb = open({
       name: TEST_DB_NAME,
     })
+    testDbQueue = getDatabaseQueue(TEST_DB_NAME)
+
+    testDb.execute('DROP TABLE IF EXISTS User;')
+    testDb.execute(
+      'CREATE TABLE User ( id REAL PRIMARY KEY, name TEXT NOT NULL, age REAL, networth REAL) STRICT;',
+    )
   } catch (e) {
     console.warn('Error resetting user database', e)
   }

example/src/tests/unit/common.ts

@@ -1,9 +1,6 @@
 import { Chance } from 'chance'
-import {
-  NitroSQLiteConnection,
-  enableSimpleNullHandling,
-} from 'react-native-nitro-sqlite'
-import { testDb as testDbInternal, resetTestDb } from '../db'
+import { enableSimpleNullHandling } from 'react-native-nitro-sqlite'
+import { resetTestDb } from '../db'
 import chai from 'chai'
 
 export function isError(e: unknown): e is Error {
@@ -13,23 +10,7 @@ export function isError(e: unknown): e is Error {
 export const expect = chai.expect
 export const chance = new Chance()
 
-export let testDb: NitroSQLiteConnection
-
 export function setupTestDb() {
   enableSimpleNullHandling(false)
-
-  try {
-    resetTestDb()
-
-    if (testDbInternal == null) throw new Error('Failed to reset test database')
-
-    testDbInternal.execute('DROP TABLE IF EXISTS User;')
-    testDbInternal.execute(
-      'CREATE TABLE User ( id REAL PRIMARY KEY, name TEXT NOT NULL, age REAL, networth REAL) STRICT;',
-    )
-
-    testDb = testDbInternal!
-  } catch (e) {
-    console.warn('Error resetting user database', e)
-  }
+  resetTestDb()
 }

example/src/tests/unit/specs/execute.spec.ts

@@ -1,9 +1,10 @@
-import { chance, expect, isError, testDb } from '../common'
+import { chance, expect, isError } from '../common'
 import {
   enableSimpleNullHandling,
   NITRO_SQLITE_NULL,
 } from 'react-native-nitro-sqlite'
 import { describe, it } from '../../MochaRNAdapter'
+import { testDb } from '../../db'
 
 export default function registerExecuteUnitTests() {
   describe('execute', () => {

example/src/tests/unit/specs/executeBatch.spec.ts

@@ -1,6 +1,7 @@
-import { chance, expect, testDb } from '../common'
+import { chance, expect } from '../common'
 import type { BatchQueryCommand } from 'react-native-nitro-sqlite'
 import { describe, it } from '../../MochaRNAdapter'
+import { testDb } from '../../db'
 
 export default function registerExecuteBatchUnitTests() {
   describe('executeBatch', () => {

example/src/tests/unit/specs/transaction.spec.ts

@@ -1,6 +1,7 @@
-import { chance, expect, isError, testDb } from '../common'
+import { chance, expect, isError } from '../common'
 import { describe, it } from '../../MochaRNAdapter'
 import type { User } from '../../../model/User'
+import { testDb, testDbQueue } from '../../db'
 
 export default function registerTransactionUnitTests() {
   describe('transaction', () => {

package/src/DatabaseQueue.ts

@@ -1,13 +1,13 @@
 import NitroSQLiteError from './NitroSQLiteError'
 
-interface QueuedOperation {
+export interface QueuedOperation {
   /**
    * Starts the operation
    */
   start: () => void
 }
 
-type DatabaseQueue = {
+export type DatabaseQueue = {
   queue: QueuedOperation[]
   inProgress: boolean
 }
@@ -47,7 +47,7 @@ export function throwIfDatabaseIsNotOpen(dbName: string) {
     )
 }
 
-function getDatabaseQueue(dbName: string) {
+export function getDatabaseQueue(dbName: string) {
   throwIfDatabaseIsNotOpen(dbName)
 
   const queue = databaseQueues.get(dbName)!