Changeset 54784 for branches/6.1/tests/phpunit/tests/query/cacheResults.php

Timestamp:

11/10/2022 02:20:47 AM (4 years ago)

Author:

peterwilsoncc

Message:

Query: Prevent ID only queries erroring when starting the loop.

Ensure only full post objects are passed to update_post_author_caches() when called within WP_Query::the_post(). This prevents an error when starting the Loop for Queries initiated with a subset of fields or IDs only.

Props konyoldeath, dd32, lozula, TimothyBlynJacobs, spacedmonkey, mxbclang, peterwilsoncc.
Merges [54771] to the 6.1 branch.
Fixes #56948.

Location:

branches/6.1

Files:

• . (modified) (1 prop)
• tests/phpunit/tests/query/cacheResults.php (modified) (1 diff)

branches/6.1/tests/phpunit/tests/query/cacheResults.php

@@ -1211 +1211 @@
                 );
         }
+
+        /**
+         * Ensure starting the loop warms the author cache.
+         *
+         * @since 6.1.1
+         * @ticket 56948
+         *
+         * @covers WP_Query::the_post
+         *
+         * @dataProvider data_author_cache_warmed_by_the_loop
+         *
+         * @param string $fields Query fields.
+         */
+        public function test_author_cache_warmed_by_the_loop( $fields ) {
+                // Update post author for the parent post.
+                self::factory()->post->update_object( self::$pages[0], array( 'post_author' => self::$author_id ) );
+
+                self::factory()->post->create(
+                        array(
+                                'post_author' => self::$author_id,
+                                'post_parent' => self::$pages[0],
+                                'post_type'   => 'page',
+                        )
+                );
+
+                $query_1 = new WP_Query(
+                        array(
+                                'post_type' => 'page',
+                                'fields'    => $fields,
+                                'author'    => self::$author_id,
+                        )
+                );
+
+                // Start the loop.
+                $start_loop_queries = get_num_queries();
+                $query_1->the_post();
+                $num_loop_queries = get_num_queries() - $start_loop_queries;
+                $this->assertSame( 2, $num_loop_queries, 'Unexpected number of queries while initializing the loop.' );
+
+                $start_author_queries = get_num_queries();
+                get_user_by( 'ID', self::$author_id );
+                $num_author_queries = get_num_queries() - $start_author_queries;
+                $this->assertSame( 0, $num_author_queries, 'Author cache is not warmed by the loop.' );
+        }
+
+        /**
+         * Data provider for test_author_cache_warmed_by_the_loop
+         *
+         * @return array[]
+         */
+        public function data_author_cache_warmed_by_the_loop() {
+                return array(
+                        'fields: empty' => array( '' ),
+                        'fields: all'   => array( 'all' ),
+                        'fields: ids'   => array( 'ids' ),
+                        /*
+                         * `id=>parent` is untested pending the resolution of an existing bug.
+                         * See https://core.trac.wordpress.org/ticket/56992
+                         */
+                );
+        }
 }