mirror of
https://github.com/langgenius/dify.git
synced 2025-10-31 19:03:09 +00:00
24 lines
941 B
Python
24 lines
941 B
Python
|
import pytest
|
||
|
|
from yarl import URL
|
||
|
|
|
||
|
|
|
||
|
|
def test_yarl_urls():
|
||
|
|
expected_1 = 'https://dify.ai/api'
|
||
|
|
assert str(URL('https://dify.ai') / 'api') == expected_1
|
||
|
|
assert str(URL('https://dify.ai/') / 'api') == expected_1
|
||
|
|
|
||
|
|
expected_2 = 'http://dify.ai:12345/api'
|
||
|
|
assert str(URL('http://dify.ai:12345') / 'api') == expected_2
|
||
|
|
assert str(URL('http://dify.ai:12345/') / 'api') == expected_2
|
||
|
|
|
||
|
|
expected_3 = 'https://dify.ai/api/v1'
|
||
|
|
assert str(URL('https://dify.ai') / 'api' / 'v1') == expected_3
|
||
|
|
assert str(URL('https://dify.ai') / 'api/v1') == expected_3
|
||
|
|
assert str(URL('https://dify.ai/') / 'api/v1') == expected_3
|
||
|
|
assert str(URL('https://dify.ai/api') / 'v1') == expected_3
|
||
|
|
assert str(URL('https://dify.ai/api/') / 'v1') == expected_3
|
||
|
|
|
||
|
|
with pytest.raises(ValueError) as e1:
|
||
|
|
str(URL('https://dify.ai') / '/api')
|
||
|
|
assert str(e1.value) == "Appending path '/api' starting from slash is forbidden"
|