chore: import upstream snapshot with attribution

python/tests/workspace/test_partial_read_errors.py

@@ -0,0 +1,362 @@
+# ========= 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
+
+from mirage.resource.ram import RAMResource
+from mirage.types import MountMode
+from mirage.workspace import Workspace
+
+
+def _make_ws():
+    ram1 = RAMResource()
+    ram2 = RAMResource()
+    ram1._store.files["/f.txt"] = b"aaa\n"
+    return Workspace(
+        {
+            "/a/": (ram1, MountMode.WRITE),
+            "/b/": (ram2, MountMode.WRITE)
+        }, )
+
+
+def _make_numbered_ws():
+    ram = RAMResource()
+    ram._store.files["/f.txt"] = b"1\n2\n"
+    ram._store.files["/g.txt"] = b"3\n4\n"
+    ram._store.files["/h.txt"] = b"hello\n"
+    return Workspace({"/a/": (ram, MountMode.WRITE)}, )
+
+
+def _run(ws, cmd):
+
+    async def _inner():
+        io = await ws.execute(cmd)
+        return await io.stdout_str(), await io.stderr_str(), io.exit_code
+
+    return asyncio.run(_inner())
+
+
+# ── single-mount: good + missing keeps partial output, GNU-style ──
+
+
+def test_cat_good_then_missing():
+    out, err, code = _run(_make_ws(), "cat /a/f.txt /a/missing.txt")
+    assert out == "aaa\n"
+    assert err == "cat: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_cat_missing_then_good():
+    out, err, code = _run(_make_ws(), "cat /a/missing.txt /a/f.txt")
+    assert out == "aaa\n"
+    assert err == "cat: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_cat_all_missing_reports_each():
+    out, err, code = _run(_make_ws(), "cat /a/m1.txt /a/m2.txt")
+    assert out == ""
+    assert err == ("cat: /a/m1.txt: No such file or directory\n"
+                   "cat: /a/m2.txt: No such file or directory\n")
+    assert code == 1
+
+
+def test_wc_good_then_missing_keeps_total():
+    out, err, code = _run(_make_ws(), "wc -l /a/f.txt /a/missing.txt")
+    assert out == "1 /a/f.txt\n1 total\n"
+    assert err == "wc: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_wc_all_missing_zero_total():
+    out, err, code = _run(_make_ws(), "wc -l /a/m1.txt /a/m2.txt")
+    assert out == "0 total\n"
+    assert err == ("wc: /a/m1.txt: No such file or directory\n"
+                   "wc: /a/m2.txt: No such file or directory\n")
+    assert code == 1
+
+
+def test_head_good_then_missing_keeps_banner():
+    out, err, code = _run(_make_ws(), "head -n 1 /a/f.txt /a/missing.txt")
+    assert out == "==> /a/f.txt <==\naaa\n"
+    assert err == "head: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_head_missing_first_no_leading_blank():
+    out, err, code = _run(_make_ws(), "head -n 1 /a/missing.txt /a/f.txt")
+    assert out == "==> /a/f.txt <==\naaa\n"
+    assert err == "head: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_tail_good_then_missing_keeps_banner():
+    out, err, code = _run(_make_ws(), "tail -n 1 /a/f.txt /a/missing.txt")
+    assert out == "==> /a/f.txt <==\naaa\n"
+    assert err == "tail: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_single_missing_operand_unchanged():
+    out, err, code = _run(_make_ws(), "cat /a/missing.txt")
+    assert out == ""
+    assert err == "cat: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+# ── cross-mount: same bytes as single-mount ──
+
+
+def test_cross_cat_good_then_missing():
+    out, err, code = _run(_make_ws(), "cat /a/f.txt /b/missing.txt")
+    assert out == "aaa\n"
+    assert err == "cat: /b/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_cross_wc_good_then_missing_keeps_total():
+    out, err, code = _run(_make_ws(), "wc -l /a/f.txt /b/missing.txt")
+    assert out == "1 /a/f.txt\n1 total\n"
+    assert err == "wc: /b/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_cross_head_good_then_missing_keeps_banner():
+    out, err, code = _run(_make_ws(), "head -n 1 /a/f.txt /b/missing.txt")
+    assert out == "==> /a/f.txt <==\naaa\n"
+    assert err == "head: /b/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_cross_tail_good_then_missing_keeps_banner():
+    out, err, code = _run(_make_ws(), "tail -n 1 /a/f.txt /b/missing.txt")
+    assert out == "==> /a/f.txt <==\naaa\n"
+    assert err == "tail: /b/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+# ── every operand is processed, not just the first (paths[0] bugs) ──
+
+
+def test_cut_processes_all_operands():
+    out, err, code = _run(_make_numbered_ws(), "cut -c1 /a/f.txt /a/g.txt")
+    assert out == "1\n2\n3\n4\n"
+    assert err == ""
+    assert code == 0
+
+
+def test_tac_reverses_each_operand():
+    out, err, code = _run(_make_numbered_ws(), "tac /a/f.txt /a/g.txt")
+    assert out == "2\n1\n4\n3\n"
+    assert err == ""
+    assert code == 0
+
+
+def test_nl_numbering_continues_across_operands():
+    out, err, code = _run(_make_numbered_ws(), "nl /a/f.txt /a/g.txt")
+    assert out == "     1\t1\n     2\t2\n     3\t3\n     4\t4\n"
+    assert err == ""
+    assert code == 0
+
+
+def test_strings_scans_all_operands():
+    ws = _make_numbered_ws()
+    _run(ws, "printf 'worlds\\n' > /a/h2.txt")
+    out, err, code = _run(ws, "strings /a/h.txt /a/h2.txt")
+    assert out == "hello\nworlds\n"
+    assert err == ""
+    assert code == 0
+
+
+def test_zcat_concatenates_all_operands():
+    ws = _make_numbered_ws()
+    _run(
+        ws, "printf 'z\\n' > /a/z1.txt && printf 'y\\n' > /a/z2.txt"
+        " && gzip /a/z1.txt /a/z2.txt")
+    out, err, code = _run(ws, "zcat /a/z1.txt.gz /a/z2.txt.gz")
+    assert out == "z\ny\n"
+    assert err == ""
+    assert code == 0
+
+
+# ── the rest of the read family keeps partial output past missing ──
+
+
+def test_nl_good_then_missing():
+    out, err, code = _run(_make_numbered_ws(), "nl /a/f.txt /a/missing.txt")
+    assert out == "     1\t1\n     2\t2\n"
+    assert err == "nl: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_nl_all_missing_reports_each():
+    out, err, code = _run(_make_numbered_ws(), "nl /a/m1.txt /a/m2.txt")
+    assert out == ""
+    assert err == ("nl: /a/m1.txt: No such file or directory\n"
+                   "nl: /a/m2.txt: No such file or directory\n")
+    assert code == 1
+
+
+def test_md5_good_then_missing():
+    out, err, code = _run(_make_numbered_ws(), "md5 /a/f.txt /a/missing.txt")
+    assert out == "6ddb4095eb719e2a9f0a3f95677d24e0  /a/f.txt\n"
+    assert err == "md5: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_sha256sum_good_then_missing():
+    out, err, code = _run(_make_numbered_ws(),
+                          "sha256sum /a/f.txt /a/missing.txt")
+    assert out == ("a6e2b7a040683432de03a18fd8a1939a2fdf8258"
+                   "5b364bfc874bdd4095c4cae1  /a/f.txt\n")
+    assert err == "sha256sum: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_tac_good_then_missing():
+    out, err, code = _run(_make_numbered_ws(), "tac /a/f.txt /a/missing.txt")
+    assert out == "2\n1\n"
+    assert err == "tac: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_rev_good_then_missing():
+    out, err, code = _run(_make_numbered_ws(), "rev /a/f.txt /a/missing.txt")
+    assert out == "1\n2\n"
+    assert err == "rev: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_cut_good_then_missing():
+    out, err, code = _run(_make_numbered_ws(),
+                          "cut -c1 /a/f.txt /a/missing.txt")
+    assert out == "1\n2\n"
+    assert err == "cut: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_expand_good_then_missing():
+    out, err, code = _run(_make_numbered_ws(),
+                          "expand /a/f.txt /a/missing.txt")
+    assert out == "1\n2\n"
+    assert err == "expand: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_unexpand_good_then_missing():
+    out, err, code = _run(_make_numbered_ws(),
+                          "unexpand /a/f.txt /a/missing.txt")
+    assert out == "1\n2\n"
+    assert err == "unexpand: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_fold_good_then_missing():
+    out, err, code = _run(_make_numbered_ws(), "fold /a/f.txt /a/missing.txt")
+    assert out == "1\n2\n"
+    assert err == "fold: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_fmt_good_then_missing():
+    out, err, code = _run(_make_numbered_ws(), "fmt /a/f.txt /a/missing.txt")
+    assert out == "1 2\n"
+    assert err == "fmt: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_strings_good_then_missing():
+    out, err, code = _run(_make_numbered_ws(),
+                          "strings /a/h.txt /a/missing.txt")
+    assert out == "hello\n"
+    assert err == "strings: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_zcat_good_then_missing():
+    ws = _make_numbered_ws()
+    _run(ws, "printf 'z\\n' > /a/z1.txt && gzip /a/z1.txt")
+    out, err, code = _run(ws, "zcat /a/z1.txt.gz /a/missing.gz")
+    assert out == "z\n"
+    assert err == "zcat: /a/missing.gz: No such file or directory\n"
+    assert code == 1
+
+
+def test_sort_still_aborts_on_missing():
+    # GNU sort needs all input before emitting anything, so no partial
+    # output; the repo reports the operand and exits 1 (GNU exits 2).
+    out, err, code = _run(_make_numbered_ws(), "sort /a/f.txt /a/missing.txt")
+    assert out == ""
+    assert err == "sort: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def _make_cross_numbered_ws():
+    ram1 = RAMResource()
+    ram2 = RAMResource()
+    ram1._store.files["/f.txt"] = b"1\n2\n"
+    return Workspace(
+        {
+            "/a/": (ram1, MountMode.WRITE),
+            "/b/": (ram2, MountMode.WRITE)
+        }, )
+
+
+def test_cross_nl_reports_own_name():
+    # STREAM commands fetch operand bytes through a native cat sub-run; the
+    # error line must still carry the command's own name, like single-mount.
+    out, err, code = _run(_make_cross_numbered_ws(),
+                          "nl /a/f.txt /b/missing.txt")
+    assert out == "     1\t1\n     2\t2\n"
+    assert err == "nl: /b/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_cross_md5_good_then_missing():
+    out, err, code = _run(_make_cross_numbered_ws(),
+                          "md5 /a/f.txt /b/missing.txt")
+    assert out == "6ddb4095eb719e2a9f0a3f95677d24e0  /a/f.txt\n"
+    assert err == "md5: /b/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_stat_good_then_missing_keeps_row():
+    out, err, code = _run(_make_ws(), "stat /a/f.txt /a/missing.txt")
+    assert "name=f.txt" in out
+    assert err == "stat: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_sed_good_then_missing_keeps_output():
+    out, err, code = _run(_make_numbered_ws(),
+                          "sed s/1/X/ /a/f.txt /a/missing.txt")
+    assert out == "X\n2\n"
+    assert err == "sed: /a/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_cross_sed_good_then_missing_keeps_output():
+    out, err, code = _run(_make_cross_numbered_ws(),
+                          "sed s/1/X/ /a/f.txt /b/missing.txt")
+    assert out == "X\n2\n"
+    assert err == "sed: /b/missing.txt: No such file or directory\n"
+    assert code == 1
+
+
+def test_cross_sort_aborts_like_single_mount():
+    out, err, code = _run(_make_cross_numbered_ws(),
+                          "sort /a/f.txt /b/missing.txt")
+    assert out == ""
+    assert err == "sort: /b/missing.txt: No such file or directory\n"
+    assert code == 1