Method for Socket disconnection added

multiparty · Mar 28, 2024 · ede15cc · ede15cc
1 parent 24c990d
commit ede15cc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 16 deletions.
diff --git a/lib/jiff-client.js b/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();

diff --git a/tests/arithmetics/arithmetics.test.ts b/tests/arithmetics/arithmetics.test.ts
@@ -48,23 +48,12 @@ describe('JIFF Arithmetic Operations', () => {
   });
 
   afterEach(async () => {
-    // Shutting Server
-    await jiffClient1.socket.onclose('io client disconnect')
-    await jiffClient2.socket.onclose('io client disconnect')
+    // Shutting down client
+    await jiffClient1.socket.disconnect()
+    await jiffClient2.socket.disconnect()
+
+    // Shutting down Server
     await jiffServer.closeAllSockets();
-    // jiffServer.freeComputation(computation_id);
-
-    // Close the server
-    // await new Promise((resolve, reject) => {
-    //   server.close((err:string) => {
-    //     if (err) {
-    //       console.error('Error closing server:', err);
-    //       reject(err);
-    //       return;
-    //     }
-    //     resolve(console.log('Server closed'));
-    //   });
-    // });
   });
 
   it('should correctly add 60 + 60 = 120', async () => {