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

Timestamp:

03/26/2020 05:50:39 PM (6 years ago)

Author:

kadamwhite

Message:

REST API: Fix _fields filtering of registered rest fields.

Use rest_is_field_included when determining which additional fields to include to permit filtering by nested field properties.

Props Dudo, kadamwhite, TimothyBlynJacobs.
Fixes #49648.

File:

• trunk/tests/phpunit/tests/rest-api/rest-controller.php (modified) (2 diffs)

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

@@ -47 +47 @@
                         )
                 );
+        }
+
+        public function tearDown() {
+                parent::tearDown();
+
+                global $wp_rest_additional_fields;
+                $wp_rest_additional_fields = array();
         }
 
@@ -438 +445 @@
                 $this->assertTrue( $listener->get_call_count( $method ) > $first_call_count );
         }
+
+        /**
+         * @dataProvider data_filter_nested_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() );
+        }
+
+        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',
+                                                ),
+                                        ),
+                                ),
+                        ),
+                );
+        }
 }