improve test

Author: nicohrubec

dev-packages/browser-integration-tests/suites/tracing/ai-providers/langgraph/mocks.js

@@ -1,5 +1,4 @@
 // Mock LangGraph graph for browser testing
-
 export class MockStateGraph {
   compile(options = {}) {
     const compiledGraph = {
@@ -11,7 +10,7 @@ export class MockStateGraph {
       builder: {
         nodes: {},
       },
-      invoke: async (input) => {
+      invoke: async input => {
         const messages = input?.messages;
         return {
           messages: [

dev-packages/browser-integration-tests/suites/tracing/ai-providers/langgraph/test.ts

@@ -8,23 +8,38 @@ import { envelopeRequestParser, waitForTransactionRequest } from '../../../../ut
 // and that gen_ai transactions are sent.
 
 sentryTest('manual LangGraph instrumentation sends gen_ai transactions', async ({ getLocalTestUrl, page }) => {
-  const transactionPromise = waitForTransactionRequest(page, event => {
-    return !!event.transaction?.includes('mock-graph');
+  const createTransactionPromise = waitForTransactionRequest(page, event => {
+    return !!event.transaction?.includes('create_agent mock-graph');
+  });
+
+  const invokeTransactionPromise = waitForTransactionRequest(page, event => {
+    return !!event.transaction?.includes('invoke_agent mock-graph');
   });
 
   const url = await getLocalTestUrl({ testDir: __dirname });
   await page.goto(url);
 
-  const req = await transactionPromise;
+  const createReq = await createTransactionPromise;
+  const invokeReq = await invokeTransactionPromise;
 
-  const eventData = envelopeRequestParser(req);
+  const createEventData = envelopeRequestParser(createReq);
+  const invokeEventData = envelopeRequestParser(invokeReq);
 
-  // Verify it's a gen_ai transaction
-  expect(eventData.transaction).toBe('create_agent mock-graph');
-  expect(eventData.contexts?.trace?.op).toBe('gen_ai.create_agent');
-  expect(eventData.contexts?.trace?.origin).toBe('auto.ai.langgraph');
-  expect(eventData.contexts?.trace?.data).toMatchObject({
+  // Verify create_agent transaction
+  expect(createEventData.transaction).toBe('create_agent mock-graph');
+  expect(createEventData.contexts?.trace?.op).toBe('gen_ai.create_agent');
+  expect(createEventData.contexts?.trace?.origin).toBe('auto.ai.langgraph');
+  expect(createEventData.contexts?.trace?.data).toMatchObject({
     'gen_ai.operation.name': 'create_agent',
     'gen_ai.agent.name': 'mock-graph',
   });
+
+  // Verify invoke_agent transaction
+  expect(invokeEventData.transaction).toBe('invoke_agent mock-graph');
+  expect(invokeEventData.contexts?.trace?.op).toBe('gen_ai.invoke_agent');
+  expect(invokeEventData.contexts?.trace?.origin).toBe('auto.ai.langgraph');
+  expect(invokeEventData.contexts?.trace?.data).toMatchObject({
+    'gen_ai.operation.name': 'invoke_agent',
+    'gen_ai.agent.name': 'mock-graph',
+  });
 });