fix test

Author: jsumners-nr

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) => {