Add tests for the 'post_status' behavior of `WP_Query`.

See #29167.

git-svn-id: https://develop.svn.wordpress.org/trunk@31047 602fd350-edb4-49c9-b593-d223f7449a82
This commit is contained in:
Boone Gorges 2015-01-04 20:35:12 +00:00
parent 04eef529b0
commit 969808e7bb
1 changed files with 302 additions and 0 deletions

View File

@ -0,0 +1,302 @@
<?php
/**
* @group query
*/
class Tests_Query_PostStatus extends WP_UnitTestCase {
public static $editor_user;
public static $author_user;
public static $editor_private_post;
public static $author_private_post;
public static $editor_privatefoo_post;
public static $author_privatefoo_post;
public static function setUpBeforeClass() {
$f = new WP_UnitTest_Factory;
self::$editor_user = $f->user->create( array( 'role' => 'editor' ) );
self::$author_user = $f->user->create( array( 'role' => 'author' ) );
self::$editor_private_post = $f->post->create( array( 'post_author' => self::$editor_user, 'post_status' => 'private' ) );
self::$author_private_post = $f->post->create( array( 'post_author' => self::$author_user, 'post_status' => 'private' ) );
// Custom status with private=true.
register_post_status( 'privatefoo', array( 'private' => true ) );
self::$editor_privatefoo_post = $f->post->create( array( 'post_author' => self::$editor_user, 'post_status' => 'privatefoo' ) );
self::$author_privatefoo_post = $f->post->create( array( 'post_author' => self::$author_user, 'post_status' => 'privatefoo' ) );
_unregister_post_status( 'privatefoo' );
self::commit_transaction();
}
public static function tearDownAfterClass() {
if ( is_multisite() ) {
wpmu_delete_user( self::$editor_user );
wpmu_delete_user( self::$author_user );
} else {
wp_delete_user( self::$editor_user );
wp_delete_user( self::$author_user );
}
wp_delete_post( self::$editor_private_post, true );
wp_delete_post( self::$author_private_post, true );
wp_delete_post( self::$editor_privatefoo_post, true );
wp_delete_post( self::$author_privatefoo_post, true );
self::commit_transaction();
}
public function test_any_should_not_include_statuses_where_exclude_from_search_is_true() {
register_post_status( 'foo', array( 'exclude_from_search' => true ) );
$q = new WP_Query( array(
'post_status' => array( 'any' ),
) );
$this->assertContains( "post_status <> 'foo'", $q->request );
}
public function test_any_should_include_statuses_where_exclude_from_search_is_false() {
register_post_status( 'foo', array( 'exclude_from_search' => false ) );
$q = new WP_Query( array(
'post_status' => array( 'any' ),
) );
$this->assertNotContains( "post_status <> 'foo'", $q->request );
}
public function test_private_should_be_included_if_perm_is_false() {
$q = new WP_Query( array(
'post_status' => array( 'private' ),
'perm' => false,
) );
$expected = array(
self::$editor_private_post,
self::$author_private_post,
);
$this->assertEqualSets( $expected, wp_list_pluck( $q->posts, 'ID' ) );
}
public function test_private_should_not_be_included_for_non_author_if_perm_is_not_false() {
// Current user is 0.
$q = new WP_Query( array(
'post_status' => array( 'private' ),
'perm' => 'editable',
) );
$this->assertEmpty( $q->posts );
}
public function test_private_should_be_included_only_for_current_user_if_perm_is_readable_and_user_cannot_read_others_posts() {
wp_set_current_user( self::$author_user );
$q = new WP_Query( array(
'post_status' => array( 'private' ),
'perm' => 'readable',
) );
$expected = array(
self::$author_private_post,
);
$this->assertEqualSets( $expected, wp_list_pluck( $q->posts, 'ID' ) );
}
public function test_private_should_be_included_for_all_users_if_perm_is_readable_and_user_can_read_others_posts() {
wp_set_current_user( self::$editor_user );
$q = new WP_Query( array(
'post_status' => array( 'private' ),
'perm' => 'readable',
) );
$expected = array(
self::$author_private_post,
self::$editor_private_post,
);
$this->assertEqualSets( $expected, wp_list_pluck( $q->posts, 'ID' ) );
}
public function test_private_should_be_included_only_for_current_user_if_perm_is_editable_and_user_cannot_read_others_posts() {
wp_set_current_user( self::$author_user );
$q = new WP_Query( array(
'post_status' => array( 'private' ),
'perm' => 'editable',
) );
$expected = array(
self::$author_private_post,
);
$this->assertEqualSets( $expected, wp_list_pluck( $q->posts, 'ID' ) );
}
public function test_private_should_be_included_for_all_users_if_perm_is_editable_and_user_can_read_others_posts() {
wp_set_current_user( self::$editor_user );
$q = new WP_Query( array(
'post_status' => array( 'private' ),
'perm' => 'editable',
) );
$expected = array(
self::$author_private_post,
self::$editor_private_post,
);
$this->assertEqualSets( $expected, wp_list_pluck( $q->posts, 'ID' ) );
}
public function test_all_public_post_stati_should_be_included_when_no_post_status_is_provided() {
register_post_status( 'foo', array( 'public' => true ) );
$q = new WP_Query( array(
'posts_per_page' => 1, // Or the query will short-circuit.
) );
foreach ( get_post_stati( array( 'public' => true ) ) as $status ) {
$this->assertContains( "post_status = '$status'", $q->request );
}
}
public function test_protected_should_not_be_included_when_not_in_the_admin() {
register_post_status( 'foo', array( 'protected' => true ) );
$q = new WP_Query( array(
'posts_per_page' => 1, // Or the query will short-circuit.
) );
$this->assertNotContains( "post_status = 'foo", $q->request );
}
public function test_protected_should_be_included_when_in_the_admin() {
set_current_screen( 'dashboard' );
register_post_status( 'foo', array( 'protected' => true, 'show_in_admin_all_list' => true ) );
$q = new WP_Query( array(
'posts_per_page' => -1, // Or the query will short-circuit.
) );
$this->assertContains( "post_status = 'foo", $q->request );
set_current_screen( 'front' );
}
public function test_private_statuses_should_be_included_when_current_user_can_read_private_posts() {
wp_set_current_user( self::$editor_user );
register_post_status( 'privatefoo', array( 'private' => true ) );
$q = new WP_Query( array(
'posts_per_page' => -1,
) );
global $wpdb;
//print_r( $wpdb->get_results( "SELECT * FROM $wpdb->posts" ) );
$this->assertContains( self::$author_privatefoo_post, wp_list_pluck( $q->posts, 'ID' ) );
$this->assertContains( self::$editor_privatefoo_post, wp_list_pluck( $q->posts, 'ID' ) );
}
public function test_private_statuses_should_not_be_included_when_current_user_cannot_read_private_posts() {
wp_set_current_user( self::$author_user );
register_post_status( 'privatefoo', array( 'private' => true ) );
$q = new WP_Query( array(
'posts_per_page' => 2, // Or the query will short-circuit.
) );
$expected = array(
self::$author_privatefoo_post,
);
$this->assertContains( self::$author_privatefoo_post, wp_list_pluck( $q->posts, 'ID' ) );
$this->assertNotContains( self::$editor_privatefoo_post, wp_list_pluck( $q->posts, 'ID' ) );
}
public function test_single_post_with_nonpublic_status_should_not_be_shown_to_logged_out_users() {
register_post_type( 'foo_pt' );
register_post_status( 'foo_ps', array( 'public' => false ) );
$p = $this->factory->post->create( array( 'post_status' => 'foo_ps' ) );
$q = new WP_Query( array(
'p' => $p,
) );
$this->assertEmpty( $q->posts );
}
public function test_single_post_with_nonpublic_and_protected_status_should_not_be_shown_for_user_who_cannot_edit_others_posts() {
register_post_type( 'foo_pt' );
register_post_status( 'foo_ps', array( 'public' => false, 'protected' => true ) );
$p = $this->factory->post->create( array( 'post_status' => 'foo_ps', 'post_author' => self::$editor_user ) );
wp_set_current_user( self::$author_user );
$q = new WP_Query( array(
'p' => $p,
) );
$this->assertEmpty( $q->posts );
}
public function test_single_post_with_nonpublic_and_protected_status_should_be_shown_for_user_who_can_edit_others_posts() {
register_post_type( 'foo_pt' );
register_post_status( 'foo_ps', array( 'public' => false, 'protected' => true ) );
$p = $this->factory->post->create( array( 'post_status' => 'foo_ps', 'post_author' => self::$author_user ) );
wp_set_current_user( self::$editor_user );
$q = new WP_Query( array(
'p' => $p,
) );
$this->assertEquals( array( $p ), wp_list_pluck( $q->posts, 'ID' ) );
}
public function test_single_post_with_nonpublic_and_private_status_should_not_be_shown_for_user_who_cannot_edit_others_posts() {
register_post_type( 'foo_pt' );
register_post_status( 'foo_ps', array( 'public' => false, 'private' => true ) );
$p = $this->factory->post->create( array( 'post_status' => 'foo_ps', 'post_author' => self::$editor_user ) );
wp_set_current_user( self::$author_user );
$q = new WP_Query( array(
'p' => $p,
) );
$this->assertEmpty( $q->posts );
}
public function test_single_post_with_nonpublic_and_private_status_should_be_shown_for_user_who_can_edit_others_posts() {
register_post_type( 'foo_pt' );
register_post_status( 'foo_ps', array( 'public' => false, 'private' => true ) );
$p = $this->factory->post->create( array( 'post_status' => 'foo_ps', 'post_author' => self::$author_user ) );
wp_set_current_user( self::$editor_user );
$q = new WP_Query( array(
'p' => $p,
) );
$this->assertEquals( array( $p ), wp_list_pluck( $q->posts, 'ID' ) );
}
public function test_single_post_with_nonpublic_and_protected_status_should_not_be_shown_for_any_user() {
register_post_type( 'foo_pt' );
register_post_status( 'foo_ps', array( 'public' => false ) );
$p = $this->factory->post->create( array( 'post_status' => 'foo_ps', 'post_author' => self::$author_user ) );
wp_set_current_user( self::$editor_user );
$q = new WP_Query( array(
'p' => $p,
) );
$this->assertEmpty( $q->posts );
}
}