fix: stabilize post-MVP travel-agent and itinerary workflows

This commit is contained in:
2026-03-08 16:51:19 +00:00
parent fb2347345f
commit 2fd11dbd26
27 changed files with 2533 additions and 794 deletions

View File

@@ -0,0 +1,38 @@
from django.contrib.auth import get_user_model
from rest_framework.authtoken.models import Token
from rest_framework.test import APIClient, APITestCase
User = get_user_model()
class MCPTokenEndpointTests(APITestCase):
def setUp(self):
self.user = User.objects.create_user(
username="mcp-token-user",
email="mcp-token@example.com",
password="password123",
)
def test_requires_authentication(self):
unauthenticated_client = APIClient()
response = unauthenticated_client.get("/auth/mcp-token/")
self.assertIn(response.status_code, [401, 403])
def test_returns_token_for_authenticated_user(self):
self.client.force_authenticate(user=self.user)
response = self.client.get("/auth/mcp-token/")
self.assertEqual(response.status_code, 200)
self.assertIn("token", response.json())
self.assertTrue(Token.objects.filter(user=self.user).exists())
def test_reuses_existing_token(self):
existing_token = Token.objects.create(user=self.user)
self.client.force_authenticate(user=self.user)
response = self.client.get("/auth/mcp-token/")
self.assertEqual(response.status_code, 200)
self.assertEqual(response.json().get("token"), existing_token.key)
self.assertEqual(Token.objects.filter(user=self.user).count(), 1)