const RealDate = global.Date

export function mockDate() {
  global.Date = new Proxy(RealDate, handler)
}

export function unmockDate() {
  global.Date = RealDate
}

let now = Date.now()

export function advanceTimeBy(ms: number) {
  now += ms
}

const handler: ProxyHandler<any> = {
  construct(target, args) {
    if (args.length === 0) {
      return new Date(now)
    } else {
      return new target(...args)
    }
  },
  get(target, propKey) {
    if (propKey === 'now') {
      return () => now
    } else {
      return target[propKey]
    }
  },
}