mirror of
https://github.com/Katakate/k7.git
synced 2026-09-21 17:43:20 +00:00
G
85 lines
3.2 KiB
Python
85 lines
3.2 KiB
Python
"""Backend-agnostic integration tests for sandbox lifecycle."""
|
|
|
|
import asyncio
|
|
|
|
import pytest
|
|
|
|
from k7.core.core import K7Core
|
|
from k7.core.models import SandboxConfig
|
|
|
|
pytestmark = pytest.mark.integration
|
|
|
|
|
|
class TestSandboxLifecycle:
|
|
async def test_create_exec_delete(self, k7_core: K7Core, test_namespace: str):
|
|
cfg = SandboxConfig(
|
|
name="integ-basic",
|
|
image="alpine:latest",
|
|
namespace=test_namespace,
|
|
)
|
|
result = await k7_core.create_sandbox(cfg)
|
|
assert result.success, f"create failed: {result.error}"
|
|
|
|
try:
|
|
for _ in range(60):
|
|
sandboxes = await k7_core.list_sandboxes(namespace=test_namespace)
|
|
if any(s.name == "integ-basic" and s.ready == "True" for s in sandboxes):
|
|
break
|
|
await asyncio.sleep(2)
|
|
|
|
exec_result = await k7_core.exec_command("integ-basic", "echo hello", namespace=test_namespace)
|
|
assert exec_result.exit_code == 0
|
|
assert "hello" in exec_result.stdout
|
|
finally:
|
|
await k7_core.delete_sandbox("integ-basic", namespace=test_namespace)
|
|
|
|
async def test_before_script(self, k7_core: K7Core, test_namespace: str):
|
|
cfg = SandboxConfig(
|
|
name="integ-before",
|
|
image="alpine:latest",
|
|
namespace=test_namespace,
|
|
before_script="touch /tmp/marker",
|
|
)
|
|
result = await k7_core.create_sandbox(cfg)
|
|
assert result.success, f"create failed: {result.error}"
|
|
|
|
try:
|
|
for _ in range(60):
|
|
sandboxes = await k7_core.list_sandboxes(namespace=test_namespace)
|
|
if any(s.name == "integ-before" and s.ready == "True" for s in sandboxes):
|
|
break
|
|
await asyncio.sleep(2)
|
|
|
|
exec_result = await k7_core.exec_command(
|
|
"integ-before", "test -f /tmp/marker && echo ok", namespace=test_namespace
|
|
)
|
|
assert exec_result.exit_code == 0
|
|
assert "ok" in exec_result.stdout
|
|
finally:
|
|
await k7_core.delete_sandbox("integ-before", namespace=test_namespace)
|
|
|
|
async def test_resource_limits(self, k7_core: K7Core, test_namespace: str):
|
|
cfg = SandboxConfig(
|
|
name="integ-limits",
|
|
image="alpine:latest",
|
|
namespace=test_namespace,
|
|
limits={"cpu": "250m", "memory": "256Mi"},
|
|
)
|
|
result = await k7_core.create_sandbox(cfg)
|
|
assert result.success, f"create failed: {result.error}"
|
|
|
|
try:
|
|
v1 = await k7_core._get_core_v1_client()
|
|
pods = None
|
|
for _ in range(60):
|
|
pods = await v1.list_namespaced_pod(namespace=test_namespace, label_selector="app=integ-limits")
|
|
if pods.items:
|
|
break
|
|
await asyncio.sleep(2)
|
|
assert pods and len(pods.items) > 0, "Pod never appeared"
|
|
container = pods.items[0].spec.containers[0]
|
|
assert container.resources.limits["cpu"] == "250m"
|
|
assert container.resources.limits["memory"] == "256Mi"
|
|
finally:
|
|
await k7_core.delete_sandbox("integ-limits", namespace=test_namespace)
|