From d73b90e6f81feddf2accab21089b18b3509fb72d Mon Sep 17 00:00:00 2001
From: srgooglo <srgooglo@ragestudio.net>
Date: Wed, 23 Feb 2022 13:56:11 +0100
Subject: [PATCH] update test

---
 utests/basicServerAndBridge.js | 49 +++++++++++++++++++++++++---------
 1 file changed, 37 insertions(+), 12 deletions(-)

diff --git a/utests/basicServerAndBridge.js b/utests/basicServerAndBridge.js
index cbbdcc5..efcbd25 100644
--- a/utests/basicServerAndBridge.js
+++ b/utests/basicServerAndBridge.js
@@ -52,28 +52,53 @@ const Controllers = [
 ]
 
 async function _main() {
-    const server = new Server(undefined, Controllers, Middlewares)
+    const server = new Server({
+        onWSClientConnection: (socket) => {
+            const authToken = socket.handshake.auth?.token
+            console.log(`AUTH TOKEN: ${authToken}`)
+
+            if (!authToken) {
+                socket.emit("unauthorized", "No auth token provided!")
+                return socket.disconnect()
+            }
+
+            if (authToken !== "123") {
+                socket.emit("unauthorized", "invalid auth token!")
+                return socket.disconnect()
+            }
+        }
+    }, Controllers, Middlewares)
+
     const clientBridge = new Bridge({
         origin: server.HTTPAddress,
         wsOrigin: server.WSAddress,
+        wsMainSocketOptions: {
+            auth: {
+                token: "123"
+            }
+        },
+    }, {
+        onUnauthorized: (reason) => {
+            console.log(reason)
+        }
     })
 
     await server.initialize()
     await clientBridge.initialize()
 
-    const test = await clientBridge.endpoints.get.test()
-    const crashTest = await clientBridge.endpoints.get.crashtest().catch(error => {
-        console.log(error)
-        return false
-    })
-    const deleteTest = await clientBridge.endpoints.delete.test({
-        a: "test"
-    })
+    // const test = await clientBridge.endpoints.get.test()
+    // const crashTest = await clientBridge.endpoints.get.crashtest().catch(error => {
+    //     console.log(error)
+    //     return false
+    // })
+    // const deleteTest = await clientBridge.endpoints.delete.test({
+    //     a: "test"
+    // })
     const wsEpicEvent = await clientBridge.wsEndpoints.epicEvent("Hello", "World")
 
-    console.log(`[get.test] > ${test}`)
-    console.log(`[get.crashtest] > ${crashTest}`)
-    console.log(`[delete.test] > ${deleteTest}`)
+    // console.log(`[get.test] > ${test}`)
+    // console.log(`[get.crashtest] > ${crashTest}`)
+    // console.log(`[delete.test] > ${deleteTest}`)
     console.log(`[ws.epicEvent] > ${wsEpicEvent}`)
 }