fix test

Author: jsumners-nr

test/versioned/langchain/runnables-streaming.js

@@ -49,9 +49,19 @@ function runStreamingEnabledTests(config) {
   } = config
 
   return async (t) => {
-    await t.test('should log tracking metrics', function(t) {
-      const { agent, langchainCoreVersion } = t.nr
-      assertPackageMetrics({ agent, pkg: '@langchain/core', version: langchainCoreVersion })
+    await t.test('should log tracking metrics', function(t, end) {
+      t.plan(5)
+      const { agent, langchainCoreVersion, prompt, model } = t.nr
+      helper.runInTransaction(agent, async () => {
+        await prompt.pipe(model).stream(inputData)
+        assertPackageMetrics({
+          agent,
+          pkg: '@langchain/core',
+          version: langchainCoreVersion,
+          subscriberType: true
+        }, { assert: t.assert })
+        end()
+      })
     })
 
     await t.test('should create langchain events for every stream call', (t, end) => {

test/versioned/langchain/runnables.js

@@ -41,9 +41,19 @@ function runRunnablesTests(config) {
     arrayParserOutput
   } = config
 
-  test('should log tracking metrics', function(t) {
-    const { agent, langchainCoreVersion } = t.nr
-    assertPackageMetrics({ agent, pkg: '@langchain/core', version: langchainCoreVersion })
+  test('should log tracking metrics', function(t, end) {
+    t.plan(5)
+    const { agent, langchainCoreVersion, model, prompt } = t.nr
+    helper.runInTransaction(agent, async () => {
+      await prompt.pipe(model).invoke(inputData)
+      assertPackageMetrics({
+        agent,
+        pkg: '@langchain/core',
+        version: langchainCoreVersion,
+        subscriberType: true
+      }, { assert: t.assert })
+      end()
+    })
   })
 
   test('should create langchain events for every invoke call', (t, end) => {

test/versioned/langchain/tools.test.js

@@ -39,10 +39,20 @@ test.afterEach((ctx) => {
   removeMatchedModules(/custom-tool\.js$/)
 })
 
-test('should log tracking metrics', function(t) {
-  const { agent } = t.nr
+test('should log tracking metrics', function(t, end) {
+  t.plan(5)
+  const { agent, tool, input } = t.nr
   const { version } = require('@langchain/core/package.json')
-  assertPackageMetrics({ agent, pkg: '@langchain/core', version })
+  helper.runInTransaction(agent, async () => {
+    await tool.call(input)
+    assertPackageMetrics({
+      agent,
+      pkg: '@langchain/core',
+      version,
+      subscriberType: true
+    }, { assert: t.assert })
+    end()
+  })
 })
 
 test('should create span on successful tools create', (t, end) => {

test/versioned/langchain/vectorstore.js

@@ -36,9 +36,19 @@ function runVectorstoreTests(config) {
     errorAssertion
   } = config
 
-  test('should log tracking metrics', function(t) {
-    const { agent, langchainCoreVersion } = t.nr
-    assertPackageMetrics({ agent, pkg: '@langchain/core', version: langchainCoreVersion })
+  test('should log tracking metrics', function(t, end) {
+    t.plan(5)
+    const { agent, langchainCoreVersion, vs } = t.nr
+    helper.runInTransaction(agent, async () => {
+      await vs.similaritySearch(searchQuery, 1)
+      assertPackageMetrics({
+        agent,
+        pkg: '@langchain/core',
+        version: langchainCoreVersion,
+        subscriberType: true
+      }, { assert: t.assert })
+      end()
+    })
   })
 
   test('should create vectorstore events for every similarity search call', (t, end) => {