re-structure
This commit is contained in:
@@ -3,7 +3,7 @@
|
||||
import pytest
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
from inference.web.services.storage_helpers import StorageHelper, get_storage_helper
|
||||
from backend.web.services.storage_helpers import StorageHelper, get_storage_helper
|
||||
from shared.storage import PREFIXES
|
||||
|
||||
|
||||
@@ -402,10 +402,10 @@ class TestGetStorageHelper:
|
||||
|
||||
def test_returns_helper_instance(self) -> None:
|
||||
"""Should return a StorageHelper instance."""
|
||||
with patch("inference.web.services.storage_helpers.get_default_storage") as mock_get:
|
||||
with patch("backend.web.services.storage_helpers.get_default_storage") as mock_get:
|
||||
mock_get.return_value = MagicMock()
|
||||
# Reset the global helper
|
||||
import inference.web.services.storage_helpers as module
|
||||
import backend.web.services.storage_helpers as module
|
||||
module._default_helper = None
|
||||
|
||||
helper = get_storage_helper()
|
||||
@@ -414,9 +414,9 @@ class TestGetStorageHelper:
|
||||
|
||||
def test_returns_same_instance(self) -> None:
|
||||
"""Should return the same instance on subsequent calls."""
|
||||
with patch("inference.web.services.storage_helpers.get_default_storage") as mock_get:
|
||||
with patch("backend.web.services.storage_helpers.get_default_storage") as mock_get:
|
||||
mock_get.return_value = MagicMock()
|
||||
import inference.web.services.storage_helpers as module
|
||||
import backend.web.services.storage_helpers as module
|
||||
module._default_helper = None
|
||||
|
||||
helper1 = get_storage_helper()
|
||||
|
||||
Reference in New Issue
Block a user