Fix \d special command and add some tests

1 files changed, 19 insertions, 0 deletions

diff --git a/tests/test_pgspecial.py b/tests/test_pgspecial.py
index e69de29b..f02efac9 100644
--- a/tests/test_pgspecial.py
+++ b/tests/test_pgspecial.py
@@ -0,0 +1,19 @@
+from pgcli.packages.sqlcompletion import suggest_type
+from test_sqlcompletion import sorted_dicts
+
+def test_d_suggests_tables_and_schemas():
+    suggestions = suggest_type('\d ', '\d ')
+    assert sorted_dicts(suggestions) == sorted_dicts([
+            {'type': 'schema'}, {'type': 'table', 'schema': []}])
+
+    suggestions = suggest_type('\d xxx', '\d xxx')
+    assert sorted_dicts(suggestions) == sorted_dicts([
+            {'type': 'schema'}, {'type': 'table', 'schema': []}])
+
+def test_d_dot_suggests_schema_qualified_tables():
+    suggestions = suggest_type('\d myschema.', '\d myschema.')
+    assert suggestions == [{'type': 'table', 'schema': 'myschema'}]
+
+    suggestions = suggest_type('\d myschema.xxx', '\d myschema.xxx')
+    assert suggestions == [{'type': 'table', 'schema': 'myschema'}]
+