mirror of
https://github.com/langgenius/dify.git
synced 2025-09-03 22:33:55 +00:00
21 lines
635 B
Python
21 lines
635 B
Python
|
import pytest
|
||
|
|
|
||
|
|
from extensions.storage.opendal_storage import is_r2_endpoint
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
("endpoint", "expected"),
|
||
|
|
[
|
||
|
|
("https://bucket.r2.cloudflarestorage.com", True),
|
||
|
|
("https://custom-domain.r2.cloudflarestorage.com/", True),
|
||
|
|
("https://bucket.r2.cloudflarestorage.com/path", True),
|
||
|
|
("https://s3.amazonaws.com", False),
|
||
|
|
("https://storage.googleapis.com", False),
|
||
|
|
("http://localhost:9000", False),
|
||
|
|
("invalid-url", False),
|
||
|
|
("", False),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_is_r2_endpoint(endpoint: str, expected: bool):
|
||
|
|
assert is_r2_endpoint(endpoint) == expected
|