Merge pull request #267 from wp-cli/266-correct-error-message

wp-cli · Aug 13, 2024 · 6c09f32 · 6c09f32
2 parents 5cf12ef + 176dc75
commit 6c09f32
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/features/db-search.feature b/features/db-search.feature
@@ -1102,5 +1102,13 @@ Feature: Search through the database
     And STDOUT should be empty
     And the return code should be 1
 
+    When I try `wp db search example.com --format=count`
+    Then STDERR should be:
+      """
+      Error: The "count" format can only be used for a single table.
+      """
+    And STDOUT should be empty
+    And the return code should be 1
+
     When I run `wp db search mail.example.com wp_options --format=ids`
     Then STDOUT should not be empty
diff --git a/src/DB_Command.php b/src/DB_Command.php
@@ -1553,7 +1553,7 @@ public function search( $args, $assoc_args ) {
 
 			if ( in_array( $format, [ 'ids', 'count' ], true ) ) {
 				if ( count( $tables ) > 1 ) {
-					WP_CLI::error( 'The "ids" format can only be used for a single table.' );
+					WP_CLI::error( "The \"{$format}\" format can only be used for a single table." );
 				}
 				$search_results = array_column( $search_results, 'primary_key_value' );
 			}