Initial project import
This commit is contained in:
@@ -0,0 +1,43 @@
|
||||
import unittest
|
||||
from types import SimpleNamespace
|
||||
from unittest import mock
|
||||
|
||||
import api.support as api_support
|
||||
|
||||
|
||||
class ImageBaseUrlApiTests(unittest.TestCase):
|
||||
def setUp(self) -> None:
|
||||
self.fake_config = SimpleNamespace(base_url="https://public.example.com")
|
||||
patcher = mock.patch.object(api_support, "config", self.fake_config)
|
||||
patcher.start()
|
||||
self.addCleanup(patcher.stop)
|
||||
|
||||
def test_prefers_configured_base_url(self) -> None:
|
||||
request = SimpleNamespace(
|
||||
url=SimpleNamespace(scheme="http", netloc="127.0.0.1:8000"),
|
||||
headers={"host": "127.0.0.1:8000"},
|
||||
)
|
||||
|
||||
self.assertEqual(api_support.resolve_image_base_url(request), "https://public.example.com")
|
||||
|
||||
def test_falls_back_to_request_host(self) -> None:
|
||||
self.fake_config.base_url = ""
|
||||
request = SimpleNamespace(
|
||||
url=SimpleNamespace(scheme="http", netloc="127.0.0.1:8000"),
|
||||
headers={"host": "internal.example:9000"},
|
||||
)
|
||||
|
||||
self.assertEqual(api_support.resolve_image_base_url(request), "http://internal.example:9000")
|
||||
|
||||
def test_falls_back_to_request_netloc_when_host_missing(self) -> None:
|
||||
self.fake_config.base_url = ""
|
||||
request = SimpleNamespace(
|
||||
url=SimpleNamespace(scheme="https", netloc="public.example.com"),
|
||||
headers={},
|
||||
)
|
||||
|
||||
self.assertEqual(api_support.resolve_image_base_url(request), "https://public.example.com")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user