Make join-condition suggestions work with multi-line queries

1 files changed, 8 insertions, 0 deletions

diff --git a/tests/test_sqlcompletion.py b/tests/test_sqlcompletion.py
index e380af6f..b64ce41b 100644
--- a/tests/test_sqlcompletion.py
+++ b/tests/test_sqlcompletion.py
@@ -402,6 +402,14 @@ def test_join_alias_dot_suggests_cols2(sql):
 
 @pytest.mark.parametrize('sql', [
     'select a.x, b.y from abc a join bcd b on ',
+    '''select a.x, b.y
+from abc a
+join bcd b on
+''',
+    '''select a.x, b.y
+from abc a
+join bcd b
+on ''',
     'select a.x, b.y from abc a join bcd b on a.id = b.id OR ',
 ])
 def test_on_suggests_aliases_and_join_conditions(sql):