Add format test

Author: fangpenlin

tests/forms/test_validator.py

@@ -1,5 +1,3 @@
-import contextlib
-import os
 import pathlib
 
 import pytest
@@ -8,22 +6,13 @@
 from click.testing import CliRunner
 from pydantic import ValidationError
 
+from ..helper import switch_cwd
 from beanhub_cli.forms.validator import format_loc
 from beanhub_cli.forms.validator import merge_index_loc
 from beanhub_cli.forms.validator import validate_doc
 from beanhub_cli.main import cli
 
 
-@contextlib.contextmanager
-def switch_cwd(cwd: pathlib.Path):
-    current_cwd = pathlib.Path.cwd()
-    try:
-        os.chdir(cwd)
-        yield
-    finally:
-        os.chdir(current_cwd)
-
-
 @pytest.mark.parametrize(
     "loc, expected",
     [
@@ -118,9 +107,11 @@ def test_bad_schema(tmp_path: pathlib.Path, schema: dict, expected_errors: list)
         yaml.dump(schema, fo)
     with pytest.raises(ValidationError) as exc:
         validate_doc(doc_file)
+
     def del_url(d):
         del d["url"]
         return d
+
     assert list(map(del_url, exc.value.errors())) == expected_errors
 
 

tests/helper.py

@@ -0,0 +1,13 @@
+import contextlib
+import os
+import pathlib
+
+
+@contextlib.contextmanager
+def switch_cwd(cwd: pathlib.Path):
+    current_cwd = pathlib.Path.cwd()
+    try:
+        os.chdir(cwd)
+        yield
+    finally:
+        os.chdir(current_cwd)

tests/test_format.py

@@ -0,0 +1,27 @@
+import pathlib
+import textwrap
+
+from click.testing import CliRunner
+
+from .helper import switch_cwd
+from beanhub_cli.main import cli
+
+
+def test_format_cmd(tmp_path: pathlib.Path, cli_runner: CliRunner):
+    beanhub_dir = tmp_path / ".beanhub"
+    beanhub_dir.mkdir()
+
+    bean_file = beanhub_dir / "sample.bean"
+    bean_file.write_text(
+        textwrap.dedent(
+            """\
+    2024-06-27   open   Assets:Cash
+    """
+        )
+    )
+
+    cli_runner.mix_stderr = False
+    with switch_cwd(tmp_path):
+        result = cli_runner.invoke(cli, ["format", str(bean_file)])
+    assert result.exit_code == 0
+    assert bean_file.read_text() == "2024-06-27 open Assets:Cash\n"