wehub-resource-sync
db620d33df
CodeQL / Analyze (csharp) (push) Waiting to run
CodeQL / Analyze (python) (push) Waiting to run
dotnet-build-and-test / dotnet-test-functions (push) Has been cancelled
dotnet-build-and-test / paths-filter (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Debug, windows-latest, net9.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Release, ubuntu-latest, net10.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Release, ubuntu-latest, net8.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Release, windows-latest, net472) (push) Has been cancelled
dotnet-build-and-test / dotnet-test (Release, integration, true, ubuntu-latest, net10.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-test (Release, integration, true, windows-latest, net472) (push) Has been cancelled
dotnet-build-and-test / dotnet-foundry-hosted-it (push) Has been cancelled
dotnet-build-and-test / dotnet-build-and-test-check (push) Has been cancelled
dotnet-build-and-test / Integration Test Report (push) Has been cancelled
52 lines
2.0 KiB
Python
52 lines
2.0 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
import pytest
|
|
|
|
from agent_framework_a2a._utils import get_uri_data
|
|
|
|
|
|
def test_get_uri_data_valid() -> None:
|
|
"""Test get_uri_data with valid data URIs."""
|
|
# Simple text/plain
|
|
uri = "data:text/plain;base64,SGVsbG8sIFdvcmxkIQ=="
|
|
assert get_uri_data(uri) == "SGVsbG8sIFdvcmxkIQ=="
|
|
|
|
# Image png
|
|
uri = "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgfFcSJAAAADUlEQVR42mP8/5+hHgAHggJ/PchI7wAAAABJRU5ErkJggg=="
|
|
assert get_uri_data(uri) == "iVBORw0KGgoAAAANSUhEUgfFcSJAAAADUlEQVR42mP8/5+hHgAHggJ/PchI7wAAAABJRU5ErkJggg=="
|
|
|
|
# Application octet-stream
|
|
uri = "data:application/octet-stream;base64,AQIDBA=="
|
|
assert get_uri_data(uri) == "AQIDBA=="
|
|
|
|
# Media type with parameters
|
|
uri = "data:text/plain;charset=utf-8;base64,SGVsbG8sIFdvcmxkIQ=="
|
|
assert get_uri_data(uri) == "SGVsbG8sIFdvcmxkIQ=="
|
|
|
|
# Media type with multiple parameters
|
|
uri = "data:text/plain;charset=utf-8;name=hello.txt;base64,SGVsbG8sIFdvcmxkIQ=="
|
|
assert get_uri_data(uri) == "SGVsbG8sIFdvcmxkIQ=="
|
|
|
|
|
|
def test_get_uri_data_invalid_format() -> None:
|
|
"""Test get_uri_data with invalid URI formats."""
|
|
invalid_uris = [
|
|
"not-a-uri",
|
|
"http://example.com",
|
|
"data:text/plain;SGVsbG8sIFdvcmxkIQ==", # Missing base64 marker
|
|
"data:base64,SGVsbG8sIFdvcmxkIQ==", # Missing media type
|
|
"data:text/plain;foo;base64,SGVsbG8sIFdvcmxkIQ==", # Parameter without value
|
|
"data:text/plain;base64;base64,SGVsbG8sIFdvcmxkIQ==", # base64 used as a parameter name
|
|
"data:text/plain;base64,SGVsbG8sIFdvcmxkIQ== extra",
|
|
"data:text/plain;base64,SGVsbG8sIFdvcmxkIQ==\n",
|
|
]
|
|
for uri in invalid_uris:
|
|
with pytest.raises(ValueError, match="Invalid data URI format"):
|
|
get_uri_data(uri)
|
|
|
|
|
|
def test_get_uri_data_empty() -> None:
|
|
"""Test get_uri_data with empty string."""
|
|
with pytest.raises(ValueError, match="Invalid data URI format"):
|
|
get_uri_data("")
|