P2P-Hiverelay / test / unit / api-gateway-stats.test.js

import test from 'brittle'
import {
  buildGatewayStatsPayload,
  buildGatewayStatsRoutePayload,
  resolveGatewayStatsRoute,
  sanitizeGatewayStats
} from 'p2p-hiverelay/core/relay-node/api-gateway-stats.js'

test('api gateway stats: route helper maps exact public stats route', (t) => {
  t.alike(resolveGatewayStatsRoute('GET', '/api/gateway'), {
    kind: 'gateway-stats'
  })

  t.is(resolveGatewayStatsRoute('POST', '/api/gateway'), null)
  t.is(resolveGatewayStatsRoute('GET', '/api/gateway/extra'), null)
  t.is(resolveGatewayStatsRoute('GET', '/api/overview'), null)
})

test('api gateway stats: route payload helper dispatches sanitized public stats', (t) => {
  let calls = 0
  const gateway = {
    getStats () {
      calls++
      return {
        cachedDrives: 2,
        totalRequests: 3,
        totalBytesServed: 4,
        cachePath: '/private/cache'
      }
    }
  }

  const result = buildGatewayStatsRoutePayload({
    route: { kind: 'gateway-stats' },
    gateway
  })
  const unknown = buildGatewayStatsRoutePayload({
    route: { kind: 'unknown' },
    gateway
  })

  t.is(calls, 1)
  t.alike(result, {
    ok: true,
    payload: {
      cachedDrives: 2,
      totalRequests: 3,
      totalBytesServed: 4
    }
  })
  t.is(unknown.status, 404)
  t.is(unknown.payload.error, 'unknown gateway stats route')
})

test('api gateway stats: missing gateway returns stable zero counters', (t) => {
  t.alike(buildGatewayStatsPayload(), {
    ok: true,
    payload: {
      cachedDrives: 0,
      totalRequests: 0,
      totalBytesServed: 0
    }
  })
})

test('api gateway stats: sanitizes public counters without raw fields', (t) => {
  const payload = sanitizeGatewayStats({
    cachedDrives: 3.7,
    totalRequests: '9',
    totalBytesServed: 1024.9,
    openDriveKeys: ['a'.repeat(64)],
    storePath: '/private/data',
    secretToken: 'do-not-leak'
  })

  t.alike(payload, {
    cachedDrives: 3,
    totalRequests: 9,
    totalBytesServed: 1024
  })
  t.absent(JSON.stringify(payload).includes('openDriveKeys'))
  t.absent(JSON.stringify(payload).includes('storePath'))
  t.absent(JSON.stringify(payload).includes('secretToken'))
  t.absent(JSON.stringify(payload).includes('do-not-leak'))
})

test('api gateway stats: malformed counters clamp to zero', (t) => {
  t.alike(sanitizeGatewayStats({
    cachedDrives: -1,
    totalRequests: Infinity,
    totalBytesServed: Number.NaN
  }), {
    cachedDrives: 0,
    totalRequests: 0,
    totalBytesServed: 0
  })
})