add more tests, and some fixes to go along with them

* indices, make them actually work
* make multiple instrumentations in one file actually work

Author: bengl

src/config.rs

@@ -23,7 +23,7 @@ macro_rules! get_arr {
     };
 }
 
-#[derive(Clone)]
+#[derive(Clone, Debug)]
 pub enum InstrumentationOperator {
     Callback,
     Promise,
@@ -52,6 +52,7 @@ impl InstrumentationOperator {
     }
 }
 
+#[derive(Debug)]
 pub struct InstrumentationConfig {
     pub module_name: String,
     pub version_range: Range,

src/function_query.rs

@@ -10,6 +10,7 @@ macro_rules! get_str {
     };
 }
 
+#[derive(Debug)]
 pub enum FunctionType {
     FunctionDeclaration,
     FunctionExpression,
@@ -27,6 +28,7 @@ impl FunctionType {
     }
 }
 
+#[derive(Debug)]
 pub enum FunctionKind {
     Sync,
     Async,
@@ -63,6 +65,7 @@ impl FunctionKind {
     }
 }
 
+#[derive(Debug)]
 pub struct FunctionQuery {
     pub name: String,
     pub class: Option<String>,

src/instrumentation.rs

@@ -70,7 +70,9 @@ impl Instrumentation {
         define_channel
     }
 
-    fn insert_tracing(&self, body: &mut BlockStmt) {
+    fn insert_tracing(&mut self, body: &mut BlockStmt) {
+        self.count += 1;
+
         let original_stmts = std::mem::take(&mut body.stmts);
 
         // Create a new BlockStmt with the original statements
@@ -102,7 +104,9 @@ impl Instrumentation {
         ];
     }
 
-    fn insert_constructor_tracing(&self, body: &mut BlockStmt) {
+    fn insert_constructor_tracing(&mut self, body: &mut BlockStmt) {
+        self.count += 1;
+
         let original_stmts = std::mem::take(&mut body.stmts);
 
         let ch_ident = ident!(format!("tr_ch_apm${}", &self.config.channel_name));

src/lib.rs

@@ -103,8 +103,14 @@ impl<'a> InstrumentationVisitor<'a> {
 macro_rules! visit_with_all {
     ($self:expr, $method:ident, $item:expr) => {
         let mut recurse = false;
+        println!(
+            "InstrumentationVisitor::visit_mut_{}: Instrumentation count is {}",
+            stringify!($method),
+            $self.instrumentations.len()
+        );
         for instr in &mut $self.instrumentations {
-            recurse = recurse || instr.$method($item);
+            let needs_recurse = instr.$method($item);
+            recurse = recurse || needs_recurse;
         }
         if recurse {
             $item.visit_mut_children_with($self);

tests/index_cjs/instrumentations.yml

@@ -0,0 +1,13 @@
+version: 1
+instrumentations:
+  - module_name: undici
+    version_range: ">=0.0.1"
+    file_path: index.mjs
+    function_query:
+      class: Undici
+      name: fetch
+      type: method
+      kind: async
+      index: 2
+    operator: tracePromise
+    channel_name: Undici_fetch

tests/index_cjs/mod.js

@@ -0,0 +1,45 @@
+{ 
+  class Undici {
+    async fetch (url) {
+      return 0;
+    }
+  }
+
+  exports.Undici0 = Undici;
+}
+{ 
+  class Undici {
+    async fetch (url) {
+      return 1;
+    }
+  }
+
+  exports.Undici1 = Undici;
+}
+{ 
+  class Undici {
+    async fetch (url) {
+      return 2;
+    }
+  }
+
+  exports.Undici2 = Undici;
+}
+{ 
+  class Undici {
+    async fetch (url) {
+      return 3;
+    }
+  }
+
+  exports.Undici3 = Undici;
+}
+{ 
+  class Undici {
+    async fetch (url) {
+      return 4;
+    }
+  }
+
+  exports.Undici4 = Undici;
+}

tests/index_cjs/test.js

@@ -0,0 +1,27 @@
+const undicis = require('./instrumented.js');
+const { assert, getContext } = require('../common/preamble.js');
+const context = getContext('orchestrion:undici:Undici_fetch');
+
+async function testOne(Undici, num, expectedCtx) {
+  const undici = new Undici;
+  const result = await undici.fetch('https://example.com');
+  assert.strictEqual(result, num);
+  assert.deepStrictEqual(context, expectedCtx);
+  delete context.start;
+  delete context.end;
+  delete context.asyncStart;
+  delete context.asyncEnd;
+}
+
+(async () => {
+  await testOne(undicis.Undici0, 0, {});
+  await testOne(undicis.Undici1, 1, {});
+  await testOne(undicis.Undici2, 2, {
+    start: true,
+    end: true,
+    asyncStart: 2,
+    asyncEnd: 2
+  });
+  await testOne(undicis.Undici3, 3, {});
+  await testOne(undicis.Undici4, 4, {});
+})();

tests/instrumentor_test.rs

@@ -27,6 +27,8 @@ make_test!(expr_cjs, false);
 
 make_test!(class_method_cjs, false);
 
+make_test!(multiple_class_method_cjs, false);
+
 make_test!(object_method_cjs, false);
 
 make_test!(constructor_cjs, false);
@@ -36,3 +38,8 @@ make_test!(constructor_mjs, true);
 make_test!(polyfill_mjs, true);
 
 make_test!(polyfill_cjs, false);
+
+make_test!(index_cjs, false);
+
+make_test!(no_index_cjs, false);
+

tests/multiple_class_method_cjs/instrumentations.yml

@@ -0,0 +1,22 @@
+version: 1
+instrumentations:
+  - module_name: undici
+    version_range: ">=0.0.1"
+    file_path: index.mjs
+    function_query:
+      class: Undici
+      name: fetch1
+      type: method
+      kind: async
+    operator: tracePromise
+    channel_name: Undici_fetch1
+  - module_name: undici
+    version_range: ">=0.0.1"
+    file_path: index.mjs
+    function_query:
+      class: Undici
+      name: fetch2
+      type: method
+      kind: async
+    operator: tracePromise
+    channel_name: Undici_fetch2

tests/multiple_class_method_cjs/mod.js

@@ -0,0 +1,11 @@
+class Undici {
+    async fetch1 (url) {
+        return 42;
+    }
+
+    async fetch2 (url) {
+        return 43;
+    }
+}
+
+module.exports = Undici;