Changeset 25554 for trunk/tests/phpunit/tests/post.php

Timestamp:

09/21/2013 05:54:36 PM (13 years ago)

Author:

wonderboymusic

Message:

Add hooks to wp_count_posts(). Adds filter docs. Adds unit test to test count_posts filter.

Props nacin, DrewAPicture.
Fixes #16603.

File:

• trunk/tests/phpunit/tests/post.php (modified) (1 diff)

trunk/tests/phpunit/tests/post.php

@@ -809 +809 @@
     }
 
+    function test_wp_count_posts_filtered() {
+        $post_type = rand_str(20);
+        register_post_type( $post_type );
+        $this->factory->post->create_many( 10, array(
+            'post_type' => $post_type,
+            'post_author' => $this->author_id
+        ) );
+        $count1 = wp_count_posts( $post_type, 'readable' );
+        $this->assertEquals( 10, $count1->publish );
+        add_filter( 'count_posts', array( $this, 'filter_wp_count_posts' ) );
+
+        $count2 = wp_count_posts( $post_type, 'readable' );
+        $this->assertEquals( 7, $count2->publish );
+
+        remove_filter( 'count_posts', array( $this, 'filter_wp_count_posts' ) );
+    }
+
+    function filter_wp_count_posts( $counts ) {
+        $counts->publish = 7;
+        return $counts;
+    }
+
     /**
      * @ticket 22074