sqlite-utils insert-files command, closes #122

tests/test_docs.py

@@ -18,5 +18,5 @@ def documented_commands():
 
 
 @pytest.mark.parametrize("command", cli.cli.commands.keys())
-def test_plugin_hooks_are_documented(documented_commands, command):
+def test_commands_are_documented(documented_commands, command):
     assert command in documented_commands

tests/test_insert_files.py

@@ -0,0 +1,85 @@
+from sqlite_utils import cli, Database
+from click.testing import CliRunner
+import pathlib
+
+
+def test_insert_files():
+    runner = CliRunner()
+    with runner.isolated_filesystem():
+        tmpdir = pathlib.Path(".")
+        print("tmpdir = ", tmpdir.resolve())
+        db_path = str(tmpdir / "files.db")
+        (tmpdir / "one.txt").write_text("This is file one", "utf-8")
+        (tmpdir / "two.txt").write_text("Two is shorter", "utf-8")
+        (tmpdir / "nested").mkdir()
+        (tmpdir / "nested" / "three.txt").write_text("Three is nested", "utf-8")
+        coltypes = (
+            "name",
+            "path",
+            "fullpath",
+            "sha256",
+            "md5",
+            "mode",
+            "content",
+            "mtime",
+            "ctime",
+            "mtime_int",
+            "ctime_int",
+            "mtime_iso",
+            "ctime_iso",
+            "size",
+        )
+        cols = []
+        for coltype in coltypes:
+            cols += ["-c", "{}:{}".format(coltype, coltype)]
+        result = runner.invoke(
+            cli.cli,
+            ["insert-files", db_path, "files", str(tmpdir)] + cols + ["--pk", "path"],
+            catch_exceptions=False,
+        )
+        assert result.exit_code == 0, result.stdout
+        db = Database(db_path)
+        rows_by_path = {r["path"]: r for r in db["files"].rows}
+        one, two, three = (
+            rows_by_path["one.txt"],
+            rows_by_path["two.txt"],
+            rows_by_path["nested/three.txt"],
+        )
+        assert {
+            "content": b"This is file one",
+            "md5": "556dfb57fce9ca301f914e2273adf354",
+            "name": "one.txt",
+            "path": "one.txt",
+            "sha256": "e34138f26b5f7368f298b4e736fea0aad87ddec69fbd04dc183b20f4d844bad5",
+            "size": 16,
+        }.items() <= one.items()
+        assert {
+            "content": b"Two is shorter",
+            "md5": "f86f067b083af1911043eb215e74ac70",
+            "name": "two.txt",
+            "path": "two.txt",
+            "sha256": "9368988ed16d4a2da0af9db9b686d385b942cb3ffd4e013f43aed2ec041183d9",
+            "size": 14,
+        }.items() <= two.items()
+        assert {
+            "content": b"Three is nested",
+            "md5": "12580f341781f5a5b589164d3cd39523",
+            "name": "three.txt",
+            "path": "nested/three.txt",
+            "sha256": "6dd45aaaaa6b9f96af19363a92c8fca5d34791d3c35c44eb19468a6a862cc8cd",
+            "size": 15,
+        }.items() <= three.items()
+        # Assert the other int/str/float columns exist and are of the right types
+        expected_types = {
+            "ctime": float,
+            "ctime_int": int,
+            "ctime_iso": str,
+            "mtime": float,
+            "mtime_int": int,
+            "mtime_iso": str,
+            "mode": int,
+            "fullpath": str,
+        }
+        for colname, expected_type in expected_types.items():
+            for row in (one, two, three):
+                assert isinstance(row[colname], expected_type)