|
| 1 | +from dataclasses import dataclass |
| 2 | +from typing import Optional |
| 3 | +from urllib.parse import parse_qs |
| 4 | +from urllib.parse import urlparse |
| 5 | + |
| 6 | +import pytest |
| 7 | +from _utils import get_client |
| 8 | + |
| 9 | +from speechmatics.voice import __version__ |
| 10 | + |
| 11 | + |
| 12 | +@dataclass |
| 13 | +class URLExample: |
| 14 | + input_url: str |
| 15 | + input_app: Optional[str] = None |
| 16 | + |
| 17 | + |
| 18 | +URLS: list[URLExample] = [ |
| 19 | + URLExample( |
| 20 | + input_url="wss://dummy/ep", |
| 21 | + input_app="dummy-0.1.2", |
| 22 | + ), |
| 23 | + URLExample( |
| 24 | + input_url="wss://dummy:1234/ep?client=amz", |
| 25 | + input_app="dummy-0.1.2", |
| 26 | + ), |
| 27 | + URLExample( |
| 28 | + input_url="wss://dummy/ep?sm-app=dummy", |
| 29 | + ), |
| 30 | + URLExample( |
| 31 | + input_url="ws://localhost:8080/ep?sm-app=dummy", |
| 32 | + input_app="dummy-0.1.2", |
| 33 | + ), |
| 34 | + URLExample( |
| 35 | + input_url="http://dummy/ep/v1/", |
| 36 | + input_app="dummy-0.1.2", |
| 37 | + ), |
| 38 | + URLExample( |
| 39 | + input_url="wss://dummy/ep", |
| 40 | + ), |
| 41 | + URLExample( |
| 42 | + input_url="wss://dummy/ep", |
| 43 | + input_app="client/a#b:c^d", |
| 44 | + ), |
| 45 | +] |
| 46 | + |
| 47 | + |
| 48 | +@pytest.mark.asyncio |
| 49 | +@pytest.mark.parametrize("test", URLS, ids=lambda s: s.input_url) |
| 50 | +async def test_url_endpoints(test: URLExample): |
| 51 | + """Test URL endpoint construction.""" |
| 52 | + |
| 53 | + # Client |
| 54 | + client = await get_client( |
| 55 | + api_key="DUMMY", |
| 56 | + connect=False, |
| 57 | + ) |
| 58 | + |
| 59 | + # Parse the input parameters |
| 60 | + input_parsed = urlparse(test.input_url) |
| 61 | + input_params = parse_qs(input_parsed.query, keep_blank_values=True) |
| 62 | + |
| 63 | + # URL test |
| 64 | + generated_url = client._get_endpoint_url(test.input_url, test.input_app) |
| 65 | + |
| 66 | + # Parse the URL |
| 67 | + parsed_url = urlparse(generated_url) |
| 68 | + parsed_params = parse_qs(parsed_url.query, keep_blank_values=True) |
| 69 | + |
| 70 | + # Check the url scheme, netloc and path are preserved |
| 71 | + assert parsed_url.scheme == input_parsed.scheme |
| 72 | + assert parsed_url.netloc == input_parsed.netloc |
| 73 | + assert parsed_url.path == input_parsed.path |
| 74 | + |
| 75 | + # Validate `sm-app` |
| 76 | + if test.input_app: |
| 77 | + assert parsed_params["sm-app"] == [test.input_app] |
| 78 | + elif "sm-app" in input_params: |
| 79 | + assert parsed_params["sm-app"] == [input_params["sm-app"][0]] |
| 80 | + else: |
| 81 | + assert parsed_params["sm-app"] == [f"voice-sdk/{__version__}"] |
| 82 | + |
| 83 | + # Validate `sm-voice-sdk` |
| 84 | + assert parsed_params["sm-voice-sdk"] == [__version__] |
| 85 | + |
| 86 | + # Check other original params are preserved |
| 87 | + for key, value in input_params.items(): |
| 88 | + if key not in ["sm-app", "sm-voice-sdk"]: |
| 89 | + assert parsed_params[key] == value |
0 commit comments