chore: import upstream snapshot with attribution

python/tests/cli/test_session_cli.py

@@ -0,0 +1,102 @@
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
+
+import re
+from contextlib import contextmanager
+
+from typer.testing import CliRunner
+
+from mirage.cli import session as session_cli
+
+_ANSI_RE = re.compile(r"\x1b\[[0-9;]*m")
+
+
+class _FakeResponse:
+    status_code = 201
+    content = b'{"session_id": "agent", "cwd": "/"}'
+
+    def json(self) -> dict:
+        return {"session_id": "agent", "cwd": "/"}
+
+
+class _FakeClient:
+
+    def __init__(self) -> None:
+        self.last_body: dict | None = None
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *_):
+        return False
+
+    def ensure_running(self, allow_spawn: bool = False) -> None:
+        return None
+
+    def request(self, method: str, path: str, json: dict | None = None):
+        self.last_body = json
+        return _FakeResponse()
+
+
+@contextmanager
+def _patched_client(fake: _FakeClient):
+    real = session_cli.make_client
+    session_cli.make_client = lambda: fake
+    try:
+        yield fake
+    finally:
+        session_cli.make_client = real
+
+
+def test_session_create_passes_mount_flags():
+    fake = _FakeClient()
+    with _patched_client(fake):
+        result = CliRunner().invoke(
+            session_cli.app,
+            [
+                "create", "demo", "--id", "agent", "-m", "/s3", "--mount",
+                "/slack"
+            ],
+        )
+    assert result.exit_code == 0, result.output
+    assert fake.last_body == {
+        "session_id": "agent",
+        "allowed_mounts": ["/s3", "/slack"],
+    }
+
+
+def test_session_create_no_mount_flag_omits_field():
+    fake = _FakeClient()
+    with _patched_client(fake):
+        result = CliRunner().invoke(
+            session_cli.app,
+            ["create", "demo", "--id", "agent"],
+        )
+    assert result.exit_code == 0, result.output
+    assert fake.last_body == {"session_id": "agent"}
+
+
+def test_session_create_without_id_or_mount_sends_empty_body():
+    fake = _FakeClient()
+    with _patched_client(fake):
+        result = CliRunner().invoke(session_cli.app, ["create", "demo"])
+    assert result.exit_code == 0, result.output
+    assert fake.last_body == {}
+
+
+def test_session_create_help_lists_mount_flag():
+    result = CliRunner().invoke(session_cli.app, ["create", "--help"])
+    assert result.exit_code == 0
+    plain = _ANSI_RE.sub("", result.output)
+    assert "--mount" in plain