diff --git a/src/wp-includes/class-wp-post-type.php b/src/wp-includes/class-wp-post-type.php index f2e14a0a93..faf7a0d496 100644 --- a/src/wp-includes/class-wp-post-type.php +++ b/src/wp-includes/class-wp-post-type.php @@ -513,7 +513,13 @@ final class WP_Post_Type { */ public function add_supports() { if ( ! empty( $this->supports ) ) { - add_post_type_support( $this->name, $this->supports ); + foreach ( $this->supports as $feature => $args ) { + if ( is_array( $args ) ) { + add_post_type_support( $this->name, $feature, $args ); + } else { + add_post_type_support( $this->name, $args ); + } + } unset( $this->supports ); } elseif ( false !== $this->supports ) { // Add default features. diff --git a/tests/phpunit/tests/post/wpPostType.php b/tests/phpunit/tests/post/wpPostType.php index 793e61cae9..041d1320c2 100644 --- a/tests/phpunit/tests/post/wpPostType.php +++ b/tests/phpunit/tests/post/wpPostType.php @@ -62,6 +62,47 @@ class Tests_WP_Post_Type extends WP_UnitTestCase { $this->assertEqualSets( array(), $post_type_supports_after ); } + /** + * Test that supports can optionally receive nested args. + * + * @ticket 40413 + */ + public function test_add_supports_custom_with_args() { + $post_type = 'cpt'; + $post_type_object = new WP_Post_Type( + $post_type, + array( + 'supports' => array( + 'support_with_args' => array( + 'arg1', + 'arg2', + ), + 'support_without_args', + ), + ) + ); + + $post_type_object->add_supports(); + $post_type_supports = get_all_post_type_supports( $post_type ); + + $post_type_object->remove_supports(); + $post_type_supports_after = get_all_post_type_supports( $post_type ); + + $this->assertEqualSets( + array( + 'support_with_args' => array( + array( + 'arg1', + 'arg2', + ), + ), + 'support_without_args' => true, + ), + $post_type_supports + ); + $this->assertEqualSets( array(), $post_type_supports_after ); + } + public function test_does_not_add_query_var_if_not_public() { $this->set_permalink_structure( '/%postname%' );