re-structure
This commit is contained in:
@@ -10,11 +10,11 @@ from unittest.mock import MagicMock, patch
|
||||
|
||||
import pytest
|
||||
|
||||
from inference.data.async_request_db import AsyncRequest
|
||||
from inference.web.workers.async_queue import AsyncTask, AsyncTaskQueue
|
||||
from inference.web.services.async_processing import AsyncProcessingService, AsyncSubmitResult
|
||||
from inference.web.config import AsyncConfig, StorageConfig
|
||||
from inference.web.rate_limiter import RateLimiter
|
||||
from backend.data.async_request_db import AsyncRequest
|
||||
from backend.web.workers.async_queue import AsyncTask, AsyncTaskQueue
|
||||
from backend.web.services.async_processing import AsyncProcessingService, AsyncSubmitResult
|
||||
from backend.web.config import AsyncConfig, StorageConfig
|
||||
from backend.web.rate_limiter import RateLimiter
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
@@ -231,7 +231,7 @@ class TestAsyncProcessingService:
|
||||
mock_db.get_request.return_value = None
|
||||
|
||||
# Mock the storage helper to return the same directory as the fixture
|
||||
with patch("inference.web.services.async_processing.get_storage_helper") as mock_storage:
|
||||
with patch("backend.web.services.async_processing.get_storage_helper") as mock_storage:
|
||||
mock_helper = MagicMock()
|
||||
mock_helper.get_uploads_base_path.return_value = temp_dir
|
||||
mock_storage.return_value = mock_helper
|
||||
|
||||
Reference in New Issue
Block a user