sentry-javascript/dev-packages/cloudflare-integration-tests/suites/tracing/d1/test.ts at 204c0ae88832d1b25821a4541945429121ced4e0 · getsentry/sentry-javascript · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
import { expect, it } from 'vitest';
import { SEMANTIC_ATTRIBUTE_SENTRY_OP, SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN } from '@sentry/core';
import { createRunner } from '../../../runner';

it('D1 database queries create spans with correct attributes', async ({ signal }) => {
  const runner = createRunner(__dirname)
    .expect(envelope => {
      const transactionEvent = envelope[1]?.[0]?.[1];
      expect(transactionEvent).toEqual(
        expect.objectContaining({
          type: 'transaction',
          transaction: 'GET /init',
          spans: [
            {
              data: {
                [SEMANTIC_ATTRIBUTE_SENTRY_OP]: 'db.query',
                [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: 'auto.db.cloudflare.d1',
                'cloudflare.d1.query_type': 'run',
                'cloudflare.d1.duration': expect.any(Number),
                'cloudflare.d1.rows_read': expect.any(Number),
                'cloudflare.d1.rows_written': expect.any(Number),
              },
              description: 'INSERT INTO users (name) VALUES (?)',
              op: 'db.query',
              origin: 'auto.db.cloudflare.d1',
              parent_span_id: expect.any(String),
              span_id: expect.any(String),
              start_timestamp: expect.any(Number),
              timestamp: expect.any(Number),
              trace_id: expect.any(String),
            },
          ],
        }),
      );
    })
    .expect(envelope => {
      const transactionEvent = envelope[1]?.[0]?.[1];
      expect(transactionEvent).toEqual(
        expect.objectContaining({
          type: 'transaction',
          transaction: 'GET /query',
          spans: [
            {
              data: {
                [SEMANTIC_ATTRIBUTE_SENTRY_OP]: 'db.query',
                [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: 'auto.db.cloudflare.d1',
                'cloudflare.d1.query_type': 'first',
              },
              description: 'SELECT * FROM users WHERE name = ?',
              op: 'db.query',
              origin: 'auto.db.cloudflare.d1',
              parent_span_id: expect.any(String),
              span_id: expect.any(String),
              start_timestamp: expect.any(Number),
              timestamp: expect.any(Number),
              trace_id: expect.any(String),
            },
          ],
        }),
      );
    })
    .start(signal);

  await runner.makeRequest('get', '/init');
  await runner.makeRequest('get', '/query');
  await runner.completed();
});