chore: import upstream snapshot with attribution

python/tests/workspace/session/__init__.py

@@ -0,0 +1,13 @@
+# ========= 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. =========

python/tests/workspace/session/test_manager.py

@@ -0,0 +1,136 @@
+# ========= 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 asyncio
+
+import pytest
+
+from mirage.workspace.session import SessionManager
+
+
+def _run(coro):
+    return asyncio.run(coro)
+
+
+def test_manager_default_session():
+    mgr = SessionManager("default")
+    s = mgr.get("default")
+    assert s.session_id == "default"
+
+
+def test_manager_default_cwd():
+    mgr = SessionManager("default")
+    assert mgr.cwd == "/"
+    mgr.cwd = "/data"
+    assert mgr.cwd == "/data"
+    assert mgr.get("default").cwd == "/data"
+
+
+def test_manager_default_env():
+    mgr = SessionManager("default")
+    assert mgr.env == {}
+    mgr.env = {"A": "1"}
+    assert mgr.env == {"A": "1"}
+    assert mgr.get("default").env == {"A": "1"}
+
+
+def test_manager_create_session():
+    mgr = SessionManager("default")
+    s = mgr.create("worker-1")
+    assert s.session_id == "worker-1"
+    assert mgr.get("worker-1") is s
+
+
+def test_manager_create_duplicate_raises():
+    mgr = SessionManager("default")
+    mgr.create("s1")
+    with pytest.raises(ValueError, match="already exists"):
+        mgr.create("s1")
+
+
+def test_manager_get_missing_raises():
+    mgr = SessionManager("default")
+    with pytest.raises(KeyError):
+        mgr.get("nonexistent")
+
+
+def test_manager_list_sessions():
+    mgr = SessionManager("default")
+    mgr.create("s1")
+    mgr.create("s2")
+    sessions = mgr.list()
+    ids = {s.session_id for s in sessions}
+    assert ids == {"default", "s1", "s2"}
+
+
+def test_manager_close_session():
+    mgr = SessionManager("default")
+    mgr.create("temp")
+    _run(mgr.close("temp"))
+    with pytest.raises(KeyError):
+        mgr.get("temp")
+
+
+def test_manager_close_default_raises():
+    mgr = SessionManager("default")
+    with pytest.raises(ValueError, match="Cannot close"):
+        _run(mgr.close("default"))
+
+
+def test_manager_close_missing_raises():
+    mgr = SessionManager("default")
+    with pytest.raises(KeyError):
+        _run(mgr.close("nonexistent"))
+
+
+def test_manager_close_all():
+    mgr = SessionManager("default")
+    mgr.create("s1")
+    mgr.create("s2")
+    _run(mgr.close_all())
+    sessions = mgr.list()
+    assert len(sessions) == 1
+    assert sessions[0].session_id == "default"
+
+
+def test_manager_sessions_isolated():
+    mgr = SessionManager("default")
+    s1 = mgr.create("s1")
+    s2 = mgr.create("s2")
+    s1.env["X"] = "from-s1"
+    s1.cwd = "/s1"
+    assert "X" not in s2.env
+    assert s2.cwd == "/"
+
+
+def test_manager_lock_for():
+    mgr = SessionManager("default")
+    lock = mgr.lock_for("default")
+    assert lock is not None
+
+    mgr.create("s1")
+    lock2 = mgr.lock_for("s1")
+    assert lock2 is not lock
+
+
+def test_manager_create_with_allowed_mounts():
+    mgr = SessionManager("default")
+    s = mgr.create("agent", allowed_mounts=frozenset({"/s3", "/slack"}))
+    assert s.allowed_mounts == frozenset({"/s3", "/slack"})
+
+
+def test_manager_create_default_unrestricted():
+    mgr = SessionManager("default")
+    s = mgr.create("worker")
+    assert s.allowed_mounts is None

python/tests/workspace/session/test_session.py

@@ -0,0 +1,157 @@
+# ========= 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. =========
+
+from mirage.workspace.session import Session
+
+
+def test_session_defaults():
+    s = Session(session_id="test")
+    assert s.session_id == "test"
+    assert s.cwd == "/"
+    assert s.env == {}
+    assert s.functions == {}
+    assert s.last_exit_code == 0
+    assert s._stdin_buffer is None
+
+
+def test_session_custom_cwd():
+    s = Session(session_id="s1", cwd="/data")
+    assert s.cwd == "/data"
+
+
+def test_session_env():
+    s = Session(session_id="s1", env={"A": "1", "B": "2"})
+    assert s.env["A"] == "1"
+    assert s.env["B"] == "2"
+
+
+def test_session_env_mutation():
+    s = Session(session_id="s1")
+    s.env["X"] = "hello"
+    assert s.env["X"] == "hello"
+    del s.env["X"]
+    assert "X" not in s.env
+
+
+def test_session_functions():
+    s = Session(session_id="s1")
+    s.functions["myfunc"] = "body"
+    assert "myfunc" in s.functions
+
+
+def test_session_exit_code():
+    s = Session(session_id="s1")
+    s.last_exit_code = 42
+    assert s.last_exit_code == 42
+
+
+def test_session_stdin_buffer():
+    s = Session(session_id="s1")
+    s._stdin_buffer = b"hello\n"
+    assert s._stdin_buffer == b"hello\n"
+    s._stdin_buffer = None
+    assert s._stdin_buffer is None
+
+
+def test_session_to_dict():
+    s = Session(session_id="s1", cwd="/data", env={"K": "V"})
+    d = s.to_dict()
+    assert d["session_id"] == "s1"
+    assert d["cwd"] == "/data"
+    assert d["env"] == {"K": "V"}
+    assert "created_at" in d
+
+
+def test_session_from_dict():
+    d = {
+        "session_id": "s2",
+        "cwd": "/tmp",
+        "env": {
+            "A": "1"
+        },
+        "created_at": 123.0
+    }
+    s = Session.from_dict(d)
+    assert s.session_id == "s2"
+    assert s.cwd == "/tmp"
+    assert s.env["A"] == "1"
+    assert s.created_at == 123.0
+
+
+def test_session_roundtrip():
+    original = Session(session_id="rt", cwd="/x", env={"K": "V"})
+    restored = Session.from_dict(original.to_dict())
+    assert restored.session_id == original.session_id
+    assert restored.cwd == original.cwd
+    assert restored.env == original.env
+
+
+def test_session_independent_envs():
+    s1 = Session(session_id="a")
+    s2 = Session(session_id="b")
+    s1.env["X"] = "1"
+    assert "X" not in s2.env
+
+
+def test_session_allowed_mounts_default_none():
+    s = Session(session_id="s")
+    assert s.allowed_mounts is None
+
+
+def test_session_allowed_mounts_set():
+    s = Session(session_id="s", allowed_mounts=frozenset({"/s3", "/slack"}))
+    assert s.allowed_mounts == frozenset({"/s3", "/slack"})
+
+
+def test_fork_copies_every_field_including_allowed_mounts():
+    original = Session(
+        session_id="orig",
+        cwd="/disk",
+        env={"FOO": "bar"},
+        functions={"f": object()},
+        last_exit_code=7,
+        shell_options={"errexit": True},
+        readonly_vars={"HOME"},
+        arrays={"ARGV": ["a", "b"]},
+        allowed_mounts=frozenset({"/s3", "/dev", "/"}),
+    )
+    forked = original.fork()
+    assert forked.session_id == "orig"
+    assert forked.cwd == "/disk"
+    assert forked.env == {"FOO": "bar"}
+    assert forked.allowed_mounts == frozenset({"/s3", "/dev", "/"})
+    assert forked.shell_options == {"errexit": True}
+    assert "HOME" in forked.readonly_vars
+    assert forked.arrays == {"ARGV": ["a", "b"]}
+    assert forked.last_exit_code == 7
+
+
+def test_fork_overrides_apply_without_mutating_original():
+    original = Session(session_id="orig", cwd="/disk", env={"FOO": "bar"})
+    forked = original.fork(cwd="/ram", env={"BAZ": "qux"})
+    assert forked.cwd == "/ram"
+    assert forked.env == {"BAZ": "qux"}
+    assert original.cwd == "/disk"
+    assert original.env == {"FOO": "bar"}
+
+
+def test_fork_deep_copies_mutable_containers():
+    original = Session(session_id="orig",
+                       env={"FOO": "bar"},
+                       arrays={"A": ["1"]})
+    forked = original.fork()
+    forked.env["NEW"] = "leaked?"
+    forked.arrays["A"].append("2")
+    assert "NEW" not in original.env
+    assert original.arrays["A"] == ["1"]

python/tests/workspace/session/test_shell_dirs.py

@@ -0,0 +1,46 @@
+# ========= 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. =========
+
+from mirage.workspace.session.session import Session
+from mirage.workspace.session.shell_dirs import change_dir, home_dir
+
+
+def test_home_dir_unset_is_none():
+    session = Session(session_id="s")
+    assert home_dir(session) is None
+
+
+def test_home_dir_from_env():
+    session = Session(session_id="s", env={"HOME": "/data"})
+    assert home_dir(session) == "/data"
+
+
+def test_home_dir_empty_env_is_none():
+    session = Session(session_id="s", env={"HOME": ""})
+    assert home_dir(session) is None
+
+
+def test_change_dir_sets_cwd_and_oldpwd():
+    session = Session(session_id="s", cwd="/data")
+    change_dir(session, "/data/sub")
+    assert session.cwd == "/data/sub"
+    assert session.env["OLDPWD"] == "/data"
+
+
+def test_change_dir_overwrites_oldpwd():
+    session = Session(session_id="s", cwd="/a")
+    change_dir(session, "/b")
+    change_dir(session, "/c")
+    assert session.cwd == "/c"
+    assert session.env["OLDPWD"] == "/b"