Function createMockDb

packages/core/src/query/__tests__/session-queries.test.ts:52–79

* Create a mock DatabaseAdapter whose prepare().get()/all() respond based on SQL patterns.

(handlers: Record<string, (args: unknown[]) => QueryResult | QueryResult[]>)

 * Create a mock DatabaseAdapter whose prepare().get()/all() respond based on SQL patterns.
 */
function createMockDb(handlers: Record<string, (args: unknown[]) => QueryResult | QueryResult[]>): DatabaseAdapter {
  return {
    prepare(sql: string) {
      return {
        get(...args: unknown[]) {
          for (const [pattern, handler] of Object.entries(handlers)) {
            if (sql.includes(pattern)) {
              return handler(args)
            }
          }
          return undefined
        },
        all(...args: unknown[]) {
          for (const [pattern, handler] of Object.entries(handlers)) {
            if (sql.includes(pattern)) {
              const result = handler(args)
              return Array.isArray(result) ? result : [result]
            }
          }
          return []
        },
        run() {
          return { changes: 0, lastInsertRowid: 0 }
        },
      }
    },
  } as unknown as DatabaseAdapter
}

// ==================== Tests ====================