test(transformer): add tests for objectName+propertyName selector

Author: crysmags

tests/object_property_named_cjs/mod.js

@@ -0,0 +1,14 @@
+/**
+ * Unless explicitly stated otherwise all files in this repository are licensed under the Apache-2.0 License.
+ * This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2025 Datadog, Inc.
+ **/
+'use strict'
+
+// Named object pattern: async arrow function assigned to a property
+// on a named identifier (not `this`).
+const conn = {}
+conn.query = async () => {
+  return 42
+}
+
+module.exports = { conn }

tests/object_property_named_cjs/test.js

@@ -0,0 +1,20 @@
+/**
+ * Unless explicitly stated otherwise all files in this repository are licensed under the Apache-2.0 License.
+ * This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2025 Datadog, Inc.
+ **/
+'use strict'
+
+const { conn } = require('./instrumented.js')
+const { assert, getContext } = require('../common/preamble.js')
+const context = getContext('orchestrion:undici:conn_query')
+
+;(async () => {
+  const result = await conn.query()
+  assert.strictEqual(result, 42)
+  assert.deepStrictEqual(context, {
+    start: true,
+    end: true,
+    asyncStart: 42,
+    asyncEnd: 42
+  })
+})()

tests/object_property_this_cjs/mod.js

@@ -0,0 +1,15 @@
+/**
+ * Unless explicitly stated otherwise all files in this repository are licensed under the Apache-2.0 License.
+ * This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2025 Datadog, Inc.
+ **/
+'use strict'
+
+// Mimics the mariadb v2 pattern: query methods are arrow functions
+// assigned to `this` inside a function constructor.
+function Connection (opts) {
+  this._query = async () => {
+    return 42
+  }
+}
+
+module.exports = { Connection }

tests/object_property_this_cjs/test.js

@@ -0,0 +1,21 @@
+/**
+ * Unless explicitly stated otherwise all files in this repository are licensed under the Apache-2.0 License.
+ * This product includes software developed at Datadog (https://www.datadoghq.com/). Copyright 2025 Datadog, Inc.
+ **/
+'use strict'
+
+const { Connection } = require('./instrumented.js')
+const { assert, getContext } = require('../common/preamble.js')
+const context = getContext('orchestrion:undici:Connection_query')
+
+;(async () => {
+  const conn = new Connection({ host: 'localhost' })
+  const result = await conn._query()
+  assert.strictEqual(result, 42)
+  assert.deepStrictEqual(context, {
+    start: true,
+    end: true,
+    asyncStart: 42,
+    asyncEnd: 42
+  })
+})()

tests/tests.test.mjs

@@ -529,3 +529,27 @@ describe('IIFE with class', () => {
     ])
   })
 })
+
+describe('object_property_this_cjs', () => {
+  test('instruments async arrow function assigned to this inside a function constructor', () => {
+    runTest('object_property_this_cjs', [
+      {
+        channelName: 'Connection_query',
+        module: { name: TEST_MODULE_NAME, versionRange: '>=0.0.1', filePath: TEST_MODULE_PATH },
+        functionQuery: { objectName: 'this', propertyName: '_query', kind: 'Async' },
+      },
+    ])
+  })
+})
+
+describe('object_property_named_cjs', () => {
+  test('instruments async arrow function assigned to a named identifier property', () => {
+    runTest('object_property_named_cjs', [
+      {
+        channelName: 'conn_query',
+        module: { name: TEST_MODULE_NAME, versionRange: '>=0.0.1', filePath: TEST_MODULE_PATH },
+        functionQuery: { objectName: 'conn', propertyName: 'query', kind: 'Async' },
+      },
+    ])
+  })
+})