wehub-resource-sync
c56bef871b
CodeQL / Analyze (python) (push) Has been cancelled
Update Platform Components Table / update (push) Has been cancelled
Docker image release / Build base image (push) Has been cancelled
Sync docs with Docusaurus / sync (push) Has been cancelled
Tests / Check if changed (push) Has been cancelled
Tests / format (push) Has been cancelled
Tests / check-imports (push) Has been cancelled
Tests / Unit / macos-latest (push) Has been cancelled
Tests / Unit / ubuntu-latest (push) Has been cancelled
Tests / Unit / windows-latest (push) Has been cancelled
Tests / mypy (push) Has been cancelled
Tests / Integration / ubuntu-latest (push) Has been cancelled
Tests / Integration / macos-latest (push) Has been cancelled
Tests / Integration / windows-latest (push) Has been cancelled
Tests / notify-slack-on-failure (push) Has been cancelled
Tests / Mark tests as completed (push) Has been cancelled
49 lines
2.1 KiB
Python
49 lines
2.1 KiB
Python
# 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)
|