Test Cases created with Jest (add/sub/mult/div)

.github/workflows/test.yml

@@ -21,4 +21,7 @@ jobs:
         with:
           node-version: '20'
       - name: Install dependencies
-        run: npm ci
+        run: npm ci
+
+      - name: Run Jest Tests
+        run: npx jest --coverage --runInBand tests/regr-tests

dist/jiff-client.js

@@ -1663,7 +1663,7 @@ module.exports = function (jiffClient) {
    * @memberof handlers
    */
   jiffClient.handlers.connected = function () {
-    console.log('Connected!', jiffClient.id); // TODO: remove debugging
+
     jiffClient.initialization_counter++;
 
     if (jiffClient.secret_key == null && jiffClient.public_key == null) {

jest.config.js

@@ -0,0 +1,9 @@
+/** @type {import('ts-jest').JestConfigWithTsJest} */
+module.exports = {
+  preset: 'ts-jest',
+  testEnvironment: 'node',
+  transform: {
+    '^.+\\.ts$': 'ts-jest'
+  },
+  moduleFileExtensions: ['ts', 'js', 'json', 'node']
+};

lib/jiff-client.js

@@ -464,6 +464,10 @@ function JIFFClient(hostname, computation_id, options) {
     }
   };
 
+  this.disconnect = async function () {
+    await this.socket.safe_disconnect(true);
+  };
+
   // Connect when all is done
   if (options.autoConnect !== false) {
     this.connect();

lib/server/socket.js

@@ -147,4 +147,19 @@
       });
     });
   };
+  JIFFServer.prototype.closeAllSockets = function () {
+    var jiff = this;
+
+    if (jiff.io) {
+      const socketIds = Object.keys(jiff.socketMaps.computationId);
+      socketIds.forEach(function (socketId) {
+        var socket = jiff.io.sockets.connected[socketId];
+        if (socket) {
+          socket.disconnect(true);
+        }
+      });
+
+      jiff.io.close(); // Optionally close the entire socket.io server
+    }
+  };
 };