diff options
author | koljonen <koljonen@outlook.com> | 2016-06-10 03:26:42 +0200 |
---|---|---|
committer | koljonen <koljonen@outlook.com> | 2016-06-10 03:26:42 +0200 |
commit | 582852adb690866e6d149ab1bcaddf88cbfb8c69 (patch) | |
tree | ff7519c714195a8850618f3ac95ab3a442fd73ba /tests/test_sqlcompletion.py | |
parent | 9b6a72e9e00bae4c1247153205e90ba549e2f1ea (diff) |
Various changes after review
Diffstat (limited to 'tests/test_sqlcompletion.py')
-rw-r--r-- | tests/test_sqlcompletion.py | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/tests/test_sqlcompletion.py b/tests/test_sqlcompletion.py index bce9fe7a..de95936b 100644 --- a/tests/test_sqlcompletion.py +++ b/tests/test_sqlcompletion.py @@ -111,9 +111,24 @@ def test_suggest_tables_views_schemas_and_functions(expression): @pytest.mark.parametrize('expression', [ - 'SELECT * FROM foo JOIN ', + 'SELECT * FROM foo JOIN bar on bar.barid = foo.barid JOIN ', + 'SELECT * FROM foo JOIN bar USING (barid) JOIN ', ]) -def test_suggest_tables_views_schemas_functions_and_joins(expression): +def test_suggest_after_join_with_two_tables(expression): + suggestions = suggest_type(expression, expression) + assert set(suggestions) == set([ + Table(schema=None), + View(schema=None), + Function(schema=None, filter='for_from_clause'), + Join(((None, 'foo', None, False), (None, 'bar', None, False)), None), + Schema(), + ]) + + +@pytest.mark.parametrize('expression', [ + 'SELECT * FROM foo JOIN ' +]) +def test_suggest_after_join_with_one_table(expression): suggestions = suggest_type(expression, expression) assert set(suggestions) == set([ Table(schema=None), |