From 3e03df43c1610c52e1d6ecfe8ade26d1faa14220 Mon Sep 17 00:00:00 2001
From: Sayali Sable <sayalisable23@gmail.com>
Date: Wed, 18 Feb 2026 22:59:45 -0500
Subject: [PATCH] feat: add test endpoint for setting user teamCode

---
 src/app.js               |  3 +++
 src/routes/testRoutes.js | 27 +++++++++++++++++++++++++++
 2 files changed, 30 insertions(+)
 create mode 100644 src/routes/testRoutes.js

diff --git a/src/app.js b/src/app.js
index 64aa02b75..66ffb66da 100644
--- a/src/app.js
+++ b/src/app.js
@@ -1,5 +1,6 @@
 const express = require('express');
 const Sentry = require('@sentry/node');
+const testRoutes = require('./routes/testRoutes');
 
 const app = express();
 const logger = require('./startup/logger');
@@ -15,6 +16,8 @@ require('./startup/compression')(app);
 require('./startup/cors')(app);
 require('./startup/bodyParser')(app);
 
+app.use('/api/test', testRoutes);
+
 const helpFeedbackRouter = require('./routes/helpFeedbackRouter');
 const helpRequestRouter = require('./routes/helpRequestRouter');
 
diff --git a/src/routes/testRoutes.js b/src/routes/testRoutes.js
new file mode 100644
index 000000000..8fe88af2d
--- /dev/null
+++ b/src/routes/testRoutes.js
@@ -0,0 +1,27 @@
+const express = require('express');
+
+const router = express.Router();
+
+router.post('/set-team-code', async (req, res) => {
+  try {
+    const userProfile = require('../models/userProfile');
+    const { userId, teamCode } = req.body;
+
+    if (!userId || !teamCode) {
+      return res.status(400).json({ error: 'userId and teamCode required' });
+    }
+
+    const updated = await userProfile.findByIdAndUpdate(userId, { teamCode }, { new: true });
+
+    if (!updated) {
+      return res.status(404).json({ error: 'User not found' });
+    }
+
+    res.json({ success: true, user: updated });
+  } catch (err) {
+    console.error('Error:', err);
+    res.status(500).json({ error: err.message });
+  }
+});
+
+module.exports = router;