packages/docs-core/lib/Services/Metrics/MetricService.spec.ts

   1 import metrics from '@proton/metrics'
   2 import { MetricService } from './MetricService'
   3 import type { Api } from '@proton/shared/lib/interfaces'
   4
   5 jest.mock('@proton/metrics')
   6
   7 describe('MetricService', () => {
   8   let metricService: MetricService
   9
  10   beforeEach(() => {
  11     const api = {} as unknown as Api
  12
  13     metricService = new MetricService(api)
  14
  15     jest.useFakeTimers()
  16   })
  17
  18   afterEach(() => {
  19     jest.useRealTimers()
  20
  21     metricService.destroy()
  22   })
  23
  24   describe('initialize', () => {
  25     it('should heartbeat once immediately', () => {
  26       const spy = jest.spyOn(metricService, 'heartbeat')
  27
  28       metricService.initialize()
  29
  30       expect(spy).toHaveBeenCalled()
  31     })
  32
  33     it('should heartbeat every 60 seconds', () => {
  34       const spy = jest.spyOn(metricService, 'heartbeat')
  35
  36       metricService.initialize()
  37
  38       expect(spy).toHaveBeenCalledTimes(1)
  39
  40       jest.advanceTimersByTime(60_000)
  41
  42       expect(spy).toHaveBeenCalledTimes(2)
  43     })
  44   })
  45
  46   describe('heartbeat', () => {
  47     it('should increment the docs_open_documents_heartbeat_total metric', () => {
  48       metricService.heartbeat()
  49
  50       expect(metrics.docs_open_documents_heartbeat_total.increment).toHaveBeenCalled()
  51     })
  52   })
  53 })