tests: add drafter tests

Author: mtth

.gitignore

@@ -1,3 +1,4 @@
 dist/
 docs/_*
+htmlcov/
 .coverage

src/git_draft/drafter.py

@@ -63,7 +63,7 @@ def __init__(self, repo: git.Repo, hook: OperationHook | None) -> None:
     @override
     def _list(self) -> Sequence[PurePosixPath]:
         # Show staged files.
-        return self._repo.git.ls_files()
+        return self._repo.git.ls_files().splitlines()
 
     @override
     def _read(self, path: PurePosixPath) -> str:

tests/git_draft/drafter_test.py

@@ -1,13 +1,49 @@
 import git
 import os.path as osp
-from pathlib import PurePosixPath
+from pathlib import Path, PurePosixPath
 import pytest
 
 from git_draft.bots import Action, Bot, Toolbox
 import git_draft.drafter as sut
 from git_draft.store import Store
 
 
+class TestToolbox:
+    @pytest.fixture(autouse=True)
+    def setup(self, repo: git.Repo) -> None:
+        self._toolbox = sut._Toolbox(repo, None)
+
+    def test_list_files(self, repo: git.Repo) -> None:
+        assert self._toolbox.list_files() == []
+        names = set(["one.txt", "two.txt"])
+        for name in names:
+            with open(Path(repo.working_dir, name), "w") as f:
+                f.write("ok")
+        repo.git.add(all=True)
+        assert set(self._toolbox.list_files()) == names
+
+    def test_read_file(self, repo: git.Repo) -> None:
+        with open(Path(repo.working_dir, "one"), "w") as f:
+            f.write("ok")
+
+        path = PurePosixPath("one")
+        with pytest.raises(git.GitCommandError):
+            assert self._toolbox.read_file(path) == ""
+
+        repo.git.add(all=True)
+        assert self._toolbox.read_file(path) == "ok"
+
+    def test_write_file(self, repo: git.Repo) -> None:
+        self._toolbox.write_file(PurePosixPath("one"), "hi")
+
+        path = Path(repo.working_dir, "one")
+        assert not path.exists()
+
+        repo.git.checkout_index(all=True)
+        with open(path) as f:
+            assert f.read() == "hi"
+
+
 class _FakeBot(Bot):
     def act(self, prompt: str, toolbox: Toolbox) -> Action:
         toolbox.write_file(PurePosixPath("PROMPT"), prompt)