Ticket #34010: 34010.3.diff

src/wp-includes/post.php

@@ -1639 +1639 @@
 }
 
 /**
+ * Get a list of post type names that support a specific feature.
+ *
+ * @since 4.5.0
+ *
+ * @global array $_wp_post_type_features
+ *
+ * @param array|string $args     Single feature or an array of features the post types should support.
+ * @param string       $operator Optional. The logical operation to perform. 'or' means
+ *                               only one element from the array needs to match; 'and'
+ *                               means all elements must match; 'not' means no elements may
+ *                               match. Default 'and'.
+ * @return array A list of post type names.
+ */
+function get_post_types_by_support( $args, $operator = 'and' ) {
+        global $_wp_post_type_features;
+
+        $features = array_fill_keys( (array) $args, true );
+
+        return array_keys( wp_filter_object_list( $_wp_post_type_features, $features, $operator ) );
+}
+
+/**
  * Update the post type for the post ID.
  *
  * The page or post cache will be cleaned for the post ID.

tests/phpunit/tests/post/types.php

@@ -379 +379 @@
 
                 $this->assertFalse( post_type_exists( 'foo' ) );
         }
+
+        /**
+         * @ticket 34010
+         */
+        public function test_get_post_types_by_support_single_feature() {
+                $this->assertEqualSets( array(
+                        'post',
+                        'page',
+                        'attachment',
+                        'nav_menu_item',
+                ), get_post_types_by_support( 'title' ) );
+        }
+
+        /**
+         * @ticket 34010
+         */
+        public function test_get_post_types_by_support_multiple_features() {
+                $this->assertEqualSets( array( 'post', 'page' ), get_post_types_by_support( array( 'thumbnail', 'author' ) ) );
+        }
+
+        /**
+         * @ticket 34010
+         */
+        public function test_get_post_types_by_support_or_operator() {
+                $this->assertEqualSets( array( 'post', 'page' ), get_post_types_by_support( array( 'post-formats', 'page-attributes' ), 'or' ) );
+        }
+
+        /**
+         * @ticket 34010
+         */
+        public function test_get_post_types_by_support_not_operator() {
+                $this->assertEqualSets( array( 'attachment', 'revision', 'nav_menu_item' ), get_post_types_by_support( array( 'thumbnail' ), 'not' ) );
+        }
+
+        /**
+         * @ticket 34010
+         */
+        public function test_get_post_types_by_support_excluding_features() {
+                $this->assertEqualSets( array(), get_post_types_by_support( array( 'post-formats', 'page-attributes' ) ) );
+        }
+
+        /**
+         * @ticket 34010
+         */
+        public function test_get_post_types_by_support_non_existant_feature() {
+                $this->assertEqualSets( array(), get_post_types_by_support( 'somefeature' ) );
+        }
 }