Implement --save plus tests for --save and --dump, refs #272

tests/test_cli_memory.py

@@ -5,6 +5,7 @@ import pytest
 
 def test_memory_basic():
     result = CliRunner().invoke(cli.cli, ["memory", "select 1 + 1"])
+    assert result.exit_code == 0
     assert result.output.strip() == '[{"1 + 1": 2}]'
 
 
@@ -26,7 +27,44 @@ def test_memory_csv(tmpdir, sql_from, use_stdin):
         ["memory", csv_path, "select * from {}".format(sql_from), "--nl"],
         input=input,
     )
+    assert result.exit_code == 0
     assert (
         result.output.strip()
         == '{"id": "1", "name": "Cleo"}\n{"id": "2", "name": "Bants"}'
     )
+
+
+def test_memory_dump():
+    result = CliRunner().invoke(
+        cli.cli,
+        ["memory", "-", "select 1", "--dump"],
+        input="id,name\n1,Cleo\n2,Bants",
+    )
+    assert result.exit_code == 0
+    assert result.output.strip() == (
+        "BEGIN TRANSACTION;\n"
+        "CREATE TABLE [stdin] (\n"
+        "   [id] TEXT,\n"
+        "   [name] TEXT\n"
+        ");\n"
+        "INSERT INTO \"stdin\" VALUES('1','Cleo');\n"
+        "INSERT INTO \"stdin\" VALUES('2','Bants');\n"
+        "CREATE VIEW t1 AS select * from [stdin];\n"
+        "CREATE VIEW t AS select * from [stdin];\n"
+        "COMMIT;"
+    )
+
+
+def test_memory_save(tmpdir):
+    save_to = str(tmpdir / "save.db")
+    result = CliRunner().invoke(
+        cli.cli,
+        ["memory", "-", "select 1", "--save", save_to],
+        input="id,name\n1,Cleo\n2,Bants",
+    )
+    assert result.exit_code == 0
+    db = Database(save_to)
+    assert list(db["stdin"].rows) == [
+        {"id": "1", "name": "Cleo"},
+        {"id": "2", "name": "Bants"},
+    ]