wordpress-develop/tests/phpunit/tests/comment/getCommentCount.php

<?php

/**
 * @group comment
 *
 * Class Tests_Get_Comment_Count
 *
 * @covers ::get_comment_count
 */
class Tests_Get_Comment_Count extends WP_UnitTestCase {

	public function test_get_comment_count() {
		$count = get_comment_count();

		$this->assertSame( 0, $count['approved'] );
		$this->assertSame( 0, $count['awaiting_moderation'] );
		$this->assertSame( 0, $count['spam'] );
		$this->assertSame( 0, $count['trash'] );
		$this->assertSame( 0, $count['post-trashed'] );
		$this->assertSame( 0, $count['total_comments'] );
		$this->assertSame( 0, $count['all'] );
	}

	public function test_get_comment_count_approved() {
		self::factory()->comment->create(
			array(
				'comment_approved' => 1,
			)
		);

		$count = get_comment_count();

		$this->assertSame( 1, $count['approved'] );
		$this->assertSame( 0, $count['awaiting_moderation'] );
		$this->assertSame( 0, $count['spam'] );
		$this->assertSame( 0, $count['trash'] );
		$this->assertSame( 0, $count['post-trashed'] );
		$this->assertSame( 1, $count['total_comments'] );
	}

	public function test_get_comment_count_awaiting() {
		self::factory()->comment->create(
			array(
				'comment_approved' => 0,
			)
		);

		$count = get_comment_count();

		$this->assertSame( 0, $count['approved'] );
		$this->assertSame( 1, $count['awaiting_moderation'] );
		$this->assertSame( 0, $count['spam'] );
		$this->assertSame( 0, $count['trash'] );
		$this->assertSame( 0, $count['post-trashed'] );
		$this->assertSame( 1, $count['total_comments'] );
	}

	public function test_get_comment_count_spam() {
		self::factory()->comment->create(
			array(
				'comment_approved' => 'spam',
			)
		);

		$count = get_comment_count();

		$this->assertSame( 0, $count['approved'] );
		$this->assertSame( 0, $count['awaiting_moderation'] );
		$this->assertSame( 1, $count['spam'] );
		$this->assertSame( 0, $count['trash'] );
		$this->assertSame( 0, $count['post-trashed'] );
		$this->assertSame( 1, $count['total_comments'] );
	}

	public function test_get_comment_count_trash() {
		self::factory()->comment->create(
			array(
				'comment_approved' => 'trash',
			)
		);

		$count = get_comment_count();

		$this->assertSame( 0, $count['approved'] );
		$this->assertSame( 0, $count['awaiting_moderation'] );
		$this->assertSame( 0, $count['spam'] );
		$this->assertSame( 1, $count['trash'] );
		$this->assertSame( 0, $count['post-trashed'] );
		$this->assertSame( 0, $count['total_comments'] );
	}

	public function test_get_comment_count_post_trashed() {
		self::factory()->comment->create(
			array(
				'comment_approved' => 'post-trashed',
			)
		);

		$count = get_comment_count();

		$this->assertSame( 0, $count['approved'] );
		$this->assertSame( 0, $count['awaiting_moderation'] );
		$this->assertSame( 0, $count['spam'] );
		$this->assertSame( 0, $count['trash'] );
		$this->assertSame( 1, $count['post-trashed'] );
		$this->assertSame( 0, $count['total_comments'] );
	}

	/**
	 * @ticket 19901
	 *
	 * @covers ::get_comment_count
	 */
	public function test_get_comment_count_validate_cache_comment_deleted() {

		$comment_id = self::factory()->comment->create();

		$count = get_comment_count();

		$this->assertSame( 1, $count['total_comments'] );

		wp_delete_comment( $comment_id, true );

		$count = get_comment_count();

		$this->assertSame( 0, $count['total_comments'] );
	}

	/**
	 * @ticket 19901
	 *
	 * @covers ::get_comment_count
	 */
	public function test_get_comment_count_validate_cache_post_deleted() {

		$post_id = self::factory()->post->create();

		$comment_id = self::factory()->comment->create(
			array(
				'comment_post_ID' => $post_id,
			)
		);

		$count = get_comment_count( $post_id );

		$this->assertSame( 1, $count['total_comments'] );

		wp_delete_post( $post_id, true );

		$count = get_comment_count( $post_id );

		$this->assertSame( 0, $count['total_comments'] );
	}

	/**
	 * @ticket 19901
	 *
	 * @covers ::get_comment_count
	 */
	public function test_get_comment_count_validate_cache_comment_status() {
		$comment_id = self::factory()->comment->create();

		$count = get_comment_count();

		$this->assertSame( 1, $count['approved'] );
		$this->assertSame( 0, $count['trash'] );
		$this->assertSame( 1, $count['total_comments'] );

		wp_set_comment_status( $comment_id, 'trash' );

		$count = get_comment_count();

		$this->assertSame( 0, $count['approved'] );
		$this->assertSame( 1, $count['trash'] );
		$this->assertSame( 0, $count['total_comments'] );
	}
}