|
| 1 | +import { afterAll, expect, test } from 'vitest'; |
| 2 | +import { cleanupChildProcesses, createRunner } from '../../../node-integration-tests/utils/runner'; |
| 3 | + |
| 4 | +afterAll(cleanupChildProcesses); |
| 5 | + |
| 6 | +test('records streaming work after the response headers have been returned', async () => { |
| 7 | + const runner = createRunner(__dirname, 'index.ts') |
| 8 | + .withMockSentryServer() |
| 9 | + .unordered() |
| 10 | + .expect({ |
| 11 | + span: container => { |
| 12 | + const root = container.items.find(item => item.is_segment); |
| 13 | + expect(root).toEqual( |
| 14 | + expect.objectContaining({ |
| 15 | + attributes: expect.objectContaining({ |
| 16 | + 'url.path': { value: '/events', type: 'string' }, |
| 17 | + 'http.response.status_code': { value: 201, type: 'integer' }, |
| 18 | + 'test.stream.completed': { value: true, type: 'boolean' }, |
| 19 | + }), |
| 20 | + }), |
| 21 | + ); |
| 22 | + }, |
| 23 | + }) |
| 24 | + .start(); |
| 25 | + expect(await runner.makeRequest('get', '/events')).toBe('data: first\n\ndata: recording=true\n\n'); |
| 26 | + await runner.completed(); |
| 27 | +}); |
| 28 | + |
| 29 | +test('records a thrown handler as an error span', async () => { |
| 30 | + const runner = createRunner(__dirname, 'index.ts') |
| 31 | + .withMockSentryServer() |
| 32 | + .unordered() |
| 33 | + .expect({ |
| 34 | + span: container => { |
| 35 | + const root = container.items.find(item => item.is_segment); |
| 36 | + expect(root).toEqual(expect.objectContaining({ status: 'error' })); |
| 37 | + }, |
| 38 | + }) |
| 39 | + .start(); |
| 40 | + await runner.makeRequest('get', '/error', { expectError: true }); |
| 41 | + await runner.completed(); |
| 42 | +}); |
| 43 | + |
| 44 | +test('captures a response stream failure with its request context and error span', async () => { |
| 45 | + let spanTraceId: string | undefined; |
| 46 | + let eventTraceId: string | undefined; |
| 47 | + const runner = createRunner(__dirname, 'index.ts') |
| 48 | + .withMockSentryServer() |
| 49 | + .unordered() |
| 50 | + .expect({ |
| 51 | + span: container => { |
| 52 | + const roots = container.items.filter(item => item.is_segment); |
| 53 | + expect(roots).toHaveLength(1); |
| 54 | + const root = roots[0]!; |
| 55 | + expect(root.status).toBe('error'); |
| 56 | + expect(root.attributes['url.path']?.value).toBe('/stream-error'); |
| 57 | + expect(root.attributes['http.response.status_code']?.value).toBe(201); |
| 58 | + spanTraceId = root.trace_id; |
| 59 | + }, |
| 60 | + }) |
| 61 | + .expect({ |
| 62 | + event: event => { |
| 63 | + expect(event.exception?.values).toHaveLength(1); |
| 64 | + const exception = event.exception!.values![0]!; |
| 65 | + expect(exception.type).toBe('Error'); |
| 66 | + expect(exception.value).toBe('stream failed'); |
| 67 | + expect(exception.mechanism).toEqual({ type: 'auto.http.bun.serve', handled: false }); |
| 68 | + expect(event.request?.method).toBe('GET'); |
| 69 | + expect(new URL(event.request!.url!).pathname).toBe('/stream-error'); |
| 70 | + eventTraceId = event.contexts?.trace?.trace_id; |
| 71 | + }, |
| 72 | + }) |
| 73 | + .start(); |
| 74 | + |
| 75 | + await expect.poll(() => runner.getPort(), { timeout: 30_000 }).toBeTypeOf('number'); |
| 76 | + const response = await fetch(`http://localhost:${runner.getPort()}/stream-error`); |
| 77 | + expect(response.status).toBe(201); |
| 78 | + // Bun versions differ in whether a failed stream rejects or closes the HTTP response body. |
| 79 | + await Promise.allSettled([response.text()]); |
| 80 | + await runner.completed(); |
| 81 | + expect(spanTraceId).toMatch(/^[a-f\d]{32}$/); |
| 82 | + expect(eventTraceId).toBe(spanTraceId); |
| 83 | +}); |
0 commit comments