fix(daytona): make disk cap visible and use SDK enum for sandbox
state - Replace logger.warning with warnings.warn for the disk cap so users actually see it (logger was suppressed by CLI's log level config) - Use SandboxState enum instead of string literals in _ensure_sandbox_ready Signed-off-by: rovle <lovre.pesut@gmail.com>
This commit is contained in:
@@ -27,11 +27,20 @@ def _patch_daytona_imports(monkeypatch):
|
||||
"""Patch the daytona SDK so DaytonaEnvironment can be imported without it."""
|
||||
import types as _types
|
||||
|
||||
import enum
|
||||
|
||||
class _SandboxState(str, enum.Enum):
|
||||
STARTED = "started"
|
||||
STOPPED = "stopped"
|
||||
ARCHIVED = "archived"
|
||||
ERROR = "error"
|
||||
|
||||
daytona_mod = _types.ModuleType("daytona")
|
||||
daytona_mod.Daytona = MagicMock
|
||||
daytona_mod.CreateSandboxFromImageParams = MagicMock
|
||||
daytona_mod.DaytonaError = type("DaytonaError", (Exception,), {})
|
||||
daytona_mod.Resources = MagicMock(name="Resources")
|
||||
daytona_mod.SandboxState = _SandboxState
|
||||
|
||||
monkeypatch.setitem(__import__("sys").modules, "daytona", daytona_mod)
|
||||
return daytona_mod
|
||||
|
||||
Reference in New Issue
Block a user