72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
"""Unit: Apple Container bottle command construction."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from bot_bottle.backend.macos_container.bottle import MacosContainerBottle
|
|
|
|
|
|
class TestMacosContainerBottle(unittest.TestCase):
|
|
def test_agent_argv_uses_container_exec(self):
|
|
bottle = MacosContainerBottle(
|
|
"bot-bottle-dev-abc",
|
|
lambda: None,
|
|
None,
|
|
agent_command="codex",
|
|
)
|
|
self.assertEqual(
|
|
[
|
|
"container", "exec", "--interactive", "--tty",
|
|
"bot-bottle-dev-abc", "codex", "run",
|
|
],
|
|
bottle.agent_argv(["run"]),
|
|
)
|
|
|
|
def test_agent_argv_includes_workdir(self):
|
|
bottle = MacosContainerBottle(
|
|
"bot-bottle-dev-abc",
|
|
lambda: None,
|
|
None,
|
|
agent_workdir="/home/node/workspace",
|
|
)
|
|
self.assertEqual(
|
|
[
|
|
"container", "exec", "--interactive", "--tty",
|
|
"--workdir", "/home/node/workspace",
|
|
"bot-bottle-dev-abc", "claude",
|
|
],
|
|
bottle.agent_argv([]),
|
|
)
|
|
|
|
def test_exec_pipes_script_to_shell(self):
|
|
bottle = MacosContainerBottle("bot-bottle-dev-abc", lambda: None, None)
|
|
with patch("bot_bottle.backend.macos_container.bottle.subprocess.run") as run:
|
|
run.return_value.returncode = 7
|
|
run.return_value.stdout = "out"
|
|
run.return_value.stderr = "err"
|
|
result = bottle.exec("echo hi", user="root")
|
|
self.assertEqual(7, result.returncode)
|
|
self.assertEqual(
|
|
[
|
|
"container", "exec", "--user", "root", "--interactive",
|
|
"bot-bottle-dev-abc", "sh", "-s",
|
|
],
|
|
run.call_args.args[0],
|
|
)
|
|
self.assertEqual("echo hi", run.call_args.kwargs["input"])
|
|
|
|
def test_cp_in_uses_container_cp(self):
|
|
bottle = MacosContainerBottle("bot-bottle-dev-abc", lambda: None, None)
|
|
with patch("bot_bottle.backend.macos_container.bottle.subprocess.run") as run:
|
|
bottle.cp_in("/tmp/src", "/home/node/src")
|
|
self.assertEqual(
|
|
["container", "cp", "/tmp/src", "bot-bottle-dev-abc:/home/node/src"],
|
|
run.call_args.args[0],
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|