stream-chat-react-native-core/src/mock-builders/DB/mock.ts

The official React Native and Expo components for Stream Chat, a service for building chat applications

import { rmSync } from 'fs';

import Sqlite3 from 'better-sqlite3';

import type { PreparedQueries } from '../../store/types';

let db: Sqlite3.Database;
const testDbName = `foobar-${process.env.JEST_WORKER_ID ?? '0'}.db`;

export const sqliteMock = {
  // better-sqlite3 has no SQLCipher, so an `encryptionKey` passed to open() is
  // simply ignored. Reporting a SQLCipher build keeps the encrypted path
  // exercisable in tests; whether the bytes on disk are actually encrypted can
  // only be verified on a device. Spy on this to test the build-missing guard.
  isSQLCipher: () => true,
  open: () => {
    db = new Sqlite3(testDbName);
    db.pragma('journal_mode = MEMORY');
    db.pragma('synchronous = OFF');
    return {
      close: () => {
        db.close();
        return {
          message: '',
          status: 0,
        };
      },
      // Mirrors op-sqlite's delete(): closes the handle and unlinks the file, so a
      // subsequent open() starts from an empty database.
      delete: () => {
        db.close();
        rmSync(testDbName, { force: true });
      },
      execute: async (queryInput: string, params: unknown[]) => {
        const query = queryInput.trim().toLowerCase();

        const stmt = db.prepare(query);
        let result: unknown[] = [];
        if (query.indexOf('select') === 0) {
          const modifiedParams = params?.map((p) => (typeof p === 'boolean' ? Number(p) : p)) || [];
          result = await new Promise((resolve) => resolve(stmt.all(modifiedParams)));

          return {
            message: '',
            rows: result,
            status: 0,
          };
        }

        if (query.indexOf('pragma') === 0) {
          const pragmaQueryTokens = query.split(' ');
          if (pragmaQueryTokens[2] === '=') {
            db.pragma(`${pragmaQueryTokens[1]} = ${pragmaQueryTokens[3]}`);
          } else {
            result = db.pragma(`${pragmaQueryTokens[1]}`) as unknown[];
          }

          return {
            message: '',
            rows: result,
            status: 0,
          };
        }

        // insert or create table query
        await new Promise((resolve) => {
          if (params) {
            const modifiedParams = params.map((p) => (typeof p === 'boolean' ? Number(p) : p));
            stmt.run(modifiedParams);
          } else {
            stmt.run();
          }
          resolve(undefined);
        });

        return {
          message: '',
          rows: result,
          status: 0,
        };
      },
      executeBatch: async (queriesArr: PreparedQueries[]) => {
        for (const queryAndParams of queriesArr) {
          const query = queryAndParams[0];
          const params = queryAndParams[1];
          const stmt = db.prepare(query);

          await new Promise((resolve) => {
            if (params) {
              const modifiedParams = params.map((p) => (typeof p === 'boolean' ? Number(p) : p));
              stmt.run(modifiedParams);
            } else {
              stmt.run();
            }
            resolve(undefined);
          });
        }

        return {
          message: '',
          staus: 0,
        };
      },
    };
  },
};