summaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/test_main.py2
-rw-r--r--tests/test_sqlcompletion.py2
2 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_main.py b/tests/test_main.py
index e8e801d5..2b5edb7a 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -317,5 +317,5 @@ def test_application_name_db_uri(tmpdir):
mock_pgexecute.return_value = None
cli = PGCli(pgclirc_file=str(tmpdir.join("rcfile")))
cli.connect_uri('postgres://bar@baz.com/?application_name=cow')
- mock_pgexecute.assert_called_with('bar', 'bar', '', 'baz.com', '', '',
+ mock_pgexecute.assert_called_with('bar', 'bar', None, 'baz.com', '', '',
application_name='cow')
diff --git a/tests/test_sqlcompletion.py b/tests/test_sqlcompletion.py
index 85fc3789..7e6a8225 100644
--- a/tests/test_sqlcompletion.py
+++ b/tests/test_sqlcompletion.py
@@ -236,7 +236,7 @@ def test_distinct_suggests_cols(text):
(
'SELECT * FROM tbl x JOIN tbl1 y ORDER BY ',
'SELECT * FROM tbl x JOIN tbl1 y ORDER BY ',
- 'BY',
+ 'ORDER BY',
)
])
def test_distinct_and_order_by_suggestions_with_aliases(text, text_before,