diff options
author | Owen Stephens <owen@owenstephens.co.uk> | 2019-10-12 17:11:32 +0100 |
---|---|---|
committer | Owen Stephens <owen@owenstephens.co.uk> | 2019-10-12 17:11:32 +0100 |
commit | a700f2b788fa8050abb23598db8282050fd18e24 (patch) | |
tree | 5dd505ec4196b381f51057d7c115ccdcb9bc0dc0 | |
parent | b401b16d9a4332b3af64a5ce12c8ce6727885917 (diff) |
fixup! Take account of table aliases when completing function args (#1048)
-rw-r--r-- | tests/test_smart_completion_multiple_schemata.py | 2 | ||||
-rw-r--r-- | tests/test_smart_completion_public_schema_only.py | 4 |
2 files changed, 4 insertions, 2 deletions
diff --git a/tests/test_smart_completion_multiple_schemata.py b/tests/test_smart_completion_multiple_schemata.py index 9125f124..9fff2a17 100644 --- a/tests/test_smart_completion_multiple_schemata.py +++ b/tests/test_smart_completion_multiple_schemata.py @@ -229,7 +229,7 @@ def test_suggested_column_names_in_function(completer): completer, "SELECT MAX( from custom.products", len("SELECT MAX(") ) assert completions_to_set(result) == completions_to_set( - testdata.columns("products", "custom") + testdata.columns_functions_and_keywords("products", "custom") ) diff --git a/tests/test_smart_completion_public_schema_only.py b/tests/test_smart_completion_public_schema_only.py index 63f1e80e..bb371543 100644 --- a/tests/test_smart_completion_public_schema_only.py +++ b/tests/test_smart_completion_public_schema_only.py @@ -287,7 +287,9 @@ def test_suggested_cased_always_qualified_column_names(completer): @parametrize("completer", completers(casing=False, qualify=no_qual)) def test_suggested_column_names_in_function(completer): result = get_result(completer, "SELECT MAX( from users", len("SELECT MAX(")) - assert completions_to_set(result) == completions_to_set(testdata.columns("users")) + assert completions_to_set(result) == completions_to_set( + (testdata.columns_functions_and_keywords("users")) + ) @parametrize("completer", completers(casing=False)) |