From ec9e40f0280391abc652d734a23d93ec65f58694 Mon Sep 17 00:00:00 2001
From: Alexander Borg <borg@servernet.se>
Date: Tue, 16 Sep 2025 21:45:06 +0200
Subject: [PATCH] Fix jwt-token

---
 server/routes/auth.js | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/server/routes/auth.js b/server/routes/auth.js
index 898f3c3..1b765b2 100644
--- a/server/routes/auth.js
+++ b/server/routes/auth.js
@@ -262,7 +262,21 @@ router.post('/login', async (req, res, next) => {
  * POST /auth/register
  * Universal registration endpoint that routes to appropriate provider
  */
-router.post('/register', validateRequest(registerSchema), ipRestriction.checkIPRestriction.bind(ipRestriction), async (req, res, next) => {
+router.post('/register', validateRequest(registerSchema), async (req, res, next) => {
+  // Set test models if available (for testing)
+  if (global.__TEST_MODELS__) {
+    ipRestriction.setModels(global.__TEST_MODELS__);
+  }
+  
+  // Apply IP restriction check
+  await new Promise((resolve, reject) => {
+    ipRestriction.checkIPRestriction(req, res, (err) => {
+      if (err) reject(err);
+      else resolve();
+    });
+  });
+  
+  // Continue with registration if IP check passed
   try {
     // Determine tenant
     const tenantId = await multiAuth.determineTenant(req);