Fix some failing tests.

dbcli · Jan 15, 2019 · 11a68b4 · 11a68b4
1 parent 16345d6
commit 11a68b4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -27,14 +27,14 @@ def test_execute_arg(executor):
     result = runner.invoke(cli, args=CLI_ARGS + ["-e", sql])
 
     assert result.exit_code == 0
-    assert "a\nabc\n" in result.output
+    assert '"abc"' in result.output
 
     result = runner.invoke(cli, args=CLI_ARGS + ["--execute", sql])
 
     assert result.exit_code == 0
-    assert "a\nabc\n" in result.output
+    assert '"abc"' in result.output
 
-    expected = "a\nabc\n"
+    expected = '"a"\n"abc"\n'
 
     assert expected in result.output
 
@@ -78,7 +78,7 @@ def test_batch_mode(executor):
     result = runner.invoke(cli, args=CLI_ARGS, input=sql)
 
     assert result.exit_code == 0
-    assert "count(*)\n3\na\nabc\n" in "".join(result.output)
+    assert '"count(*)"\n"3"\n"a"\n"abc"\n' in "".join(result.output)
 
 
 @dbtest