summaryrefslogtreecommitdiffstats
path: root/tests/test_sqlcompletion.py
diff options
context:
space:
mode:
authorJoakim Koljonen <koljonen@outlook.com>2016-08-23 20:14:00 +0200
committerJoakim Koljonen <koljonen@outlook.com>2016-08-23 20:14:00 +0200
commit8d2265a65d85fde0e62b0002b937d6564a85142b (patch)
treea6f6d4c75f3f2ea19bdaf786207d8ce754f26926 /tests/test_sqlcompletion.py
parentbeb9ceae946c3db9cff6d21d0f35485e853411d6 (diff)
Temporary hack for sqlparse crashing after AS
And a regression test.
Diffstat (limited to 'tests/test_sqlcompletion.py')
-rw-r--r--tests/test_sqlcompletion.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/tests/test_sqlcompletion.py b/tests/test_sqlcompletion.py
index 4063ea7f..d9fffef5 100644
--- a/tests/test_sqlcompletion.py
+++ b/tests/test_sqlcompletion.py
@@ -55,6 +55,14 @@ def test_where_in_suggests_columns(expression):
suggestions = suggest_type(expression, expression)
assert set(suggestions) == cols_etc('tabl')
+@pytest.mark.parametrize('expression', [
+ 'SELECT 1 AS ',
+ 'SELECT 1 FROM tabl AS ',
+])
+def test_after_as(expression):
+ suggestions = suggest_type(expression, expression)
+ assert set(suggestions) == set()
+
def test_where_equals_any_suggests_columns_or_keywords():
text = 'SELECT * FROM tabl WHERE foo = ANY('