test-purges.php 2.43 KB
Newer Older
Erick Hitter's avatar
Erick Hitter committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111
<?php
/**
 * Test purge methods.
 *
 * @package WP_Revisions_Control
 */

/**
 * Class TestPurges.
 */
class TestPurges extends WP_UnitTestCase {
	/**
	 * Plugin's limit meta key.
	 *
	 * @var string
	 */
	protected static $meta_key = '_wp_rev_ctl_limit';

	/**
	 * Test revision purging.
	 */
	public function test_purge_all() {
		$post_id    = $this->factory->post->create();
		$iterations = 10;

		for ( $i = 0; $i < $iterations; $i++ ) {
			wp_update_post(
				array(
					'ID'           => $post_id,
					'post_content' => wp_rand(),
				)
			);
		}

		$revisions_to_purge = count( wp_get_post_revisions( $post_id ) );
		$this->assertEquals(
			$iterations,
			$revisions_to_purge,
			'Failed to assert that there are revisions to purge.'
		);

		$purge = WP_Revisions_Control::get_instance()->do_purge_all( $post_id );
		$revisions_remaining = count( wp_get_post_revisions( $post_id ) );

		$this->assertEquals(
			0,
			$revisions_remaining,
			'Failed to assert that all revisions were purged.'
		);

		$this->assertEquals(
			10,
			$purge['count'],
			'Failed to assert that response includes expected count of purged revisions.'
		);

		$this->assertEquals(
			'Removed 10 revisions associated with this post.',
			$purge['success'],
			'Failed to assert that response includes expected success message.'
		);
	}

	/**
	 * Test revision purging.
	 */
	public function test_purge_excess() {
		$post_id    = $this->factory->post->create();
		$iterations = 10;
		$limit      = 4;

		for ( $i = 0; $i < $iterations; $i++ ) {
			wp_update_post(
				array(
					'ID'           => $post_id,
					'post_content' => wp_rand(),
				)
			);
		}

		$revisions_to_purge = count( wp_get_post_revisions( $post_id ) );
		$this->assertEquals(
			$iterations,
			$revisions_to_purge,
			'Failed to assert that there are revisions to purge.'
		);

		update_post_meta( $post_id, static::$meta_key, $limit );

		$this->assertEquals(
			$limit,
			wp_revisions_to_keep( get_post( $post_id ) ),
			'Failed to assert that post is limited to a given number of revisions.'
		);

		$purge = WP_Revisions_Control::get_instance()->do_purge_excess( $post_id );
		$revisions_remaining = count( wp_get_post_revisions( $post_id ) );

		$this->assertEquals(
			4,
			$revisions_remaining,
			'Failed to assert that specified number of revisions were retained.'
		);

		$this->assertEquals(
			6,
			$purge['count'],
			'Failed to assert that response includes expected count of purged revisions.'
		);
	}
}