adding support for multiples -c / --command parameter to pgcli w/tests

Author: DiegoDAF

AUTHORS

@@ -144,6 +144,7 @@ Contributors:
     * Jay Knight (jay-knight)
     * fbdb
     * Charbel Jacquin (charbeljc)
+    * Diego
 
 Creator:
 --------

tests/features/command_option.feature

@@ -26,3 +26,13 @@ Feature: run the cli with -c/--command option,
      When we run pgcli with -c "SELECT 1; SELECT 2"
       then we see both query results
       and pgcli exits successfully
+
+  Scenario: run pgcli with multiple -c options
+     When we run pgcli with multiple -c options
+      then we see all command outputs
+      and pgcli exits successfully
+
+  Scenario: run pgcli with mixed -c and --command options
+     When we run pgcli with mixed -c and --command
+      then we see all command outputs
+      and pgcli exits successfully

tests/features/steps/command_option.py

@@ -121,3 +121,72 @@ def step_pgcli_exits_with_error(context):
     # Clean up
     context.cmd_output = None
     context.exit_code = None
+
+
+@when("we run pgcli with multiple -c options")
+def step_run_pgcli_with_multiple_c(context):
+    """Run pgcli with multiple -c flags."""
+    cmd = [
+        "pgcli",
+        "-h", context.conf["host"],
+        "-p", str(context.conf["port"]),
+        "-U", context.conf["user"],
+        "-d", context.conf["dbname"],
+        "-c", "SELECT 'first' as result",
+        "-c", "SELECT 'second' as result",
+        "-c", "SELECT 'third' as result"
+    ]
+    try:
+        context.cmd_output = subprocess.check_output(
+            cmd,
+            cwd=context.package_root,
+            stderr=subprocess.STDOUT,
+            timeout=10
+        )
+        context.exit_code = 0
+    except subprocess.CalledProcessError as e:
+        context.cmd_output = e.output
+        context.exit_code = e.returncode
+    except subprocess.TimeoutExpired as e:
+        context.cmd_output = b"Command timed out"
+        context.exit_code = -1
+
+
+@when("we run pgcli with mixed -c and --command")
+def step_run_pgcli_with_mixed_options(context):
+    """Run pgcli with mixed -c and --command flags."""
+    cmd = [
+        "pgcli",
+        "-h", context.conf["host"],
+        "-p", str(context.conf["port"]),
+        "-U", context.conf["user"],
+        "-d", context.conf["dbname"],
+        "-c", "SELECT 'from_c' as source",
+        "--command", "SELECT 'from_command' as source"
+    ]
+    try:
+        context.cmd_output = subprocess.check_output(
+            cmd,
+            cwd=context.package_root,
+            stderr=subprocess.STDOUT,
+            timeout=10
+        )
+        context.exit_code = 0
+    except subprocess.CalledProcessError as e:
+        context.cmd_output = e.output
+        context.exit_code = e.returncode
+    except subprocess.TimeoutExpired as e:
+        context.cmd_output = b"Command timed out"
+        context.exit_code = -1
+
+
+@then("we see all command outputs")
+def step_see_all_command_outputs(context):
+    """Verify that all command outputs are present."""
+    output = context.cmd_output.decode('utf-8')
+    # Should contain output from all commands
+    assert "first" in output or "from_c" in output, f"Expected 'first' or 'from_c' in output, but got: {output}"
+    assert "second" in output or "from_command" in output, f"Expected 'second' or 'from_command' in output, but got: {output}"
+    # For the 3-command test, also check for third
+    if "third" in output or "result" in output:
+        assert "third" in output, f"Expected 'third' in output for 3-command test, but got: {output}"