unit tests

tests/unit/classes/class_spec.js

@@ -7,7 +7,7 @@ describe("File js/class", function () {
 		let clone;
 		let dom;
 
-		before(function (done) {
+		beforeAll(function (done) {
 			dom = new JSDOM(
 				`<script>var Log = {log: function() {}};</script>\
 					<script src="file://${path.join(__dirname, "..", "..", "..", "js", "class.js")}">`,
@@ -87,14 +87,14 @@ describe("File js/class", function () {
 		describe("Test lockstring code", function () {
 			let log;
 
-			before(function () {
+			beforeAll(function () {
 				log = dom.window.Log.log;
 				dom.window.Log.log = function cmp(str) {
 					expect(str).to.equal("lockStrings");
 				};
 			});
 
-			after(function () {
+			afterAll(function () {
 				dom.window.Log.log = log;
 			});
 

tests/unit/classes/translator_spec.js

@@ -3,11 +3,12 @@ const path = require("path");
 const helmet = require("helmet");
 const { JSDOM } = require("jsdom");
 const express = require("express");
+const sockets = new Set();
 
 describe("Translator", function () {
 	let server;
 
-	before(function () {
+	beforeAll(function () {
 		const app = express();
 		app.use(helmet());
 		app.use(function (req, res, next) {
@@ -17,9 +18,20 @@ describe("Translator", function () {
 		app.use("/translations", express.static(path.join(__dirname, "..", "..", "..", "tests", "configs", "data")));
 
 		server = app.listen(3000);
+
+		server.on('connection', (socket) => {
+			sockets.add(socket);
+		});
+
 	});
 
-	after(function () {
+	afterAll(function () {
+		for (const socket of sockets) {
+			socket.destroy();
+	
+			sockets.delete(socket);
+		}
+
 		server.close();
 	});