test.ts

import { type Page, expect } from '@playwright/test';
import type { Event } from '@sentry/core';

import { sentryTest } from '../../../../utils/fixtures';
import { getFirstSentryEnvelopeRequest, shouldSkipTracingTest } from '../../../../utils/helpers';

async function mockSupabaseRoute(page: Page) {
  await page.route('**/rest/v1/rpc**', route => {
    return route.fulfill({
      status: 200,
      body: JSON.stringify({
        foo: ['bar', 'baz'],
      }),
      headers: {
        'Content-Type': 'application/json',
      },
    });
  });
}

sentryTest('should capture Supabase queue spans from client.schema(...).rpc', async ({ getLocalTestUrl, page }) => {
  await mockSupabaseRoute(page);

  if (shouldSkipTracingTest()) {
    return;
  }

  const url = await getLocalTestUrl({ testDir: __dirname });

  const event = await getFirstSentryEnvelopeRequest<Event>(page, url);
  const queueSpans = event.spans?.filter(({ op }) => op?.startsWith('queue'));

  expect(queueSpans).toHaveLength(2);

  expect(queueSpans![0]).toMatchObject({
    description: 'supabase.db.rpc',
    parent_span_id: event.contexts?.trace?.span_id,
    span_id: expect.any(String),
    start_timestamp: expect.any(Number),
    timestamp: expect.any(Number),
    trace_id: event.contexts?.trace?.trace_id,
    data: expect.objectContaining({
      'sentry.op': 'queue.publish',
      'sentry.origin': 'auto.db.supabase',
      'messaging.destination.name': 'todos',
      'messaging.message.id': 'Test Todo',
    }),
  });

  expect(queueSpans![1]).toMatchObject({
    description: 'supabase.db.rpc',
    parent_span_id: event.contexts?.trace?.span_id,
    span_id: expect.any(String),
    start_timestamp: expect.any(Number),
    timestamp: expect.any(Number),
    trace_id: event.contexts?.trace?.trace_id,
    data: expect.objectContaining({
      'sentry.op': 'queue.process',
      'sentry.origin': 'auto.db.supabase',
      'messaging.destination.name': 'todos',
    }),
  });
});