From d610873755f6bbd3444d02bcdfbf4e8b6311de9b Mon Sep 17 00:00:00 2001
From: Tony Gorez <gorez.tony@gmail.com>
Date: Wed, 15 May 2024 20:57:26 +0200
Subject: [PATCH] test: use nodejs api for mocks

Signed-off-by: Tony Gorez <gorez.tony@gmail.com>
---
 test/AstAnalyser.spec.js | 43 ++++++++++++++++------------------------
 1 file changed, 17 insertions(+), 26 deletions(-)

diff --git a/test/AstAnalyser.spec.js b/test/AstAnalyser.spec.js
index bf43205..3742505 100644
--- a/test/AstAnalyser.spec.js
+++ b/test/AstAnalyser.spec.js
@@ -177,6 +177,7 @@ describe("AstAnalyser", (t) => {
 
     describe("intialize", () => {
       const analyser = new AstAnalyser();
+
       it("should throw if initialize is not a function", () => {
         assert.throws(() => {
           analyser.analyse("const foo = 'bar';", {
@@ -185,29 +186,24 @@ describe("AstAnalyser", (t) => {
         });
       });
 
-      it("should call the initialize function", () => {
-        let hasBeenCalled = false;
-        const initialize = () => {
-          assert.strictEqual(hasBeenCalled, false);
-          hasBeenCalled = true;
-        };
+      it("should call the initialize function", (t) => {
+        const initialize = t.mock.fn();
 
         analyser.analyse("const foo = 'bar';", {
           initialize
         });
-        assert.strictEqual(hasBeenCalled, true);
+
+        assert.strictEqual(initialize.mock.callCount(), 1);
       });
 
-      it("should pass the source file as first argument", () => {
-        let hasBeenCalled = false;
-        const initialize = (source) => {
-          assert.strictEqual(source instanceof SourceFile, true);
-        };
+      it("should pass the source file as first argument", (t) => {
+        const initialize = t.mock.fn();
 
         analyser.analyse("const foo = 'bar';", {
           initialize
         });
-        assert.strictEqual(hasBeenCalled, true);
+
+        assert.strictEqual(initialize.mock.calls[0].arguments[0] instanceof SourceFile, true);
       });
     });
 
@@ -221,29 +217,24 @@ describe("AstAnalyser", (t) => {
         });
       });
 
-      it("should call the finalize function", () => {
-        let hasBeenCalled = false;
-        const finalize = () => {
-          assert.strictEqual(hasBeenCalled, false);
-          hasBeenCalled = true;
-        };
+      it("should call the finalize function", (t) => {
+        const finalize = t.mock.fn();
 
         analyser.analyse("const foo = 'bar';", {
           finalize
         });
-        assert.strictEqual(hasBeenCalled, true);
+
+        assert.strictEqual(finalize.mock.callCount(), 1);
       });
 
-      it("should pass the source file as first argument", () => {
-        let hasBeenCalled = false;
-        const finalize = (source) => {
-          assert.strictEqual(source instanceof SourceFile, true);
-        };
+      it("should pass the source file as first argument", (t) => {
+        const finalize = t.mock.fn();
 
         analyser.analyse("const foo = 'bar';", {
           finalize
         });
-        assert.strictEqual(hasBeenCalled, true);
+
+        assert.strictEqual(finalize.mock.calls[0].arguments[0] instanceof SourceFile, true);
       });
     });
   });