refactor: rename prepare.py → resolve_plan.py in both backends
lint / lint (push) Failing after 1m45s
test / unit (pull_request) Successful in 35s
test / integration (pull_request) Successful in 47s

This commit is contained in:
2026-06-08 14:12:48 +00:00
parent b872985a65
commit f95eabeb86
5 changed files with 12 additions and 12 deletions
+2 -2
View File
@@ -29,7 +29,7 @@ from .. import ActiveAgent, BottleBackend, BottleSpec
from . import cleanup as _cleanup from . import cleanup as _cleanup
from . import enumerate as _enumerate from . import enumerate as _enumerate
from . import launch as _launch from . import launch as _launch
from . import prepare as _prepare from . import resolve_plan as _resolve_plan
from .bottle import DockerBottle from .bottle import DockerBottle
from .bottle_cleanup_plan import DockerBottleCleanupPlan from .bottle_cleanup_plan import DockerBottleCleanupPlan
from .bottle_plan import DockerBottlePlan from .bottle_plan import DockerBottlePlan
@@ -49,7 +49,7 @@ class DockerBottleBackend(BottleBackend["DockerBottlePlan", "DockerBottleCleanup
return shutil.which("docker") is not None return shutil.which("docker") is not None
def _resolve_plan(self, spec: BottleSpec, *, stage_dir: Path) -> DockerBottlePlan: def _resolve_plan(self, spec: BottleSpec, *, stage_dir: Path) -> DockerBottlePlan:
return _prepare.resolve_plan(spec, stage_dir=stage_dir) return _resolve_plan.resolve_plan(spec, stage_dir=stage_dir)
@contextmanager @contextmanager
def launch(self, plan: DockerBottlePlan) -> Generator[DockerBottle, None, None]: def launch(self, plan: DockerBottlePlan) -> Generator[DockerBottle, None, None]:
+2 -2
View File
@@ -17,7 +17,7 @@ from .. import ActiveAgent, Bottle, BottleBackend, BottleSpec
from . import cleanup as _cleanup from . import cleanup as _cleanup
from . import enumerate as _enumerate from . import enumerate as _enumerate
from . import launch as _launch from . import launch as _launch
from . import prepare as _prepare from . import resolve_plan as _resolve_plan
from . import smolvm as _smolvm from . import smolvm as _smolvm
from .bottle import SmolmachinesBottle from .bottle import SmolmachinesBottle
from .bottle_cleanup_plan import SmolmachinesBottleCleanupPlan from .bottle_cleanup_plan import SmolmachinesBottleCleanupPlan
@@ -44,7 +44,7 @@ class SmolmachinesBottleBackend(
def _resolve_plan( def _resolve_plan(
self, spec: BottleSpec, *, stage_dir: Path self, spec: BottleSpec, *, stage_dir: Path
) -> SmolmachinesBottlePlan: ) -> SmolmachinesBottlePlan:
return _prepare.resolve_plan(spec, stage_dir=stage_dir) return _resolve_plan.resolve_plan(spec, stage_dir=stage_dir)
@contextmanager @contextmanager
def launch( def launch(
+8 -8
View File
@@ -50,16 +50,16 @@ class TestSmolmachinesResolveEnv(unittest.TestCase):
try: try:
with ( with (
patch("bot_bottle.backend.smolmachines.prepare.resolve_env", patch("bot_bottle.backend.smolmachines.resolve_plan.resolve_env",
return_value=resolved) as mock_resolve, return_value=resolved) as mock_resolve,
patch("bot_bottle.backend.smolmachines.prepare.smolmachines_preflight"), patch("bot_bottle.backend.smolmachines.resolve_plan.smolmachines_preflight"),
patch("bot_bottle.backend.smolmachines.prepare.smolmachines_bundle_subnet", patch("bot_bottle.backend.smolmachines.resolve_plan.smolmachines_bundle_subnet",
return_value=("10.99.0.0/24", "10.99.0.1", "10.99.0.2")), return_value=("10.99.0.0/24", "10.99.0.1", "10.99.0.2")),
patch("bot_bottle.backend.smolmachines.prepare.GitGate") as mock_gg, patch("bot_bottle.backend.smolmachines.resolve_plan.GitGate") as mock_gg,
patch("bot_bottle.backend.smolmachines.prepare.Egress") as mock_eg, patch("bot_bottle.backend.smolmachines.resolve_plan.Egress") as mock_eg,
patch("bot_bottle.backend.smolmachines.prepare.Supervise"), patch("bot_bottle.backend.smolmachines.resolve_plan.Supervise"),
patch( patch(
"bot_bottle.backend.smolmachines.prepare.agent_provision_plan" "bot_bottle.backend.smolmachines.resolve_plan.agent_provision_plan"
) as mock_app, ) as mock_app,
): ):
mock_gg.return_value.prepare.return_value = MagicMock() mock_gg.return_value.prepare.return_value = MagicMock()
@@ -75,7 +75,7 @@ class TestSmolmachinesResolveEnv(unittest.TestCase):
) )
mock_app.side_effect = lambda **kw: _make_provision(**kw) # type: ignore mock_app.side_effect = lambda **kw: _make_provision(**kw) # type: ignore
from bot_bottle.backend.smolmachines.prepare import resolve_plan from bot_bottle.backend.smolmachines.resolve_plan import resolve_plan
plan = resolve_plan(spec, stage_dir=stage) plan = resolve_plan(spec, stage_dir=stage)
mock_resolve.assert_called_once_with(manifest, "myagent") mock_resolve.assert_called_once_with(manifest, "myagent")