summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorOwen Stephens <owen@owenstephens.co.uk>2019-10-12 17:11:32 +0100
committerOwen Stephens <owen@owenstephens.co.uk>2019-10-12 17:11:32 +0100
commita700f2b788fa8050abb23598db8282050fd18e24 (patch)
tree5dd505ec4196b381f51057d7c115ccdcb9bc0dc0
parentb401b16d9a4332b3af64a5ce12c8ce6727885917 (diff)
fixup! Take account of table aliases when completing function args (#1048)
-rw-r--r--tests/test_smart_completion_multiple_schemata.py2
-rw-r--r--tests/test_smart_completion_public_schema_only.py4
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))