Ticket #49648: 49648.1.diff

src/wp-includes/rest-api/endpoints/class-wp-rest-controller.php

@@ -449 +449 @@
                                 continue;
                         }
 
-                        if ( ! in_array( $field_name, $requested_fields, true ) ) {
+                        if ( ! rest_is_field_included( $field_name, $requested_fields ) ) {
                                 continue;
                         }
 

tests/phpunit/tests/rest-api.php

@@ -560 +560 @@
                 );
         }
 
+        /**
+         * Ensure inclusion of deeply nested fields may be controlled with request['_fields'].
+         *
+         * @ticket 49648
+         */
+        public function test_rest_filter_response_fields_deeply_nested_field_filter() {
+                $response = new WP_REST_Response();
+
+                $response->set_data(
+                        array(
+                                'field' => array(
+                                        'a' => array(
+                                                'i' => 'value i',
+                                                'ii' => 'value ii',
+                                        ),
+                                        'b' => array(
+                                                'iii' => 'value iii',
+                                                'iv' => 'value iv',
+                                        ),
+                                ),
+                        )
+                );
+                $request = array(
+                        '_fields' => 'field.a.i,field.b.iv',
+                );
+
+                $response = rest_filter_response_fields( $response, null, $request );
+                $this->assertEquals(
+                        array(
+                                'field' => array(
+                                        'a' => array(
+                                                'i' => 'value i',
+                                        ),
+                                        'b' => array(
+                                                'iv' => 'value iv',
+                                        ),
+                                ),
+                        ),
+                        $response->get_data()
+                );
+        }
+
         /**
          * Ensure that specifying a single top-level key in _fields includes that field and all children.
          *

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

@@ -288 +288 @@
                 $controller->prepare_item_for_response( array(), $request );
 
                 $this->assertGreaterThan( 0, $listener->get_call_count( $method ) );
+
+                global $wp_rest_additional_fields;
+                $wp_rest_additional_fields = array();
         }
 
         public function test_filtering_fields_for_response_by_context_returns_fields_with_no_context() {
@@ -316 +319 @@
                 $controller->prepare_item_for_response( array(), $request );
 
                 $this->assertGreaterThan( 0, $listener->get_call_count( $method ) );
+
+                global $wp_rest_additional_fields;
+                $wp_rest_additional_fields = array();
         }
 
         public function test_filtering_fields_for_response_by_context_returns_fields_with_no_schema() {
@@ -341 +347 @@
                 $controller->prepare_item_for_response( array(), $request );
 
                 $this->assertGreaterThan( 0, $listener->get_call_count( $method ) );
+
+                global $wp_rest_additional_fields;
+                $wp_rest_additional_fields = array();
         }
 
         /**
@@ -415 +424 @@
                 $controller->prepare_item_for_response( $item, $request );
 
                 $this->assertTrue( $listener->get_call_count( $method ) > $first_call_count );
+
+                global $wp_rest_additional_fields;
+                $wp_rest_additional_fields = array();
+        }
+
+        /**
+         * @dataProvider data_filter_registered_rest_fields
+         * @ticket 49648
+         */
+        public function test_filter_nested_registered_rest_fields( $filter, $expected ) {
+                $controller = new WP_REST_Test_Controller();
+
+                register_rest_field(
+                        'type',
+                        'field',
+                        array(
+                                'schema' => array(
+                                        'type'        => 'object',
+                                        'description' => 'A complex object',
+                                        'context'     => array( 'view', 'edit' ),
+                                        'properties'  => array(
+                                                'a' => array(
+                                                        'i'  => 'string',
+                                                        'ii' => 'string',
+                                                ),
+                                                'b' => array(
+                                                        'iii' => 'string',
+                                                        'iv'  => 'string',
+                                                ),
+                                        ),
+                                ),
+                                'get_callback' => array( $this, 'register_nested_rest_field_get_callback' ),
+                        ),
+                );
+
+                $request = new WP_REST_Request( 'GET', '/wp/v2/testroute' );
+                $request->set_param( '_fields', $filter );
+
+                $response = $controller->prepare_item_for_response( array(), $request );
+                $response = rest_filter_response_fields( $response, rest_get_server(), $request );
+
+                $this->assertEquals( $expected, $response->get_data() );
+
+                global $wp_rest_additional_fields;
+                $wp_rest_additional_fields = array();
+        }
+
+        public function register_nested_rest_field_get_callback() {
+                return array(
+                        'a' => array(
+                                'i' => 'value i',
+                                'ii' => 'value ii',
+                        ),
+                        'b' => array(
+                                'iii' => 'value iii',
+                                'iv' => 'value iv',
+                        ),
+                );
+        }
+
+        public function data_filter_nested_registered_rest_fields() {
+                return array(
+                        array(
+                                'field',
+                                array(
+                                        'field' => array(
+                                                'a' => array(
+                                                        'i' => 'value i',
+                                                        'ii' => 'value ii',
+                                                ),
+                                                'b' => array(
+                                                        'iii' => 'value iii',
+                                                        'iv' => 'value iv',
+                                                ),
+                                        ),
+                                ),
+                        ),
+                        array(
+                                'field.a',
+                                array(
+                                        'field' => array(
+                                                'a' => array(
+                                                        'i' => 'value i',
+                                                        'ii' => 'value ii',
+                                                ),
+                                        ),
+                                ),
+                        ),
+                        array(
+                                'field.b',
+                                array(
+                                        'field' => array(
+                                                'b' => array(
+                                                        'iii' => 'value iii',
+                                                        'iv' => 'value iv',
+                                                ),
+                                        ),
+                                ),
+                        ),
+                        array(
+                                'field.a.i,field.b.iv',
+                                array(
+                                        'field' => array(
+                                                'a' => array(
+                                                        'i' => 'value i',
+                                                ),
+                                                'b' => array(
+                                                        'iv' => 'value iv',
+                                                ),
+                                        ),
+                                ),
+                        ),
+                        array(
+                                'field.a,field.b.iii',
+                                array(
+                                        'field' => array(
+                                                'a' => array(
+                                                        'i' => 'value i',
+                                                        'ii' => 'value ii',
+                                                ),
+                                                'b' => array(
+                                                        'iii' => 'value iii',
+                                                ),
+                                        ),
+                                ),
+                        ),
+                );
         }
 }