Suggest function signature after `CREATE FUNCTION`

1 files changed, 3 insertions, 3 deletions

diff --git a/tests/test_smart_completion_public_schema_only.py b/tests/test_smart_completion_public_schema_only.py
index 2c1751a7..a8130b1b 100644
--- a/tests/test_smart_completion_public_schema_only.py
+++ b/tests/test_smart_completion_public_schema_only.py
@@ -87,10 +87,10 @@ def test_function_column_name(completer):
         )
 
 
-@parametrize('text', ['ALTER FUNCTION set_ret', 'DROP FUNCTION set_ret'])
+@parametrize('action', ['ALTER', 'DROP', 'CREATE', 'CREATE OR REPLACE'])
 @parametrize('completer', completers())
-def test_drop_alter_function(completer, text):
-    assert get_result(completer, text) == [
+def test_drop_alter_function(completer, action):
+    assert get_result(completer, action + ' FUNCTION set_ret') == [
         function('set_returning_func(x integer, y integer)', -len('set_ret'))
     ]