Add tests for \c and \connect to suggest databases

1 files changed, 8 insertions, 1 deletions

diff --git a/tests/test_pgspecial.py b/tests/test_pgspecial.py
index f08fd304..fc41ad58 100644
--- a/tests/test_pgspecial.py
+++ b/tests/test_pgspecial.py
@@ -1,3 +1,4 @@
+import pytest
 from pgcli.packages.sqlcompletion import suggest_type
 from test_sqlcompletion import sorted_dicts
 
@@ -67,4 +68,10 @@ def test_schema_qualified_dT_suggests_datatypes():
     text = '\\dT foo.'
     suggestions = suggest_type(text, text)
     assert sorted_dicts(suggestions) == sorted_dicts(
-        [{'type': 'datatype', 'schema': 'foo'}])
\ No newline at end of file
+        [{'type': 'datatype', 'schema': 'foo'}])
+
+
+@pytest.mark.parametrize('command', ['\\c ', '\\connect '])
+def test_c_suggests_databases(command):
+    suggestions = suggest_type(command, command)
+    assert suggestions == [{'type': 'database'}]
\ No newline at end of file