Make join-condition suggestions work with multi-line queries

1 files changed, 3 insertions, 1 deletions

diff --git a/tests/test_smart_completion_public_schema_only.py b/tests/test_smart_completion_public_schema_only.py
index cacc6912..5bc745f2 100644
--- a/tests/test_smart_completion_public_schema_only.py
+++ b/tests/test_smart_completion_public_schema_only.py
@@ -532,13 +532,15 @@ def test_join_functions_using_suggests_common_columns(completer, complete_event)
          Completion(text='y', start_position=0, display_meta='column')])
 
 
-def test_join_functions_on_suggests_columns(completer, complete_event):
+def test_join_functions_on_suggests_columns_and_join_conditions(completer, complete_event):
     text = '''SELECT * FROM set_returning_func() f1
               INNER JOIN set_returning_func() f2 ON f1.'''
     pos = len(text)
     result = set(completer.get_completions(
         Document(text=text, cursor_position=pos), complete_event))
     assert set(result) == set([
+         Completion(text='y = f2.y', start_position=0, display_meta='name join'),
+         Completion(text='x = f2.x', start_position=0, display_meta='name join'),
          Completion(text='x', start_position=0, display_meta='column'),
          Completion(text='y', start_position=0, display_meta='column')])