diff --git a/tests/phpunit/tests/multisite/updatePostsCount.php b/tests/phpunit/tests/multisite/updatePostsCount.php index 850d9fff59..270ac60c49 100644 --- a/tests/phpunit/tests/multisite/updatePostsCount.php +++ b/tests/phpunit/tests/multisite/updatePostsCount.php @@ -21,10 +21,21 @@ if ( is_multisite() ) : * @covers ::_update_posts_count_on_delete */ public function test_update_posts_count() { + $blog_id = self::factory()->blog->create(); + switch_to_blog( $blog_id ); + $original_post_count = (int) get_site()->post_count; $post_id = self::factory()->post->create(); + $post_count_after_creating = get_site()->post_count; + + wp_delete_post( $post_id ); + + $post_count_after_deleting = get_site()->post_count; + + restore_current_blog(); + /* * Check that posts count is updated when a post is created: * add_action( 'transition_post_status', '_update_posts_count_on_transition_post_status', 10, 3 ); @@ -32,9 +43,7 @@ if ( is_multisite() ) : * Check that _update_posts_count_on_transition_post_status() is called on that filter, * which then calls update_posts_count() to update the count. */ - $this->assertSame( $original_post_count + 1, get_site()->post_count, 'Post count should be incremented by 1.' ); - - wp_delete_post( $post_id ); + $this->assertSame( $original_post_count + 1, $post_count_after_creating, 'Post count should be incremented by 1.' ); /* * Check that posts count is updated when a post is deleted: @@ -43,7 +52,7 @@ if ( is_multisite() ) : * Check that _update_posts_count_on_delete() is called on that filter, * which then calls update_posts_count() to update the count. */ - $this->assertSame( $original_post_count, get_site()->post_count, 'Post count should match the original count.' ); + $this->assertSame( $original_post_count, $post_count_after_deleting, 'Post count should match the original count.' ); } }