Improve robustness and fix test scenarios

Co-authored-by: swissspidy <[email protected]>

Author: Copilot

features/ability-category.feature

@@ -3,15 +3,8 @@ Feature: Manage WordPress ability categories
   Background:
     Given a WP install
 
-  @require-wp-6.9
+  @less-than-wp-6.9
   Scenario: Ability category commands require WordPress 6.9+
-    When I try `wp core download --version=6.8 --force`
-    And I run `wp core version`
-    Then STDOUT should contain:
-      """
-      6.8
-      """
-
     When I try `wp ability category list`
     Then STDERR should contain:
       """

features/ability.feature

@@ -3,15 +3,8 @@ Feature: Manage WordPress abilities
   Background:
     Given a WP install
 
-  @require-wp-6.9
+  @less-than-wp-6.9
   Scenario: Ability commands require WordPress 6.9+
-    When I try `wp core download --version=6.8 --force`
-    And I run `wp core version`
-    Then STDOUT should contain:
-      """
-      6.8
-      """
-
     When I try `wp ability list`
     Then STDERR should contain:
       """

src/Ability_Category_Command.php

@@ -81,7 +81,15 @@ class Ability_Category_Command extends WP_CLI_Command {
 	public function list_( $args, $assoc_args ) {
 		$formatter = $this->get_formatter( $assoc_args );
 
-		$categories = wp_get_ability_categories();
+		// Get all registered ability categories
+		if ( function_exists( 'wp_get_ability_categories' ) ) {
+			$categories = wp_get_ability_categories();
+		} elseif ( function_exists( 'wp_ability_categories' ) ) {
+			$registry = wp_ability_categories();
+			$categories = $registry->get_all();
+		} else {
+			$categories = array();
+		}
 
 		if ( empty( $categories ) ) {
 			$categories = array();

src/Ability_Command.php

@@ -101,7 +101,15 @@ class Ability_Command extends WP_CLI_Command {
 	public function list_( $args, $assoc_args ) {
 		$formatter = $this->get_formatter( $assoc_args );
 
-		$abilities = wp_get_abilities();
+		// Get all registered abilities
+		if ( function_exists( 'wp_get_abilities' ) ) {
+			$abilities = wp_get_abilities();
+		} elseif ( function_exists( 'wp_abilities' ) ) {
+			$registry = wp_abilities();
+			$abilities = $registry->get_all();
+		} else {
+			$abilities = array();
+		}
 
 		if ( empty( $abilities ) ) {
 			$abilities = array();
@@ -291,7 +299,16 @@ public function execute( $args, $assoc_args ) {
 		}
 
 		// Execute the ability
-		$result = wp_execute_ability( $ability_name, $input );
+		// First try wp_execute_ability if it exists, otherwise call the callback directly
+		if ( function_exists( 'wp_execute_ability' ) ) {
+			$result = wp_execute_ability( $ability_name, $input );
+		} else {
+			$ability = wp_get_ability( $ability_name );
+			if ( ! $ability || ! is_callable( $ability->callback ) ) {
+				WP_CLI::error( "Cannot execute ability {$ability_name}." );
+			}
+			$result = call_user_func( $ability->callback, $input );
+		}
 
 		if ( is_wp_error( $result ) ) {
 			WP_CLI::error( $result->get_error_message() );