dify

dify/api/tests/unit_tests/libs/test_yarl.py

@@ -0,0 +1,29 @@
+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
+
+    expected_4 = "api"
+    assert str(URL("") / "api") == expected_4
+
+    expected_5 = "/api"
+    assert str(URL("/") / "api") == expected_5
+
+    with pytest.raises(ValueError) as e1:
+        str(URL("https://dify.ai") / "/api")
+    assert str(e1.value) == "Appending path '/api' starting from slash is forbidden"