Watch out for leading whitespace

1 files changed, 6 insertions, 1 deletions

diff --git a/tests/test_pgspecial.py b/tests/test_pgspecial.py
index 39f9f979..66d9a5ad 100644
--- a/tests/test_pgspecial.py
+++ b/tests/test_pgspecial.py
@@ -18,7 +18,6 @@ def test_dn_suggests_schemata():
     suggestions = suggest_type('\\dn xxx', '\\dn xxx')
     assert suggestions == [{'type': 'schema'}]
 
-
 def test_d_suggests_tables_and_schemas():
     suggestions = suggest_type('\d ', '\d ')
     assert sorted_dicts(suggestions) == sorted_dicts([
@@ -42,3 +41,9 @@ def test_df_suggests_schema_or_function():
 
     suggestions = suggest_type('\\df myschema.xxx', '\\df myschema.xxx')
     assert suggestions == [{'type': 'function', 'schema': 'myschema'}]
+
+def test_leading_whitespace_ok():
+    cmd = '\\dn '
+    whitespace = '   '
+    suggestions = suggest_type(whitespace + cmd, whitespace + cmd)
+    assert suggestions == suggest_type(cmd, cmd)