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