re-structure

tests/web/test_admin_auth.py

@@ -8,9 +8,9 @@ from unittest.mock import MagicMock, patch
 
 from fastapi import HTTPException
 
-from inference.data.repositories import TokenRepository
-from inference.data.admin_models import AdminToken
-from inference.web.core.auth import (
+from backend.data.repositories import TokenRepository
+from backend.data.admin_models import AdminToken
+from backend.web.core.auth import (
     get_token_repository,
     reset_token_repository,
     validate_admin_token,
@@ -81,7 +81,7 @@ class TestTokenRepository:
 
     def test_is_valid_active_token(self):
         """Test valid active token."""
-        with patch("inference.data.repositories.token_repository.BaseRepository._session") as mock_ctx:
+        with patch("backend.data.repositories.token_repository.BaseRepository._session") as mock_ctx:
             mock_session = MagicMock()
             mock_ctx.return_value.__enter__.return_value = mock_session
 
@@ -98,7 +98,7 @@ class TestTokenRepository:
 
     def test_is_valid_inactive_token(self):
         """Test inactive token."""
-        with patch("inference.data.repositories.token_repository.BaseRepository._session") as mock_ctx:
+        with patch("backend.data.repositories.token_repository.BaseRepository._session") as mock_ctx:
             mock_session = MagicMock()
             mock_ctx.return_value.__enter__.return_value = mock_session
 
@@ -115,7 +115,7 @@ class TestTokenRepository:
 
     def test_is_valid_expired_token(self):
         """Test expired token."""
-        with patch("inference.data.repositories.token_repository.BaseRepository._session") as mock_ctx:
+        with patch("backend.data.repositories.token_repository.BaseRepository._session") as mock_ctx:
             mock_session = MagicMock()
             mock_ctx.return_value.__enter__.return_value = mock_session
 
@@ -138,7 +138,7 @@ class TestTokenRepository:
         This verifies the fix for comparing timezone-aware and naive datetimes.
         The auth API now creates tokens with timezone-aware expiration dates.
         """
-        with patch("inference.data.repositories.token_repository.BaseRepository._session") as mock_ctx:
+        with patch("backend.data.repositories.token_repository.BaseRepository._session") as mock_ctx:
             mock_session = MagicMock()
             mock_ctx.return_value.__enter__.return_value = mock_session
 
@@ -157,7 +157,7 @@ class TestTokenRepository:
 
     def test_is_valid_not_expired_token_timezone_aware(self):
         """Test non-expired token with timezone-aware datetime."""
-        with patch("inference.data.repositories.token_repository.BaseRepository._session") as mock_ctx:
+        with patch("backend.data.repositories.token_repository.BaseRepository._session") as mock_ctx:
             mock_session = MagicMock()
             mock_ctx.return_value.__enter__.return_value = mock_session
 
@@ -175,7 +175,7 @@ class TestTokenRepository:
 
     def test_is_valid_token_not_found(self):
         """Test token not found."""
-        with patch("inference.data.repositories.token_repository.BaseRepository._session") as mock_ctx:
+        with patch("backend.data.repositories.token_repository.BaseRepository._session") as mock_ctx:
             mock_session = MagicMock()
             mock_ctx.return_value.__enter__.return_value = mock_session
             mock_session.get.return_value = None