|
|
|
@@ -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() |