Merge branch 'refactor/backend-architecture' into development
This commit is contained in:
@@ -2,8 +2,8 @@ import os
|
||||
import unittest
|
||||
from unittest.mock import Mock, patch
|
||||
|
||||
from api.services.memoir_images.provider import LiblibImageProvider
|
||||
from api.services.memoir_images.settings import DEFAULT_LIBLIB_TEMPLATE_UUID
|
||||
from app.features.memoir.memoir_images.provider import LiblibImageProvider
|
||||
from app.features.memoir.memoir_images.settings import DEFAULT_LIBLIB_TEMPLATE_UUID
|
||||
|
||||
|
||||
def _make_provider(http_client=None):
|
||||
@@ -193,7 +193,6 @@ class DownloadImageTest(unittest.TestCase):
|
||||
"https://liblibai-tmp-image.liblib.cloud/img/demo.png"
|
||||
)
|
||||
|
||||
@patch.dict(os.environ, {"MEMOIR_IMAGE_DOWNLOAD_HOSTS": "cdn.example.com"}, clear=False)
|
||||
def test_download_fetches_binary_payload(self):
|
||||
http_client = Mock()
|
||||
resp = Mock()
|
||||
@@ -201,7 +200,14 @@ class DownloadImageTest(unittest.TestCase):
|
||||
resp.raise_for_status = Mock()
|
||||
http_client.get.return_value = resp
|
||||
|
||||
provider = _make_provider(http_client)
|
||||
provider = LiblibImageProvider(
|
||||
http_client=http_client,
|
||||
access_key="test-ak",
|
||||
secret_key="test-sk",
|
||||
base_url="https://openapi.liblibai.cloud",
|
||||
template_uuid="tpl-uuid",
|
||||
allowed_download_hosts=("cdn.example.com",),
|
||||
)
|
||||
payload = provider.download_image({"image_url": "https://cdn.example.com/1.png"})
|
||||
|
||||
self.assertEqual(payload, b"png-bytes")
|
||||
@@ -217,7 +223,7 @@ class DownloadImageTest(unittest.TestCase):
|
||||
|
||||
|
||||
class ProviderResourceManagementTest(unittest.TestCase):
|
||||
@patch("api.services.memoir_images.provider.httpx.Client")
|
||||
@patch("app.adapters.image_gen.liblib_provider.httpx.Client")
|
||||
def test_provider_closes_owned_http_client(self, httpx_client_cls):
|
||||
http_client = Mock()
|
||||
httpx_client_cls.return_value = http_client
|
||||
|
||||
Reference in New Issue
Block a user