Changeset 47138 for trunk/tests/phpunit/tests/rest-api/rest-server.php

Timestamp:

01/30/2020 08:20:30 PM (7 years ago)

Author:

kadamwhite

Message:

REST API: Reuse previously-generated embedded objects when building collection response.

Store each generated embedded object in a temporary cache when querying for linked resources so that repeated links to the same resource do not trigger repeated queries or processing.

Props TimothyBlynJacobs.
Fixes #48838.

File:

• trunk/tests/phpunit/tests/rest-api/rest-server.php (modified) (1 diff)

trunk/tests/phpunit/tests/rest-api/rest-server.php

@@ -676 +676 @@
                 $this->assertEquals( 'Test message', $up_data['message'] );
                 $this->assertEquals( 403, $up_data['data']['status'] );
+        }
+
+        /**
+         * @ticket 48838
+         */
+        public function test_link_embedding_clears_cache() {
+                $post_id = self::factory()->post->create();
+
+                $response = new WP_REST_Response();
+                $response->add_link( 'post', rest_url( 'wp/v2/posts/' . $post_id ), array( 'embeddable' => true ) );
+
+                $data = rest_get_server()->response_to_data( $response, true );
+                $this->assertArrayHasKey( 'post', $data['_embedded'] );
+                $this->assertCount( 1, $data['_embedded']['post'] );
+
+                wp_update_post(
+                        array(
+                                'ID'         => $post_id,
+                                'post_title' => 'My Awesome Title',
+                        )
+                );
+
+                $data = rest_get_server()->response_to_data( $response, true );
+                $this->assertArrayHasKey( 'post', $data['_embedded'] );
+                $this->assertCount( 1, $data['_embedded']['post'] );
+                $this->assertEquals( 'My Awesome Title', $data['_embedded']['post'][0]['title']['rendered'] );
+        }
+
+        /**
+         * @ticket 48838
+         */
+        public function test_link_embedding_cache() {
+                $response = new WP_REST_Response(
+                        array(
+                                'id' => 1,
+                        )
+                );
+                $response->add_link(
+                        'author',
+                        rest_url( 'wp/v2/users/1' ),
+                        array( 'embeddable' => true )
+                );
+                $response->add_link(
+                        'author',
+                        rest_url( 'wp/v2/users/1' ),
+                        array( 'embeddable' => true )
+                );
+
+                $mock = new MockAction();
+                add_filter( 'rest_post_dispatch', array( $mock, 'filter' ) );
+
+                $data = rest_get_server()->response_to_data( $response, true );
+
+                $this->assertArrayHasKey( '_embedded', $data );
+                $this->assertArrayHasKey( 'author', $data['_embedded'] );
+                $this->assertCount( 2, $data['_embedded']['author'] );
+
+                $this->assertCount( 1, $mock->get_events() );
+        }
+
+        /**
+         * @ticket 48838
+         */
+        public function test_link_embedding_cache_collection() {
+                $response = new WP_REST_Response(
+                        array(
+                                array(
+                                        'id'     => 1,
+                                        '_links' => array(
+                                                'author' => array(
+                                                        array(
+                                                                'href'       => rest_url( 'wp/v2/users/1' ),
+                                                                'embeddable' => true,
+                                                        ),
+                                                ),
+                                        ),
+                                ),
+                                array(
+                                        'id'     => 2,
+                                        '_links' => array(
+                                                'author' => array(
+                                                        array(
+                                                                'href'       => rest_url( 'wp/v2/users/1' ),
+                                                                'embeddable' => true,
+                                                        ),
+                                                ),
+                                        ),
+                                ),
+                        )
+                );
+
+                $mock = new MockAction();
+                add_filter( 'rest_post_dispatch', array( $mock, 'filter' ) );
+
+                $data = rest_get_server()->response_to_data( $response, true );
+
+                $embeds = wp_list_pluck( $data, '_embedded' );
+                $this->assertCount( 2, $embeds );
+                $this->assertArrayHasKey( 'author', $embeds[0] );
+                $this->assertArrayHasKey( 'author', $embeds[1] );
+
+                $this->assertCount( 1, $mock->get_events() );
         }