1  import assert from 'node:assert/strict'
 2  import { describe, it } from 'node:test'
 3  import { isHeartbeatSource, isInternalHeartbeatRun } from '@/lib/server/runtime/heartbeat-source'
 4  
 5  describe('heartbeat-source', () => {
 6    it('treats scheduled heartbeat polls as heartbeat traffic', () => {
 7      assert.equal(isHeartbeatSource('heartbeat'), true)
 8      assert.equal(isInternalHeartbeatRun(true, 'heartbeat'), true)
 9    })
10  
11    it('treats wake-triggered heartbeat polls as heartbeat traffic', () => {
12      assert.equal(isHeartbeatSource('heartbeat-wake'), true)
13      assert.equal(isInternalHeartbeatRun(true, 'heartbeat-wake'), true)
14    })
15  
16    it('does not classify other sources as heartbeat traffic', () => {
17      assert.equal(isHeartbeatSource('task'), false)
18      assert.equal(isHeartbeatSource('chat'), false)
19      assert.equal(isInternalHeartbeatRun(false, 'heartbeat'), false)
20      assert.equal(isInternalHeartbeatRun(true, 'task'), false)
21    })
22  })