diff options
Diffstat (limited to 'test/test_completion_engine.py')
| -rw-r--r-- | test/test_completion_engine.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/test/test_completion_engine.py b/test/test_completion_engine.py index 9e7c608..8b06ed3 100644 --- a/test/test_completion_engine.py +++ b/test/test_completion_engine.py @@ -393,6 +393,17 @@ def test_join_using_suggests_common_columns(col_list): 'tables': [(None, 'abc', None), (None, 'def', None)], 'drop_unique': True}] +@pytest.mark.parametrize('sql', [ + 'SELECT * FROM abc a JOIN def d ON a.id = d.id JOIN ghi g ON g.', + 'SELECT * FROM abc a JOIN def d ON a.id = d.id AND a.id2 = d.id2 JOIN ghi g ON d.id = g.id AND g.', +]) +def test_two_join_alias_dot_suggests_cols1(sql): + suggestions = suggest_type(sql, sql) + assert sorted_dicts(suggestions) == sorted_dicts([ + {'type': 'column', 'tables': [(None, 'ghi', 'g')]}, + {'type': 'table', 'schema': 'g'}, + {'type': 'view', 'schema': 'g'}, + {'type': 'function', 'schema': 'g'}]) def test_2_statements_2nd_current(): suggestions = suggest_type('select * from a; select * from ', |