assert probes in proberunner

.gitignore

@@ -106,3 +106,4 @@ dist
 temp.js
 temp/
 .vscode/
+.idea/

src/ProbeRunner.js

@@ -1,3 +1,6 @@
+// Import Native Dependencies
+import assert from "node:assert";
+
 // Import all the probes
 import isUnsafeCallee from "./probes/isUnsafeCallee.js";
 import isLiteral from "./probes/isLiteral.js";
@@ -71,7 +74,17 @@ export class ProbeRunner {
   ) {
     this.sourceFile = sourceFile;
 
-    // TODO: Assert/validate all probes? (Good first issue)
+    for (const probe of probes) {
+      assert(
+        typeof probe.validateNode === "function" || Array.isArray(probe.validateNode),
+        `Invalid probe ${probe.name}: validateNode must be a function or an array of functions`
+      );
+      assert(
+        typeof probe.main === "function",
+        `Invalid probe ${probe.name}: main must be a function`
+      );
+    }
+
     this.probes = probes;
   }
 

test/ProbeRunner.spec.js

@@ -14,12 +14,57 @@ describe("ProbeRunner", () => {
       assert.strictEqual(pr.probes, ProbeRunner.Defaults);
     });
 
-    it("should use provided probes", () => {
-      const probes = [{}];
+    it("should use provided probes with validate node as func", () => {
+      const fakeProbe = [
+        {
+          validateNode: (node) => [node.type === "CallExpression"],
+          main: mock.fn(),
+          teardown: mock.fn()
+        }
+      ];
 
-      const pr = new ProbeRunner(null, probes);
+      const pr = new ProbeRunner(null, fakeProbe);
       assert.strictEqual(pr.sourceFile, null);
-      assert.strictEqual(pr.probes, probes);
+      assert.strictEqual(pr.probes, fakeProbe);
+    });
+
+    it("should use provided probe with validate node as Array", () => {
+      const fakeProbe = [
+        {
+          validateNode: [],
+          main: mock.fn(),
+          teardown: mock.fn()
+        }];
+
+      const pr = new ProbeRunner(null, fakeProbe);
+      assert.strictEqual(pr.sourceFile, null);
+      assert.strictEqual(pr.probes, fakeProbe);
+    });
+
+    it("should fail if main not present", () => {
+      const fakeProbe = {
+        validateNode: (node) => [node.type === "CallExpression"],
+        teardown: mock.fn()
+      };
+
+      function instantiateProbeRunner() {
+        return new ProbeRunner(null, [fakeProbe]);
+      }
+
+      assert.throws(instantiateProbeRunner, Error, "Invalid probe");
+    });
+
+    it("should fail if validate not present", () => {
+      const fakeProbe = {
+        main: mock.fn(),
+        teardown: mock.fn()
+      };
+
+      function instantiateProbeRunner() {
+        return new ProbeRunner(null, [fakeProbe]);
+      }
+
+      assert.throws(instantiateProbeRunner, Error, "Invalid probe");
     });
   });