Changeset 43445 for branches/4.9/tests/phpunit/tests/rest-api/rest-attachments-controller.php

Timestamp:

07/13/2018 06:50:51 AM (7 years ago)

Author:

pento

Message:

REST API: Filter responses based on the _fields parameter, before data is processed.

Historically, the REST API would generate the entire response object, including running expensive filters, then it would apply the _fields parameter, discarding the fields that weren't specificed.

This change causes _fields to be applied earlier, so that only requested fields are processed.

Merges [43087] to the 4.9 branch.

Props danielbachhuber.
See #43874.

Location:

branches/4.9

Files:

• . (modified) (1 prop)
• tests/phpunit/tests/rest-api/rest-attachments-controller.php (modified) (1 diff)

branches/4.9/tests/phpunit/tests/rest-api/rest-attachments-controller.php

@@ -1075 +1075 @@
     }
 
+    public function test_prepare_item_limit_fields() {
+        $attachment_id = $this->factory->attachment->create_object(
+            $this->test_file, 0, array(
+                'post_mime_type' => 'image/jpeg',
+                'post_excerpt'   => 'A sample caption',
+                'post_author'    => self::$editor_id,
+            )
+        );
+        wp_set_current_user( self::$editor_id );
+        $endpoint = new WP_REST_Attachments_Controller( 'post' );
+        $request  = new WP_REST_Request( 'GET', sprintf( '/wp/v2/media/%d', $attachment_id ) );
+        $request->set_param( 'context', 'edit' );
+        $request->set_param( '_fields', 'id,slug' );
+        $obj      = get_post( $attachment_id );
+        $response = $endpoint->prepare_item_for_response( $obj, $request );
+        $this->assertEquals( array(
+            'id',
+            'slug',
+        ), array_keys( $response->get_data() ) );
+    }
+
     public function test_get_item_schema() {
         $request = new WP_REST_Request( 'OPTIONS', '/wp/v2/media' );