From 1c022c910ef0924699841e272ebe6e881b37e7f1 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Fri, 14 Jan 2022 15:44:37 +0100 Subject: Merging upstream version 1.24.2. Signed-off-by: Daniel Baumann --- test/test_completion_engine.py | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'test/test_completion_engine.py') 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 ', -- cgit v1.2.3