Changeset 43087 for trunk/tests/phpunit/tests/rest-api/rest-attachments-controller.php

Timestamp:

05/02/2018 01:24:30 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.

Props danielbachhuber.
See #43874.

File:

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

trunk/tests/phpunit/tests/rest-api/rest-attachments-controller.php

@@ -1177 +1177 @@
         $this->check_post_data( $attachment, $data, 'view', $response->get_links() );
         $this->check_post_data( $attachment, $data, 'embed', $response->get_links() );
+    }
+
+    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() ) );
     }