1 files changed, 20 insertions, 0 deletions
diff --git a/tests/test_smart_completion.py b/tests/test_smart_completion.py
index b306b36a..425344b9 100644
--- a/tests/test_smart_completion.py
+++ b/tests/test_smart_completion.py
@@ -197,6 +197,16 @@ def test_suggested_aliases_after_on(completer, complete_event):
         Completion(text='u', start_position=0),
         Completion(text='o', start_position=0)])
 
+def test_suggested_aliases_after_on_right_side(completer, complete_event):
+    text = 'SELECT u.name, o.id FROM users u JOIN orders o ON o.user_id = '
+    position = len('SELECT u.name, o.id FROM users u JOIN orders o ON o.user_id = ')
+    result = set(completer.get_completions(
+        Document(text=text, cursor_position=position),
+        complete_event))
+    assert set(result) == set([
+        Completion(text='u', start_position=0),
+        Completion(text='o', start_position=0)])
+
 def test_suggested_tables_after_on(completer, complete_event):
     text = 'SELECT users.name, orders.id FROM users JOIN orders ON '
     position = len('SELECT users.name, orders.id FROM users JOIN orders ON ')
@@ -207,6 +217,16 @@ def test_suggested_tables_after_on(completer, complete_event):
         Completion(text='users', start_position=0),
         Completion(text='orders', start_position=0)])
 
+def test_suggested_tables_after_on_right_side(completer, complete_event):
+    text = 'SELECT users.name, orders.id FROM users JOIN orders ON orders.user_id = '
+    position = len('SELECT users.name, orders.id FROM users JOIN orders ON orders.user_id = ')
+    result = set(completer.get_completions(
+        Document(text=text, cursor_position=position),
+        complete_event))
+    assert set(result) == set([
+        Completion(text='users', start_position=0),
+        Completion(text='orders', start_position=0)])
+
 def test_table_names_after_from(completer, complete_event):
     text = 'SELECT * FROM '
     position = len('SELECT * FROM ')