index.ts

import Vue, { nextTick } from 'vue'
import { MockFirebase, MockedReference } from 'firebase-mock'
import firebase from '../../../../node_modules/firebase/index'
import { walkSet } from '@posva/vuefire-core'

export { Vue, MockFirebase, MockedReference }

export * from './mock'

type FirestoreReference =
  | firebase.firestore.CollectionReference
  | firebase.firestore.DocumentReference
  | firebase.firestore.Query

export function spyUnbind(ref: FirestoreReference) {
  const unbindSpy = jest.fn()
  const onSnapshot = ref.onSnapshot.bind(ref)
  ref.onSnapshot =
    // @ts-ignore
    fn => {
      // @ts-ignore
      const unbind = onSnapshot(fn)
      return () => {
        unbindSpy()
        unbind()
      }
    }
  return unbindSpy
}

export function spyOnSnapshot(ref: FirestoreReference) {
  const onSnapshot = ref.onSnapshot.bind(ref)
  // @ts-ignore
  return (ref.onSnapshot = jest.fn((...args) => onSnapshot(...args)))
}

export function spyOnSnapshotCallback(ref: FirestoreReference) {
  const onSnapshot = ref.onSnapshot.bind(ref)
  const spy = jest.fn()
  ref.onSnapshot = (fn: any) =>
    // @ts-ignore
    onSnapshot((...args) => {
      spy()
      fn(...args)
    })
  return spy
}

// This makes sure some tests fail by delaying callbacks
export function delayUpdate(ref: firebase.firestore.DocumentReference, time = 0) {
  const onSnapshot = ref.onSnapshot.bind(ref)
  // @ts-ignore
  ref.onSnapshot = fn =>
    // @ts-ignore
    onSnapshot(async (...args) => {
      await delay(time)
      fn(...args)
    })
}

export function tick() {
  return new Promise(resolve => {
    nextTick(resolve)
  })
}

export function delay(time: number) {
  return new Promise(resolve => setTimeout(resolve, time))
}

type WalkSet = typeof walkSet
export const createOps = (localWalkSet: WalkSet = walkSet) => ({
  add: jest.fn((array, index, data) => array.splice(index, 0, data)),
  set: jest.fn(localWalkSet),
  remove: jest.fn((array, index) => array.splice(index, 1)),
})