expo-sqlite-mock 2.1.0 → 3.0.0

package/README.md

@@ -7,7 +7,8 @@ Use [expo-sqlite](https://docs.expo.dev/versions/latest/sdk/sqlite/) with jest.
 
 ## Notice
 
-- **~2.0.0** is for expo-sqlite >=52.
+- **~3.0.0** is for expo-sqlite ~53.
+- **~2.0.0** is for expo-sqlite ~52.
 - **~1.0.0** is for expo-sqlite ~51.
 
 ## Usage
@@ -52,6 +53,13 @@ it("test", async () => {
 
 ## Changelog
 
+- **3.0.0**
+  - Compatible with expo-sqlite ~15 and expo ~53.
+
+- **2.2.0**
+  - Compatible with drizzle
+  - Clean up the code
+
 - **2.1.0**
   - Support custom SQLite database location by setting the `EXPO_SQLITE_MOCK` environment variable.
   - Update peer dependencies.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "expo-sqlite-mock",
-  "version": "2.1.0",
+  "version": "3.0.0",
   "license": "MIT",
   "homepage": "https://github.com/zfben/expo-sqlite-mock",
   "repository": {
@@ -24,9 +24,9 @@
   ],
   "peerDependencies": {
     "better-sqlite3": "*",
-    "jest": "*",
+    "jest": "~29",
     "expo-sqlite": "*",
-    "expo": ">=52"
+    "expo": ">=53"
   },
   "devDependencies": {
     "typescript": "*",
@@ -34,13 +34,14 @@
     "expo-sqlite": "*",
     "better-sqlite3": "*",
     "@types/better-sqlite3": "*",
-    "@types/react": "^18",
-    "react": "^18",
-    "react-test-renderer": "^18",
-    "react-native": "~0.77",
-    "@testing-library/react-native": "*",
-    "jest-expo": "~52",
-    "@types/jest": "*",
-    "jest": "*"
+    "@types/react": "*",
+    "react": "*",
+    "react-test-renderer": "*",
+    "react-native": "~0.80",
+    "@testing-library/react-native": "~13",
+    "jest-expo": "~53",
+    "@types/jest": "~29",
+    "jest": "~29",
+    "drizzle-orm": "*"
   }
 }

package/src/ExpoSQLiteNext.ts

@@ -1,15 +1,15 @@
-import assert from 'node:assert'
 import sqlite3 from 'better-sqlite3'
 
-import type { SQLiteOpenOptions } from 'expo-sqlite/src/NativeDatabase'
 import type {
-  SQLiteBindBlobParams,
+  SQLiteBindValue,
+  SQLiteOpenOptions,
   SQLiteBindParams,
-  SQLiteBindPrimitiveParams,
-  SQLiteColumnNames,
-  SQLiteColumnValues,
-  SQLiteRunResult,
-} from 'expo-sqlite/src/NativeStatement'
+  SQLiteRunResult, } from 'expo-sqlite'
+
+type SQLiteBindBlobParams = Record<string, Uint8Array>
+type SQLiteBindPrimitiveParams = Record<string, Exclude<SQLiteBindValue, Uint8Array>>
+type SQLiteColumnNames = string[]
+type SQLiteColumnValues = any[]
 
 const dbs: NativeDatabase[] = []
 
@@ -18,19 +18,18 @@ export const mockedExpoSqliteNext = {
     for (const db of dbs) await db.closeAsync()
   },
 
-  NativeDatabase: jest
-    .fn()
-    .mockImplementation(
-      (databaseName: string, options?: SQLiteOpenOptions, serializedData?: Uint8Array) =>
-        new NativeDatabase(databaseName, options, serializedData)
-    ),
+  NativeDatabase: (
+    databaseName: string,
+    options?: SQLiteOpenOptions,
+    serializedData?: Uint8Array
+  ) => new NativeDatabase(databaseName, options, serializedData),
 
-  NativeStatement: jest.fn().mockImplementation(() => new NativeStatement()),
+  NativeStatement: () => new NativeStatement(),
 
   defaultDatabaseDirectory: '.',
 
-  ensureDatabasePathExistsAsync: jest.fn().mockImplementation(async (databasePath: string) => {}),
-  ensureDatabasePathExistsSync: jest.fn().mockImplementation((databasePath: string) => {}),
+  ensureDatabasePathExistsAsync: async (_databasePath: string) => {},
+  ensureDatabasePathExistsSync: (_databasePath: string) => {},
 }
 
 //#region async sqlite3
@@ -41,7 +40,11 @@ export const mockedExpoSqliteNext = {
 class NativeDatabase {
   private readonly sqlite3Db: sqlite3.Database
 
-  constructor(databaseName: string, options?: SQLiteOpenOptions, serializedData?: Uint8Array) {
+  constructor(
+    _databaseName: string,
+    _options?: SQLiteOpenOptions,
+    serializedData?: Uint8Array
+  ) {
     if (serializedData != null) {
       this.sqlite3Db = new sqlite3(Buffer.from(serializedData))
     } else {
@@ -52,11 +55,12 @@ class NativeDatabase {
 
   //#region Asynchronous API
 
-  initAsync = jest.fn().mockResolvedValue(null)
+  initAsync = () => Promise.resolve()
   isInTransactionAsync = async () => this.sqlite3Db.inTransaction
   closeAsync = async () => this.sqlite3Db.close()
   execAsync = async (source: string) => this.sqlite3Db.exec(source)
-  serializeAsync = async (databaseName: string) => this.sqlite3Db.serialize({ attached: databaseName })
+  serializeAsync = async (databaseName: string) =>
+    this.sqlite3Db.serialize({ attached: databaseName })
   prepareAsync = async (nativeStatement: NativeStatement, source: string) => {
     nativeStatement.sqlite3Stmt = this.sqlite3Db.prepare(source)
   }
@@ -65,11 +69,12 @@ class NativeDatabase {
 
   //#region Synchronous API
 
-  initSync = jest.fn()
+  initSync = () => {}
   isInTransactionSync = () => this.sqlite3Db.inTransaction
   closeSync = () => this.sqlite3Db.close()
   execSync = (source: string) => this.sqlite3Db.exec(source)
-  serializeSync = (databaseName: string) => this.sqlite3Db.serialize({ attached: databaseName })
+  serializeSync = (databaseName: string) =>
+    this.sqlite3Db.serialize({ attached: databaseName })
   prepareSync = (nativeStatement: NativeStatement, source: string) => {
     nativeStatement.sqlite3Stmt = this.sqlite3Db.prepare(source)
   }
@@ -82,40 +87,43 @@ class NativeDatabase {
 class NativeStatement {
   public sqlite3Stmt: sqlite3.Statement | null = null
   private iterator: ReturnType<sqlite3.Statement['iterate']> | null = null
+  private cachedRows: any[] = []
 
   //#region Asynchronous API
 
-  public runAsync = jest
-    .fn()
-    .mockImplementation(
-      (
-        database: NativeDatabase,
-        bindParams: SQLiteBindPrimitiveParams,
-        bindBlobParams: SQLiteBindBlobParams,
-        shouldPassAsArray: boolean
-      ): Promise<SQLiteRunResult & { firstRowValues: SQLiteColumnValues }> =>
-        Promise.resolve(this._run(normalizeSQLite3Args(bindParams, bindBlobParams, shouldPassAsArray)))
+  public runAsync = (
+    _database: NativeDatabase,
+    bindParams: SQLiteBindPrimitiveParams,
+    bindBlobParams: SQLiteBindBlobParams,
+    shouldPassAsArray: boolean
+  ): Promise<SQLiteRunResult & { firstRowValues: SQLiteColumnValues }> =>
+    Promise.resolve(
+      this._run(
+        normalizeSQLite3Args(bindParams, bindBlobParams, shouldPassAsArray)
+      )
     )
-  public stepAsync = jest.fn().mockImplementation((database: NativeDatabase): Promise<any> => {
-    assert(this.sqlite3Stmt)
+  public stepAsync = (_database: NativeDatabase): Promise<any> => {
     if (this.iterator == null) {
       this.iterator = this.sqlite3Stmt.iterate()
       // Since the first row is retrieved by `_run()`, we need to skip the first row here.
       this.iterator.next()
     }
     const result = this.iterator.next()
-    const columnValues = result.done === false ? Object.values(result.value as Record<string, any>) : null
+    const columnValues =
+      result.done === false
+        ? Object.values(result.value as Record<string, any>)
+        : null
     return Promise.resolve(columnValues)
-  })
-  public getAllAsync = (database: NativeDatabase) => Promise.resolve(this._allValues())
-  public getColumnNamesAsync = async (database: NativeDatabase) => {
-    assert(this.sqlite3Stmt)
+  }
+  public getAllAsync = (_database: NativeDatabase) =>
+    Promise.resolve(this._allValues())
+  public getColumnNamesAsync = async (_database: NativeDatabase) => {
     return this.sqlite3Stmt.columns().map(column => column.name)
   }
-  public resetAsync = async (database: NativeDatabase) => {
+  public resetAsync = async (_database: NativeDatabase) => {
     this._reset()
   }
-  public finalizeAsync = async (database: NativeDatabase) => {
+  public finalizeAsync = async (_database: NativeDatabase) => {
     this._finalize()
   }
 
@@ -123,19 +131,17 @@ class NativeStatement {
 
   //#region Synchronous API
 
-  public runSync = jest
-    .fn()
-    .mockImplementation(
-      (
-        database: NativeDatabase,
-        bindParams: SQLiteBindPrimitiveParams,
-        bindBlobParams: SQLiteBindBlobParams,
-        shouldPassAsArray: boolean
-      ): SQLiteRunResult & { firstRowValues: SQLiteColumnValues } =>
-        this._run(normalizeSQLite3Args(bindParams, bindBlobParams, shouldPassAsArray))
+  public runSync = (
+    _database: NativeDatabase,
+    bindParams: SQLiteBindPrimitiveParams,
+    bindBlobParams: SQLiteBindBlobParams,
+    shouldPassAsArray: boolean
+  ): SQLiteRunResult & { firstRowValues: SQLiteColumnValues } =>
+    this._run(
+      normalizeSQLite3Args(bindParams, bindBlobParams, shouldPassAsArray)
     )
-  public stepSync = (database: NativeDatabase): any => {
-    assert(this.sqlite3Stmt)
+
+  public stepSync = (_database: NativeDatabase): any => {
     if (this.iterator == null) {
       this.iterator = this.sqlite3Stmt.iterate()
       // Since the first row is retrieved by `_run()`, we need to skip the first row here.
@@ -143,33 +149,48 @@ class NativeStatement {
     }
 
     const result = this.iterator.next()
-    const columnValues = result.done === false ? Object.values(result.value as Record<string, any>) : null
+    const columnValues =
+      result.done === false
+        ? Object.values(result.value as Record<string, any>)
+        : null
     return columnValues
   }
-  public getAllSync = (database: NativeDatabase) => this._allValues()
-  public getColumnNamesSync = (database: NativeDatabase) => {
-    assert(this.sqlite3Stmt)
+  public getAllSync = (_database: NativeDatabase) => this._allValues()
+  public getColumnNamesSync = (_database: NativeDatabase) => {
     return this.sqlite3Stmt.columns().map(column => column.name)
   }
-  public resetSync = (database: NativeDatabase) => {
+  public resetSync = (_database: NativeDatabase) => {
     this._reset()
   }
-  public finalizeSync = (database: NativeDatabase) => {
+  public finalizeSync = (_database: NativeDatabase) => {
     this._finalize()
   }
 
   //#endregion
 
-  private _run = (...params: any[]): SQLiteRunResult & { firstRowValues: SQLiteColumnValues } => {
-    assert(this.sqlite3Stmt)
+  private _run = (
+    ...params: any[]
+  ): SQLiteRunResult & { firstRowValues: SQLiteColumnValues } => {
     this.sqlite3Stmt.bind(...params)
+
+    // Avoid insert with returning being executed multiple times
+    if (this.sqlite3Stmt.source.includes(' returning ')) {
+      this.cachedRows = this.sqlite3Stmt.all()
+
+      return {
+        lastInsertRowId: this.cachedRows[0].id,
+        changes: params.length,
+        firstRowValues: Object.values(this.cachedRows[0]),
+      }
+    }
+
     const result = this.sqlite3Stmt.run()
 
     // better-sqlite3 does not support run() returning the first row, use get() instead.
     let firstRow: any
     try {
       firstRow = this.sqlite3Stmt.get()
-    } catch {
+    } catch (_) {
       // better-sqlite3 may throw `TypeError: This statement does not return data. Use run() instead`
       firstRow = null
     }
@@ -181,7 +202,9 @@ class NativeStatement {
   }
 
   private _allValues = (): SQLiteColumnNames[] => {
-    assert(this.sqlite3Stmt)
+    if (this.sqlite3Stmt.source.includes(' returning '))
+      return this.cachedRows.slice(1).map(row => Object.values(row))
+
     const sqlite3Stmt = this.sqlite3Stmt as any
     // Since the first row is retrieved by `_run()`, we need to skip the first row here.
     return sqlite3Stmt
@@ -191,7 +214,6 @@ class NativeStatement {
   }
 
   private _reset = () => {
-    assert(this.sqlite3Stmt)
     this.iterator?.return?.()
     this.iterator = this.sqlite3Stmt.iterate()
   }