restructure project
This commit is contained in:
@@ -8,9 +8,9 @@ from unittest.mock import MagicMock, patch
|
||||
|
||||
from fastapi import HTTPException
|
||||
|
||||
from src.data.admin_db import AdminDB
|
||||
from src.data.admin_models import AdminToken
|
||||
from src.web.core.auth import (
|
||||
from inference.data.admin_db import AdminDB
|
||||
from inference.data.admin_models import AdminToken
|
||||
from inference.web.core.auth import (
|
||||
get_admin_db,
|
||||
reset_admin_db,
|
||||
validate_admin_token,
|
||||
@@ -81,7 +81,7 @@ class TestAdminDB:
|
||||
|
||||
def test_is_valid_admin_token_active(self):
|
||||
"""Test valid active token."""
|
||||
with patch("src.data.admin_db.get_session_context") as mock_ctx:
|
||||
with patch("inference.data.admin_db.get_session_context") as mock_ctx:
|
||||
mock_session = MagicMock()
|
||||
mock_ctx.return_value.__enter__.return_value = mock_session
|
||||
|
||||
@@ -98,7 +98,7 @@ class TestAdminDB:
|
||||
|
||||
def test_is_valid_admin_token_inactive(self):
|
||||
"""Test inactive token."""
|
||||
with patch("src.data.admin_db.get_session_context") as mock_ctx:
|
||||
with patch("inference.data.admin_db.get_session_context") as mock_ctx:
|
||||
mock_session = MagicMock()
|
||||
mock_ctx.return_value.__enter__.return_value = mock_session
|
||||
|
||||
@@ -115,7 +115,7 @@ class TestAdminDB:
|
||||
|
||||
def test_is_valid_admin_token_expired(self):
|
||||
"""Test expired token."""
|
||||
with patch("src.data.admin_db.get_session_context") as mock_ctx:
|
||||
with patch("inference.data.admin_db.get_session_context") as mock_ctx:
|
||||
mock_session = MagicMock()
|
||||
mock_ctx.return_value.__enter__.return_value = mock_session
|
||||
|
||||
@@ -132,7 +132,7 @@ class TestAdminDB:
|
||||
|
||||
def test_is_valid_admin_token_not_found(self):
|
||||
"""Test token not found."""
|
||||
with patch("src.data.admin_db.get_session_context") as mock_ctx:
|
||||
with patch("inference.data.admin_db.get_session_context") as mock_ctx:
|
||||
mock_session = MagicMock()
|
||||
mock_ctx.return_value.__enter__.return_value = mock_session
|
||||
mock_session.get.return_value = None
|
||||
|
||||
Reference in New Issue
Block a user