Просмотр исходного кода

fix: split dir for opendal tests (#11627)

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
tags/0.14.0
yihong 10 месяцев назад
Родитель
Сommit
194bc60429
Аккаунт пользователя с таким Email не найден

+ 4
- 0
api/tests/unit_tests/oss/__mock/base.py Просмотреть файл

@@ -13,6 +13,10 @@ def get_example_bucket() -> str:
return "dify"


def get_opendal_bucket() -> str:
return "./dify"


def get_example_filename() -> str:
return "test.txt"


+ 17
- 18
api/tests/unit_tests/oss/opendal/test_opendal.py Просмотреть файл

@@ -10,7 +10,7 @@ from tests.unit_tests.oss.__mock.base import (
get_example_data,
get_example_filename,
get_example_filepath,
get_example_folder,
get_opendal_bucket,
)


@@ -20,20 +20,24 @@ class TestOpenDAL:
"""Executed before each test method."""
self.storage = OpenDALStorage(
scheme=OpenDALScheme.FS,
root=get_example_folder(),
root=get_opendal_bucket(),
)

def teardown_method(self, method):
"""Clean up after each test method."""
try:
if self.storage.exists(get_example_filename()):
self.storage.delete(get_example_filename())
@pytest.fixture(scope="class", autouse=True)
def teardown_class(self, request):
"""Clean up after all tests in the class."""

filepath = Path(get_example_filepath())
if filepath.exists():
filepath.unlink()
except:
pass
def cleanup():
folder = Path(get_opendal_bucket())
if folder.exists() and folder.is_dir():
for item in folder.iterdir():
if item.is_file():
item.unlink()
elif item.is_dir():
item.rmdir()
folder.rmdir()

return cleanup()

def test_save_and_exists(self):
"""Test saving data and checking existence."""
@@ -66,17 +70,12 @@ class TestOpenDAL:
def test_download(self):
"""Test downloading data to a file."""
filename = get_example_filename()
filepath = get_example_filepath()
filepath = str(Path(get_opendal_bucket()) / filename)
data = get_example_data()

self.storage.save(filename, data)
self.storage.download(filename, filepath)

downloaded_path = Path(filepath)
assert downloaded_path.exists()
downloaded_data = downloaded_path.read_bytes()
assert downloaded_data == data

def test_delete(self):
"""Test deleting a file."""
filename = get_example_filename()

Загрузка…
Отмена
Сохранить