server/tests/api/server/open-telemetry.ts

   1 /* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */
   2
   3 import { expect } from 'chai'
   4 import { expectLogContain, expectLogDoesNotContain, MockHTTP } from '@server/tests/shared'
   5 import { HttpStatusCode, VideoPrivacy } from '@shared/models'
   6 import { cleanupTests, createSingleServer, makeRawRequest, PeerTubeServer, setAccessTokensToServers } from '@shared/server-commands'
   7
   8 describe('Open Telemetry', function () {
   9   let server: PeerTubeServer
  10
  11   describe('Metrics', function () {
  12     const metricsUrl = 'http://localhost:9091/metrics'
  13
  14     it('Should not enable open telemetry metrics', async function () {
  15       server = await createSingleServer(1)
  16
  17       let hasError = false
  18       try {
  19         await makeRawRequest(metricsUrl, HttpStatusCode.NOT_FOUND_404)
  20       } catch (err) {
  21         hasError = err.message.includes('ECONNREFUSED')
  22       }
  23
  24       expect(hasError).to.be.true
  25
  26       await server.kill()
  27     })
  28
  29     it('Should enable open telemetry metrics', async function () {
  30       server = await createSingleServer(1, {
  31         open_telemetry: {
  32           metrics: {
  33             enabled: true
  34           }
  35         }
  36       })
  37
  38       const res = await makeRawRequest(metricsUrl, HttpStatusCode.OK_200)
  39       expect(res.text).to.contain('peertube_job_queue_total')
  40
  41       await server.kill()
  42     })
  43   })
  44
  45   describe('Tracing', function () {
  46     let mockHTTP: MockHTTP
  47     let mockPort: number
  48
  49     before(async function () {
  50       mockHTTP = new MockHTTP()
  51       mockPort = await mockHTTP.initialize()
  52     })
  53
  54     it('Should enable open telemetry tracing', async function () {
  55       server = await createSingleServer(1)
  56
  57       await expectLogDoesNotContain(server, 'Registering Open Telemetry tracing')
  58
  59       await server.kill()
  60     })
  61
  62     it('Should enable open telemetry metrics', async function () {
  63       server = await createSingleServer(1, {
  64         open_telemetry: {
  65           tracing: {
  66             enabled: true,
  67             jaeger_exporter: {
  68               endpoint: 'http://localhost:' + mockPort
  69             }
  70           }
  71         }
  72       })
  73
  74       await expectLogContain(server, 'Registering Open Telemetry tracing')
  75     })
  76
  77     it('Should upload a video and correctly works', async function () {
  78       await setAccessTokensToServers([ server ])
  79
  80       const { uuid } = await server.videos.quickUpload({ name: 'video', privacy: VideoPrivacy.PUBLIC })
  81
  82       const video = await server.videos.get({ id: uuid })
  83
  84       expect(video.name).to.equal('video')
  85     })
  86
  87     after(async function () {
  88       await mockHTTP.terminate()
  89     })
  90   })
  91
  92   after(async function () {
  93     await cleanupTests([ server ])
  94   })
  95 })