Function createMockCollection

packages/angular-db/tests/inject-live-query.test.ts:67–165

(
  initial: Array<T & Record<`id`, K>> = [],
  initialStatus: CollectionStatus = `ready`,
)

}

function createMockCollection<T extends object, K extends string | number>(
  initial: Array<T & Record<`id`, K>> = [],
  initialStatus: CollectionStatus = `ready`,
): Collection<T, K, Record<string, never>> &
  NonSingleResult & {
    __setStatus: (s: CollectionStatus) => void
    __replaceAll: (rows: Array<T & Record<`id`, K>>) => void
    __upsert: (row: T & Record<`id`, K>) => void
    __delete: (key: K) => void
  } {
  const map = new Map<K, T>()
  for (const r of initial) {
    map.set(r.id, r)
  }

  let status: CollectionStatus = initialStatus
  const subs = new Set<(changes: Array<any>) => void>()
  const readySubs = new Set<() => void>()
  const id = `mock-col-` + Math.random().toString(36).slice(2)

  const notify = (changes: Array<any> = []) => {
    for (const cb of subs) cb(changes)
  }

  const notifyReady = () => {
    for (const cb of readySubs) cb()
  }

  const api: any = {
    id,
    get status() {
      return status
    },
    entries: () => Array.from(map.entries()),
    values: () => Array.from(map.values()),
    get: (key: K) => map.get(key),
    has: (key: K) => map.has(key),
    size: () => map.size,
    subscribeChanges: (cb: (changes: Array<any>) => void) => {
      subs.add(cb)
      return {
        unsubscribe: () => subs.delete(cb),
      }
    },
    onFirstReady: (cb: () => void) => {
      if (status === `ready`) {
        setTimeout(cb, 0)
      } else {
        readySubs.add(cb)
      }
      return () => readySubs.delete(cb)
    },
    preload: () => Promise.resolve(),
    startSyncImmediate: () => {
      const wasNotReady = status !== `ready`
      if (status === `idle`) {
        status = `ready`
      }