chore: import upstream snapshot with attribution

python/tests/resource/redis/__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/resource/redis/test_redis_provider.py

@@ -0,0 +1,193 @@
+# ========= 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 os
+
+import pytest
+import pytest_asyncio
+
+from mirage.resource.redis import RedisResource
+from mirage.types import MountMode
+from mirage.workspace import Workspace
+
+REDIS_URL = os.environ.get("REDIS_URL", "")
+pytestmark = pytest.mark.skipif(not REDIS_URL, reason="REDIS_URL not set")
+
+
+@pytest_asyncio.fixture()
+async def ws():
+    resource = RedisResource(url=REDIS_URL, key_prefix="test:integ:")
+    await resource._store.clear()
+    await resource._store.add_dir("/")
+    w = Workspace(
+        {"/data": (resource, MountMode.WRITE)},
+        mode=MountMode.WRITE,
+    )
+    yield w
+    await resource._store.clear()
+    await resource._store.close()
+
+
+async def _run(ws, cmd, stdin=None):
+    io = await ws.execute(cmd, stdin=stdin)
+    stdout = io.stdout
+    if stdout is None:
+        return ""
+    if isinstance(stdout, bytes):
+        return stdout.decode(errors="replace")
+    chunks = [chunk async for chunk in stdout]
+    return b"".join(chunks).decode(errors="replace")
+
+
+@pytest.mark.asyncio
+async def test_echo_and_cat(ws):
+    await _run(ws, "echo hello > /data/hello.txt")
+    result = await _run(ws, "cat /data/hello.txt")
+    assert result == "hello\n"
+
+
+@pytest.mark.asyncio
+async def test_ls(ws):
+    await _run(ws, "echo a > /data/a.txt")
+    await _run(ws, "echo b > /data/b.txt")
+    result = await _run(ws, "ls /data/")
+    assert "a.txt" in result
+    assert "b.txt" in result
+
+
+@pytest.mark.asyncio
+async def test_stat(ws):
+    await _run(ws, "echo hello > /data/f.txt")
+    result = await _run(ws, "stat /data/f.txt")
+    assert "name=f.txt" in result
+    assert "size=" in result
+
+
+@pytest.mark.asyncio
+async def test_head(ws):
+    await _run(ws, "echo 'line1\nline2\nline3' > /data/f.txt")
+    result = await _run(ws, "head -n 2 /data/f.txt")
+    assert "line1" in result
+    assert "line2" in result
+
+
+@pytest.mark.asyncio
+async def test_grep(ws):
+    await _run(
+        ws,
+        "echo 'hello world\nfoo bar\nhello again' "
+        "> /data/f.txt",
+    )
+    result = await _run(ws, "grep hello /data/f.txt")
+    assert "hello world" in result
+    assert "hello again" in result
+    assert "foo bar" not in result
+
+
+@pytest.mark.asyncio
+async def test_wc(ws):
+    await _run(ws, "echo 'hello world' > /data/f.txt")
+    result = await _run(ws, "wc /data/f.txt")
+    assert "1" in result
+
+
+@pytest.mark.asyncio
+async def test_mkdir(ws):
+    await _run(ws, "mkdir /data/sub")
+    result = await _run(ws, "ls /data/")
+    assert "sub" in result
+
+
+@pytest.mark.asyncio
+async def test_mkdir_p(ws):
+    await _run(ws, "mkdir -p /data/a/b/c")
+    result = await _run(ws, "stat /data/a/b/c")
+    assert "directory" in result
+
+
+@pytest.mark.asyncio
+async def test_cp(ws):
+    await _run(ws, "echo data > /data/src.txt")
+    await _run(ws, "cp /data/src.txt /data/dst.txt")
+    result = await _run(ws, "cat /data/dst.txt")
+    assert "data" in result
+
+
+@pytest.mark.asyncio
+async def test_mv(ws):
+    await _run(ws, "echo data > /data/old.txt")
+    await _run(ws, "mv /data/old.txt /data/new.txt")
+    result = await _run(ws, "cat /data/new.txt")
+    assert "data" in result
+    ls_result = await _run(ws, "ls /data/")
+    assert "old.txt" not in ls_result
+
+
+@pytest.mark.asyncio
+async def test_rm(ws):
+    await _run(ws, "echo data > /data/f.txt")
+    await _run(ws, "rm /data/f.txt")
+    result = await _run(ws, "ls /data/")
+    assert "f.txt" not in result
+
+
+@pytest.mark.asyncio
+async def test_tree(ws):
+    await _run(ws, "echo a > /data/a.txt")
+    await _run(ws, "mkdir /data/sub")
+    await _run(ws, "echo b > /data/sub/b.txt")
+    result = await _run(ws, "tree /data/")
+    assert "a.txt" in result
+    assert "sub" in result
+    assert "b.txt" in result
+
+
+@pytest.mark.asyncio
+async def test_find(ws):
+    await _run(ws, "echo a > /data/a.txt")
+    await _run(ws, "mkdir /data/sub")
+    await _run(ws, "echo b > /data/sub/b.py")
+    result = await _run(ws, "find /data/ -name '*.py'")
+    assert "b.py" in result
+    assert "a.txt" not in result
+
+
+@pytest.mark.asyncio
+async def test_du(ws):
+    await _run(ws, "echo hello > /data/f.txt")
+    result = await _run(ws, "du /data/")
+    assert result.strip() != ""
+
+
+@pytest.mark.asyncio
+async def test_pipe(ws):
+    await _run(ws, "echo 'a\nb\na\nc\na' > /data/f.txt")
+    result = await _run(ws, "grep a /data/f.txt | wc -l")
+    assert result.strip() == "3"
+
+
+@pytest.mark.asyncio
+async def test_cd_and_pwd(ws):
+    await _run(ws, "echo hi > /data/f.txt")
+    await _run(ws, "cd /data/")
+    result = await _run(ws, "pwd")
+    assert "/data" in result
+
+
+@pytest.mark.asyncio
+async def test_data_persists_across_commands(ws):
+    await _run(ws, "echo persistent > /data/p.txt")
+    result1 = await _run(ws, "cat /data/p.txt")
+    result2 = await _run(ws, "cat /data/p.txt")
+    assert result1 == result2 == "persistent\n"

python/tests/resource/redis/test_redis_store.py

@@ -0,0 +1,138 @@
+# ========= 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 os
+
+import pytest
+import pytest_asyncio
+
+from mirage.resource.redis.store import RedisStore
+
+REDIS_URL = os.environ.get("REDIS_URL", "")
+pytestmark = pytest.mark.skipif(not REDIS_URL, reason="REDIS_URL not set")
+
+
+@pytest_asyncio.fixture()
+async def store():
+    s = RedisStore(url=REDIS_URL, key_prefix="test:store:")
+    await s.clear()
+    await s.add_dir("/")
+    yield s
+    await s.clear()
+    await s.close()
+
+
+@pytest.mark.asyncio
+async def test_get_set_file(store):
+    await store.set_file("/a.txt", b"hello")
+    assert await store.get_file("/a.txt") == b"hello"
+
+
+@pytest.mark.asyncio
+async def test_get_file_missing(store):
+    assert await store.get_file("/nope") is None
+
+
+@pytest.mark.asyncio
+async def test_del_file(store):
+    await store.set_file("/a.txt", b"data")
+    await store.del_file("/a.txt")
+    assert await store.get_file("/a.txt") is None
+
+
+@pytest.mark.asyncio
+async def test_has_file(store):
+    assert await store.has_file("/a.txt") is False
+    await store.set_file("/a.txt", b"x")
+    assert await store.has_file("/a.txt") is True
+
+
+@pytest.mark.asyncio
+async def test_list_files(store):
+    await store.set_file("/a.txt", b"a")
+    await store.set_file("/b.txt", b"b")
+    await store.set_file("/sub/c.txt", b"c")
+    files = await store.list_files()
+    assert "/a.txt" in files
+    assert "/b.txt" in files
+    assert "/sub/c.txt" in files
+
+
+@pytest.mark.asyncio
+async def test_list_files_prefix(store):
+    await store.set_file("/a.txt", b"a")
+    await store.set_file("/sub/b.txt", b"b")
+    files = await store.list_files("/sub/")
+    assert "/sub/b.txt" in files
+    assert "/a.txt" not in files
+
+
+@pytest.mark.asyncio
+async def test_file_len(store):
+    await store.set_file("/a.txt", b"hello")
+    assert await store.file_len("/a.txt") == 5
+
+
+@pytest.mark.asyncio
+async def test_has_dir(store):
+    assert await store.has_dir("/") is True
+    assert await store.has_dir("/sub") is False
+
+
+@pytest.mark.asyncio
+async def test_add_remove_dir(store):
+    await store.add_dir("/sub")
+    assert await store.has_dir("/sub") is True
+    await store.remove_dir("/sub")
+    assert await store.has_dir("/sub") is False
+
+
+@pytest.mark.asyncio
+async def test_list_dirs(store):
+    await store.add_dir("/a")
+    await store.add_dir("/b")
+    dirs = await store.list_dirs()
+    assert "/" in dirs
+    assert "/a" in dirs
+    assert "/b" in dirs
+
+
+@pytest.mark.asyncio
+async def test_get_set_modified(store):
+    await store.set_modified("/a.txt", "2026-01-01T00:00:00")
+    result = await store.get_modified("/a.txt")
+    assert result == "2026-01-01T00:00:00"
+
+
+@pytest.mark.asyncio
+async def test_get_modified_missing(store):
+    assert await store.get_modified("/nope") is None
+
+
+@pytest.mark.asyncio
+async def test_del_modified(store):
+    await store.set_modified("/a.txt", "2026-01-01")
+    await store.del_modified("/a.txt")
+    assert await store.get_modified("/a.txt") is None
+
+
+@pytest.mark.asyncio
+async def test_clear(store):
+    await store.set_file("/a.txt", b"data")
+    await store.add_dir("/sub")
+    await store.set_modified("/a.txt", "2026-01-01")
+    await store.clear()
+    assert await store.get_file("/a.txt") is None
+    assert await store.has_dir("/sub") is False
+    assert await store.get_modified("/a.txt") is None