sentry-javascript/dev-packages/node-integration-tests/suites/tracing/prisma-orm-v5/test.ts at ba5f19d817841a0c4308b06a400fab71dd18d167 · 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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
import { afterAll, describe, expect } from 'vitest';
import { cleanupChildProcesses, createEsmAndCjsTests } from '../../../utils/runner';

afterAll(() => {
  cleanupChildProcesses();
});

describe('Prisma ORM v5 Tests', () => {
  createEsmAndCjsTests(
    __dirname,
    'scenario.mjs',
    'instrument.mjs',
    (createRunner, test, _mode, cwd) => {
      test('should instrument PostgreSQL queries from Prisma ORM', { timeout: 75_000 }, async () => {
        await createRunner()
          .withDockerCompose({
            workingDirectory: [cwd],
            setupCommand: 'yarn prisma generate && yarn prisma migrate dev -n sentry-test',
          })
          .expect({
            transaction: transaction => {
              expect(transaction.transaction).toBe('Test Transaction');
              const spans = transaction.spans || [];
              expect(spans.length).toBeGreaterThanOrEqual(5);

              expect(spans).toContainEqual(
                expect.objectContaining({
                  data: {
                    method: 'create',
                    model: 'User',
                    name: 'User.create',
                    'sentry.origin': 'auto.db.otel.prisma',
                  },
                  description: 'prisma:client:operation',
                  status: 'ok',
                }),
              );

              expect(spans).toContainEqual(
                expect.objectContaining({
                  data: {
                    'sentry.origin': 'auto.db.otel.prisma',
                  },
                  description: 'prisma:client:serialize',
                  status: 'ok',
                }),
              );

              expect(spans).toContainEqual(
                expect.objectContaining({
                  data: {
                    'sentry.origin': 'auto.db.otel.prisma',
                  },
                  description: 'prisma:client:connect',
                  status: 'ok',
                }),
              );
              expect(spans).toContainEqual(
                expect.objectContaining({
                  data: {
                    method: 'findMany',
                    model: 'User',
                    name: 'User.findMany',
                    'sentry.origin': 'auto.db.otel.prisma',
                  },
                  description: 'prisma:client:operation',
                  status: 'ok',
                }),
              );
              expect(spans).toContainEqual(
                expect.objectContaining({
                  data: {
                    'sentry.origin': 'auto.db.otel.prisma',
                  },
                  description: 'prisma:client:serialize',
                  status: 'ok',
                }),
              );
              expect(spans).toContainEqual(
                expect.objectContaining({
                  data: {
                    'db.statement': expect.stringContaining('INSERT INTO'),
                    'db.system': 'postgresql',
                    'otel.kind': 'CLIENT',
                    'sentry.op': 'db',
                    'sentry.origin': 'auto.db.otel.prisma',
                  },
                  op: 'db',
                  description: expect.stringContaining('INSERT INTO'),
                  status: 'ok',
                }),
              );
              expect(spans).toContainEqual(
                expect.objectContaining({
                  data: {
                    'db.statement': expect.stringContaining('SELECT'),
                    'db.system': 'postgresql',
                    'otel.kind': 'CLIENT',
                    'sentry.op': 'db',
                    'sentry.origin': 'auto.db.otel.prisma',
                  },
                  op: 'db',
                  description: expect.stringContaining('SELECT'),
                  status: 'ok',
                }),
              );
              expect(spans).toContainEqual(
                expect.objectContaining({
                  data: {
                    'db.statement': expect.stringContaining('DELETE'),
                    'db.system': 'postgresql',
                    'otel.kind': 'CLIENT',
                    'sentry.op': 'db',
                    'sentry.origin': 'auto.db.otel.prisma',
                  },
                  op: 'db',
                  description: expect.stringContaining('DELETE'),
                  status: 'ok',
                }),
              );
            },
          })
          .start()
          .completed();
      });
    },
    {
      additionalDependencies: {
        '@prisma/client': '5.22.0',
        prisma: '5.22.0',
      },
      copyPaths: ['prisma', 'docker-compose.yml'],
    },
  );
});