mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-08-12 21:29:02 +08:00
fix: split dir for opendal tests (#11627)
Signed-off-by: yihong0618 <zouzou0208@gmail.com>
This commit is contained in:
parent
430ca3322b
commit
194bc60429
@ -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"
|
||||
|
||||
|
@ -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()
|
||||
|
Loading…
x
Reference in New Issue
Block a user