68 lines
2.5 KiB
TypeScript
68 lines
2.5 KiB
TypeScript
import { describe, test } from '@jest/globals'
|
|
import request from 'supertest'
|
|
import { PrometheusServer } from '../../../src/prometheus'
|
|
import { serve } from '../../../src/adapters/http/fastify'
|
|
import { Server, createServer } from 'http'
|
|
import type { HttpServer } from '../../../src/http'
|
|
import { Metric } from '../../../src/metrics'
|
|
|
|
class TestablePrometheusServer extends PrometheusServer {
|
|
public serverFactory: HttpServer['serverFactory']
|
|
}
|
|
|
|
function createTestServer(): { http: Server; prometheus: HttpServer } {
|
|
const http = createServer()
|
|
const prometheus = new TestablePrometheusServer(0, undefined, false, 'homebridge')
|
|
prometheus.serverFactory = (handler) => http.on('request', handler)
|
|
serve(prometheus).catch((err: Error) => {
|
|
if (!('code' in err) || (err as unknown as { code: unknown }).code !== 'ERR_SERVER_ALREADY_LISTEN') {
|
|
console.debug(err)
|
|
}
|
|
})
|
|
|
|
return { http, prometheus }
|
|
}
|
|
|
|
describe('Fastify HTTP adapter', () => {
|
|
test('Serves 503 everywhere while metrics are not available', () => {
|
|
return request(createTestServer().http)
|
|
.get('/any-url')
|
|
.expect(503)
|
|
.expect('Content-Type', 'text/plain; charset=utf-8')
|
|
.expect('Retry-After', '15')
|
|
.expect('Metrics discovery pending')
|
|
})
|
|
|
|
test('Serves 404 on / when metrics are available', () => {
|
|
const testServer = createTestServer()
|
|
testServer.prometheus.updateMetrics([])
|
|
|
|
return request(testServer.http)
|
|
.get('/')
|
|
.expect(404)
|
|
.expect('Content-Type', 'text/plain; charset=utf-8')
|
|
.expect('Not found. Try /metrics')
|
|
})
|
|
|
|
test('Serves metrics', () => {
|
|
const testServer = createTestServer()
|
|
const timestamp = new Date('2020-01-01 00:00:00 UTC')
|
|
testServer.prometheus.updateMetrics([
|
|
new Metric('metric', 0.1, timestamp, { name: 'metric' }),
|
|
new Metric('total_something', 100, timestamp, { name: 'counter' }),
|
|
])
|
|
|
|
return request(testServer.http)
|
|
.get('/metrics')
|
|
.expect(200)
|
|
.expect('Content-Type', 'text/plain; charset=utf-8; version=0.0.4')
|
|
.expect(
|
|
[
|
|
'# TYPE homebridge_metric gauge',
|
|
'homebridge_metric{name="metric"} 0.1 1577836800000',
|
|
'# TYPE homebridge_something_total counter',
|
|
'homebridge_something_total{name="counter"} 100 1577836800000',
|
|
].join('\n'),
|
|
)
|
|
})
|
|
})
|