Schema name should be escaped in SET SCHEMA.

1 files changed, 16 insertions, 0 deletions

diff --git a/tests/test_smart_completion_public_schema_only.py b/tests/test_smart_completion_public_schema_only.py
index d263aba2..4c43dd23 100644
--- a/tests/test_smart_completion_public_schema_only.py
+++ b/tests/test_smart_completion_public_schema_only.py
@@ -859,3 +859,19 @@ def test_keyword_after_alter(completer):
     expected = Completion('COLUMN', start_position=0, display_meta='keyword')
     completions = result_set(completer, text)
     assert expected in set(completions)
+
+
+@parametrize('completer', completers())
+def test_set_statement(completer):
+    text = ('SET ')
+    result = result_set(completer, text)
+    expected = set([schema(u"'public'")])
+    assert result == expected
+
+
+@parametrize('completer', completers())
+def test_set_schema(completer):
+    text = ('SET SCHEMA ')
+    result = result_set(completer, text)
+    expected = set([schema(u"'public'")])
+    assert result == expected