Add a test to verify error message.

2 files changed, 19 insertions, 0 deletions

diff --git a/tests/features/basic_commands.feature b/tests/features/basic_commands.feature
index 1f63a8ad..99f893e2 100644
--- a/tests/features/basic_commands.feature
+++ b/tests/features/basic_commands.feature
@@ -10,6 +10,11 @@ Feature: run the cli,
      When we send source command
       then we see help output
 
+  Scenario: run partial select command
+     When we send partial select command
+       then we see error message
+       then we see dbcli prompt
+
   Scenario: check our application_name
      When we run query to check application_name
       then we see found
diff --git a/tests/features/steps/basic_commands.py b/tests/features/steps/basic_commands.py
index b9ab0046..bda163f6 100644
--- a/tests/features/steps/basic_commands.py
+++ b/tests/features/steps/basic_commands.py
@@ -82,6 +82,20 @@ def step_send_help(context):
     context.cli.sendline("\?")
 
 
+@when("we send partial select command")
+def step_send_partial_select_command(context):
+    """
+    Send `SELECT a` to see completion.
+    """
+    context.cli.sendline("SELECT a")
+
+
+@then("we see error message")
+def step_see_error_message(context):
+    wrappers.expect_exact(context, 'column "a" does not exist', timeout=2)
+    wrappers.wait_prompt(context)
+
+
 @when("we send source command")
 def step_send_source_command(context):
     context.tmpfile_sql_help = tempfile.NamedTemporaryFile(prefix="pgcli_")