forked from wrenn/python-sdk
80 lines
3.0 KiB
Python
80 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from wrenn.capsule import Capsule, ExecResult
|
|
|
|
from .conftest import requires_auth
|
|
|
|
# --- Tests ---
|
|
|
|
|
|
@requires_auth
|
|
class TestAsyncCapsuleLifecycle:
|
|
@pytest.mark.asyncio
|
|
async def test_async_create_exec_destroy(self, async_minimal_capsule: Capsule):
|
|
result = await async_minimal_capsule.async_exec("echo", args=["async_hello"])
|
|
assert isinstance(result, ExecResult)
|
|
assert result.exit_code == 0
|
|
assert "async_hello" in result.stdout
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_upload_download(self, async_minimal_capsule: Capsule):
|
|
content = b"Async upload test"
|
|
await async_minimal_capsule.async_upload("/tmp/async_test.txt", content)
|
|
downloaded = await async_minimal_capsule.async_download("/tmp/async_test.txt")
|
|
assert downloaded == content
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_run_code(self, async_python_capsule: Capsule):
|
|
r = await async_python_capsule.async_run_code("42 * 2")
|
|
assert r.text == "84"
|
|
|
|
|
|
@requires_auth
|
|
class TestAsyncFilesystem:
|
|
@pytest.mark.asyncio
|
|
async def test_async_list_dir(self, async_minimal_capsule: Capsule):
|
|
await async_minimal_capsule.async_mkdir("/tmp/async_ls_test")
|
|
await async_minimal_capsule.async_upload("/tmp/async_ls_test/file.txt", b"data")
|
|
entries = await async_minimal_capsule.async_list_dir("/tmp/async_ls_test")
|
|
|
|
assert isinstance(entries, list)
|
|
assert any(e.name == "file.txt" for e in entries)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_mkdir(self, async_minimal_capsule: Capsule):
|
|
entry = await async_minimal_capsule.async_mkdir("/tmp/async_mkdir_test")
|
|
assert entry.type == "directory"
|
|
assert entry.name == "async_mkdir_test"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_remove(self, async_minimal_capsule: Capsule):
|
|
await async_minimal_capsule.async_upload("/tmp/async_rm.txt", b"bye")
|
|
|
|
entries = await async_minimal_capsule.async_list_dir("/tmp")
|
|
assert any(e.name == "async_rm.txt" for e in entries)
|
|
|
|
await async_minimal_capsule.async_remove("/tmp/async_rm.txt")
|
|
entries = await async_minimal_capsule.async_list_dir("/tmp")
|
|
assert not any(e.name == "async_rm.txt" for e in entries)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_full_filesystem_roundtrip(
|
|
self, async_minimal_capsule: Capsule
|
|
):
|
|
await async_minimal_capsule.async_mkdir("/tmp/async_rt")
|
|
await async_minimal_capsule.async_upload(
|
|
"/tmp/async_rt/file.txt", b"async content"
|
|
)
|
|
|
|
entries = await async_minimal_capsule.async_list_dir("/tmp/async_rt")
|
|
assert any(e.name == "file.txt" for e in entries)
|
|
|
|
data = await async_minimal_capsule.async_download("/tmp/async_rt/file.txt")
|
|
assert data == b"async content"
|
|
|
|
await async_minimal_capsule.async_remove("/tmp/async_rt/file.txt")
|
|
entries = await async_minimal_capsule.async_list_dir("/tmp/async_rt")
|
|
assert not any(e.name == "file.txt" for e in entries)
|