chore: import upstream snapshot with attribution

test/hooks/tool_result_offloading/test_stores.py

@@ -0,0 +1,48 @@
+# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from pathlib import Path
+
+import pytest
+
+from haystack.hooks.tool_result_offloading import FileSystemToolResultStore
+
+
+class TestFileSystemToolResultStore:
+    def test_write_returns_path_and_persists_content(self, tmp_path):
+        store = FileSystemToolResultStore(root=tmp_path)
+        reference = store.write(key="a.txt", content="hello")
+        assert reference == str(tmp_path / "a.txt")
+        assert Path(reference).read_text(encoding="utf-8") == "hello"
+
+    def test_write_creates_missing_directories(self, tmp_path):
+        store = FileSystemToolResultStore(root=tmp_path / "nested" / "dir")
+        reference = store.write(key="a.txt", content="hi")
+        assert Path(reference).read_text(encoding="utf-8") == "hi"
+
+    def test_write_allows_nested_keys_within_root(self, tmp_path):
+        store = FileSystemToolResultStore(root=tmp_path)
+        reference = store.write(key="sub/dir/a.txt", content="ok")
+        assert Path(reference).read_text(encoding="utf-8") == "ok"
+
+    def test_write_rejects_parent_traversal_key(self, tmp_path):
+        store = FileSystemToolResultStore(root=tmp_path / "root")
+        with pytest.raises(ValueError, match="outside the store root"):
+            store.write(key="../escape.txt", content="x")
+        assert not (tmp_path / "escape.txt").exists()
+
+    def test_write_rejects_absolute_key(self, tmp_path):
+        store = FileSystemToolResultStore(root=tmp_path / "root")
+        with pytest.raises(ValueError, match="outside the store root"):
+            store.write(key=str(tmp_path / "outside.txt"), content="x")
+
+    def test_read_round_trips_written_content(self, tmp_path):
+        store = FileSystemToolResultStore(root=tmp_path)
+        reference = store.write(key="a.txt", content="round trip")
+        assert store.read(reference) == "round trip"
+
+    def test_to_dict_from_dict_roundtrip(self, tmp_path):
+        store = FileSystemToolResultStore(root=tmp_path)
+        restored = FileSystemToolResultStore.from_dict(store.to_dict())
+        assert restored.root == Path(tmp_path)