Posts, Post Types: Ensure default terms are added by wp_publish_post().

Transitioning posts from `auto-draft` to `publish` via `wp_publish_post()` could result in published posts without the default category or custom taxonomy default terms.

Props frank-klein, TimothyBlynJacobs, peterwilsoncc.
Fixes #51292.



git-svn-id: https://develop.svn.wordpress.org/trunk@49000 602fd350-edb4-49c9-b593-d223f7449a82
This commit is contained in:
Peter Wilson 2020-09-19 01:20:48 +00:00
parent 2f491fba50
commit 1ffdfdaea7
3 changed files with 366 additions and 0 deletions

View File

@ -4375,6 +4375,33 @@ function wp_publish_post( $post ) {
return; return;
} }
// Ensure at least one term is applied for taxonomies with a default term.
foreach ( get_object_taxonomies( $post->post_type, 'object' ) as $taxonomy => $tax_object ) {
// Skip taxonomy if no default term is set.
if (
'category' !== $taxonomy &&
empty( $tax_object->default_term )
) {
continue;
}
// Do not modify previously set terms.
if ( ! empty( get_the_terms( $post, $taxonomy ) ) ) {
continue;
}
if ( 'category' === $taxonomy ) {
$default_term_id = (int) get_option( 'default_category', 0 );
} else {
$default_term_id = (int) get_option( 'default_term_' . $taxonomy, 0 );
}
if ( ! $default_term_id ) {
continue;
}
wp_set_post_terms( $post->ID, array( $default_term_id ), $taxonomy );
}
$wpdb->update( $wpdb->posts, array( 'post_status' => 'publish' ), array( 'ID' => $post->ID ) ); $wpdb->update( $wpdb->posts, array( 'post_status' => 'publish' ), array( 'ID' => $post->ID ) );
clean_post_cache( $post->ID ); clean_post_cache( $post->ID );

View File

@ -0,0 +1,153 @@
<?php
/**
* @group post
*/
class Tests_WPPublishPost extends WP_UnitTestCase {
/**
* Auto-draft post ID.
*
* @var int
*/
public static $auto_draft_id;
/**
* Create shared fixtures.
*
* @param WP_UnitTest_Factory $factory Test suite factory.
*/
public static function wpSetUpBeforeClass( $factory ) {
self::$auto_draft_id = $factory->post->create( array( 'post_status' => 'auto-draft' ) );
}
/**
* Ensure wp_publish_post does not add default category in error.
*
* @ticket 51292
*/
function test_wp_publish_post_respects_current_categories() {
$post_id = self::$auto_draft_id;
$category_id = $this->factory->term->create( array( 'taxonomy' => 'category' ) );
wp_set_post_categories( $post_id, $category_id );
wp_publish_post( $post_id );
$post_categories = get_the_category( $post_id );
$this->assertCount( 1, $post_categories );
$this->assertSame(
$category_id,
$post_categories[0]->term_id,
'wp_publish_post replaced set category.'
);
}
/**
* Ensure wp_publish_post adds default category.
*
* @covers wp_publish_post
* @ticket 51292
*/
function test_wp_publish_post_adds_default_category() {
$post_id = self::$auto_draft_id;
wp_publish_post( $post_id );
$post_categories = get_the_category( $post_id );
$this->assertCount( 1, $post_categories );
$this->assertSame(
(int) get_option( 'default_category' ),
$post_categories[0]->term_id,
'wp_publish_post failed to add default category.'
);
}
/**
* Ensure wp_publish_post adds default category when tagged.
*
* @covers wp_publish_post
* @ticket 51292
*/
function test_wp_publish_post_adds_default_category_when_tagged() {
$post_id = self::$auto_draft_id;
$tag_id = $this->factory->term->create( array( 'taxonomy' => 'post_tag' ) );
wp_set_post_tags( $post_id, array( $tag_id ) );
wp_publish_post( $post_id );
$post_categories = get_the_category( $post_id );
$this->assertCount( 1, $post_categories );
$this->assertSame(
(int) get_option( 'default_category' ),
$post_categories[0]->term_id,
'wp_publish_post failed to add default category.'
);
}
/**
* Ensure wp_publish_post does not add default term in error.
*
* @covers wp_publish_post
* @ticket 51292
*/
function test_wp_publish_post_respects_current_terms() {
// Create custom taxonomy to test with.
register_taxonomy(
'tax_51292',
'post',
array(
'hierarchical' => true,
'public' => true,
'default_term' => array(
'name' => 'Default 51292',
'slug' => 'default-51292',
),
)
);
$post_id = self::$auto_draft_id;
$term_id = $this->factory->term->create( array( 'taxonomy' => 'tax_51292' ) );
wp_set_object_terms( $post_id, array( $term_id ), 'tax_51292' );
wp_publish_post( $post_id );
$post_terms = get_the_terms( $post_id, 'tax_51292' );
$this->assertCount( 1, $post_terms );
$this->assertSame(
$term_id,
$post_terms[0]->term_id,
'wp_publish_post replaced set term for custom taxonomy.'
);
}
/**
* Ensure wp_publish_post adds default term.
*
* @covers wp_publish_post
* @ticket 51292
*/
function test_wp_publish_post_adds_default_term() {
// Create custom taxonomy to test with.
register_taxonomy(
'tax_51292',
'post',
array(
'hierarchical' => true,
'public' => true,
'default_term' => array(
'name' => 'Default 51292',
'slug' => 'default-51292',
),
)
);
$post_id = self::$auto_draft_id;
wp_publish_post( $post_id );
$post_terms = get_the_terms( $post_id, 'tax_51292' );
$this->assertCount( 1, $post_terms );
$this->assertSame(
get_term_by( 'slug', 'default-51292', 'tax_51292' )->term_id,
$post_terms[0]->term_id,
'wp_publish_post failed to add default term for custom taxonomy.'
);
}
}

View File

@ -0,0 +1,186 @@
<?php
/**
* @group taxonomy
*/
class Tests_Term_termCount extends WP_UnitTestCase {
/**
* Term ID for testing attachment counts.
*
* @var int
*/
public static $attachment_term;
/**
* Post IDs of shared posts.
*
* @var int[]
*/
public static $post_ids;
/**
* Create shared fixtures.
*
* @param WP_UnitTest_Factory $factory Test suite factory.
*/
public static function wpSetUpBeforeClass( $factory ) {
$statuses = array( 'publish', 'auto-draft', 'draft', 'private' );
foreach ( $statuses as $status ) {
self::$post_ids[ $status ] = $factory->post->create( array( 'post_status' => $status ) );
}
register_taxonomy( 'wp_test_tax_counts', array( 'post', 'attachment' ) );
self::$attachment_term = $factory->term->create( array( 'taxonomy' => 'wp_test_tax_counts' ) );
}
public function setUp() {
parent::setUp();
register_taxonomy( 'wp_test_tax_counts', array( 'post', 'attachment' ) );
}
/**
* Term counts increments correctly when post status becomes published.
*
* @covers wp_publish_post
* @covers wp_count_terms
* @dataProvider data_term_counts_incremented_on_publish
* @ticket 40351
* @ticket 51292
*
* @param string $original_post_status Post status prior to change to publish.
* @param int $change Expected change upon publish.
*/
public function test_term_counts_incremented_on_publish( $original_post_status, $change ) {
$post_id = self::$post_ids[ $original_post_status ];
$term_count = get_term( get_option( 'default_category' ) )->count;
wp_publish_post( $post_id );
$expected = $term_count + $change;
$this->assertSame( $expected, get_term( get_option( 'default_category' ) )->count );
}
/**
* Data provider for test_term_count_changes_for_post_statuses.
*
* @return array[] {
* @type string $original_post_status Post status prior to change to publish.
* @type int $change Expected change upon publish.
* }
*/
function data_term_counts_incremented_on_publish() {
return array(
// 0. Published post
array( 'publish', 0 ),
// 1. Auto draft
array( 'auto-draft', 1 ),
// 2. Draft
array( 'draft', 1 ),
// 3. Private post
array( 'private', 1 ),
);
}
/**
* Term counts increments correctly when post status becomes published.
*
* @covers wp_publish_post
* @dataProvider data_term_counts_incremented_on_publish_with_attachments
* @ticket 40351
* @ticket 51292
*
* @param string $original_post_status Post status prior to change to publish.
* @param int $change Expected change upon publish.
*/
public function test_term_counts_incremented_on_publish_with_attachments( $original_post_status, $change ) {
$post_id = self::$post_ids[ $original_post_status ];
wp_add_object_terms( $post_id, self::$attachment_term, 'wp_test_tax_counts' );
$attachment_id = self::factory()->attachment->create_object(
array(
'file' => 'image.jpg',
'post_parent' => $post_id,
'post_status' => 'inherit',
)
);
wp_add_object_terms( $attachment_id, self::$attachment_term, 'wp_test_tax_counts' );
$term_count = get_term( self::$attachment_term )->count;
wp_publish_post( $post_id );
$expected = $term_count + $change;
$this->assertSame( $expected, get_term( self::$attachment_term )->count );
}
/**
* Data provider for test_term_count_changes_for_post_statuses_with_attachments.
*
* @return array[] {
* @type string $original_post_status Post status prior to change to publish.
* @type int $change Expected change upon publish.
* }
*/
function data_term_counts_incremented_on_publish_with_attachments() {
return array(
// 0. Published post
array( 'publish', 0 ),
// 1. Auto draft
array( 'auto-draft', 2 ),
// 2. Draft
array( 'draft', 2 ),
// 3. Private post
array( 'private', 2 ),
);
}
/**
* Term counts increments correctly when post status becomes published.
*
* @covers wp_publish_post
* @dataProvider data_term_counts_incremented_on_publish_with_untermed_attachments
* @ticket 40351
* @ticket 51292
*
* @param string $original_post_status Post status prior to change to publish.
* @param int $change Expected change upon publish.
*/
public function test_term_counts_incremented_on_publish_with_untermed_attachments( $original_post_status, $change ) {
$post_id = self::$post_ids[ $original_post_status ];
wp_add_object_terms( $post_id, self::$attachment_term, 'wp_test_tax_counts' );
$attachment_id = self::factory()->attachment->create_object(
array(
'file' => 'image.jpg',
'post_parent' => $post_id,
'post_status' => 'inherit',
)
);
$term_count = get_term( self::$attachment_term )->count;
wp_publish_post( $post_id );
$expected = $term_count + $change;
$this->assertSame( $expected, get_term( self::$attachment_term )->count );
}
/**
* Data provider for test_term_count_changes_for_post_statuses_with_untermed_attachments.
*
* @return array[] {
* @type string $original_post_status Post status prior to change to publish.
* @type int $change Expected change upon publish.
* }
*/
function data_term_counts_incremented_on_publish_with_untermed_attachments() {
return array(
// 0. Published post
array( 'publish', 0 ),
// 1. Auto draft
array( 'auto-draft', 1 ),
// 2. Draft
array( 'draft', 1 ),
// 3. Private post
array( 'private', 1 ),
);
}
}