mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 03:55:55 +00:00
3ba727f0e4
Add comprehensive tests for B-class issues (core functionality untested): Pipeline: - test_pool.py: QueryPool ID generation, caching, async context (12 tests) - test_ratelimit.py: Fixed timing-sensitive test tolerance - test_pipelinemgr.py: Use real Pydantic StageProcessResult instead of Mock Utils: - test_version.py: Version comparison functions (20 tests) - test_logcache.py: Log page management and retrieval (18 tests) - test_httpclient.py: HTTP session pool management (10 tests) - test_proxy.py: Proxy configuration from env and config (10 tests) - test_image.py: URL parsing and base64 extraction (12 tests) - test_pkgmgr.py: Pip command generation (8 tests) Discover: - test_engine.py: I18nString, Metadata, Component manifest (15 tests) Test count: 1193 → 1298 (+105 tests) Note: Some B-class issues cannot be tested due to circular import bugs filed as GitHub issues #2175 (pipeline) and #2176 (persistence).
143 lines
4.8 KiB
Python
143 lines
4.8 KiB
Python
"""
|
|
Unit tests for image utility functions.
|
|
|
|
Tests URL parsing and base64 extraction without network calls.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
import base64
|
|
|
|
from langbot.pkg.utils.image import (
|
|
get_qq_image_downloadable_url,
|
|
extract_b64_and_format,
|
|
)
|
|
|
|
|
|
class TestGetQQImageDownloadableUrl:
|
|
"""Tests for get_qq_image_downloadable_url function."""
|
|
|
|
def test_basic_url(self):
|
|
"""Parse basic image URL."""
|
|
url = "http://example.com/image.jpg"
|
|
result_url, query = get_qq_image_downloadable_url(url)
|
|
|
|
assert result_url == "http://example.com/image.jpg"
|
|
assert query == {}
|
|
|
|
def test_url_with_query_params(self):
|
|
"""Parse URL with query parameters."""
|
|
url = "http://example.com/image.jpg?param1=value1¶m2=value2"
|
|
result_url, query = get_qq_image_downloadable_url(url)
|
|
|
|
assert result_url == "http://example.com/image.jpg"
|
|
assert query == {"param1": ["value1"], "param2": ["value2"]}
|
|
|
|
def test_url_with_port(self):
|
|
"""Parse URL with port number."""
|
|
url = "http://example.com:8080/image.jpg"
|
|
result_url, query = get_qq_image_downloadable_url(url)
|
|
|
|
assert result_url == "http://example.com:8080/image.jpg"
|
|
|
|
def test_url_with_path(self):
|
|
"""Parse URL with complex path."""
|
|
url = "http://example.com/path/to/image.jpg"
|
|
result_url, query = get_qq_image_downloadable_url(url)
|
|
|
|
assert result_url == "http://example.com/path/to/image.jpg"
|
|
|
|
def test_url_with_fragment(self):
|
|
"""Parse URL with fragment (fragment is not part of query)."""
|
|
url = "http://example.com/image.jpg#fragment"
|
|
result_url, query = get_qq_image_downloadable_url(url)
|
|
|
|
# Fragment is not included in query string parsing
|
|
assert "http://example.com/image.jpg" in result_url
|
|
|
|
def test_https_url(self):
|
|
"""Parse HTTPS URL - note: function returns http:// regardless of input scheme."""
|
|
url = "https://example.com/image.jpg"
|
|
result_url, query = get_qq_image_downloadable_url(url)
|
|
|
|
# The function constructs URL with http:// scheme
|
|
assert "example.com/image.jpg" in result_url
|
|
|
|
|
|
class TestExtractB64AndFormat:
|
|
"""Tests for extract_b64_and_format function."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_jpeg_data_uri(self):
|
|
"""Extract base64 and format from JPEG data URI."""
|
|
# Create a simple base64 string
|
|
original_data = b"test image data"
|
|
b64_data = base64.b64encode(original_data).decode()
|
|
data_uri = f"data:image/jpeg;base64,{b64_data}"
|
|
|
|
result_b64, result_format = await extract_b64_and_format(data_uri)
|
|
|
|
assert result_b64 == b64_data
|
|
assert result_format == "jpeg"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_png_data_uri(self):
|
|
"""Extract base64 and format from PNG data URI."""
|
|
original_data = b"test png data"
|
|
b64_data = base64.b64encode(original_data).decode()
|
|
data_uri = f"data:image/png;base64,{b64_data}"
|
|
|
|
result_b64, result_format = await extract_b64_and_format(data_uri)
|
|
|
|
assert result_b64 == b64_data
|
|
assert result_format == "png"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_gif_data_uri(self):
|
|
"""Extract base64 and format from GIF data URI."""
|
|
original_data = b"test gif data"
|
|
b64_data = base64.b64encode(original_data).decode()
|
|
data_uri = f"data:image/gif;base64,{b64_data}"
|
|
|
|
result_b64, result_format = await extract_b64_and_format(data_uri)
|
|
|
|
assert result_b64 == b64_data
|
|
assert result_format == "gif"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_webp_data_uri(self):
|
|
"""Extract base64 and format from WebP data URI."""
|
|
original_data = b"test webp data"
|
|
b64_data = base64.b64encode(original_data).decode()
|
|
data_uri = f"data:image/webp;base64,{b64_data}"
|
|
|
|
result_b64, result_format = await extract_b64_and_format(data_uri)
|
|
|
|
assert result_b64 == b64_data
|
|
assert result_format == "webp"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_complex_base64(self):
|
|
"""Handle base64 with special characters."""
|
|
# Base64 can include + and / characters
|
|
original_data = bytes(range(256)) # All byte values
|
|
b64_data = base64.b64encode(original_data).decode()
|
|
data_uri = f"data:image/png;base64,{b64_data}"
|
|
|
|
result_b64, result_format = await extract_b64_and_format(data_uri)
|
|
|
|
assert result_b64 == b64_data
|
|
# Verify we can decode back to original
|
|
assert base64.b64decode(result_b64) == original_data
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_empty_base64(self):
|
|
"""Handle empty base64 string."""
|
|
data_uri = "data:image/png;base64,"
|
|
|
|
result_b64, result_format = await extract_b64_and_format(data_uri)
|
|
|
|
assert result_b64 == ""
|
|
assert result_format == "png"
|