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

Timestamp:

12/04/2015 11:35:54 PM (9 years ago)

Author:

wonderboymusic

Message:

REST API: Core typically sends nocache headers on all auth'ed responses, as in wp, admin-ajax, etc. Because the REST API infrastructure is hooked in pre-wp, we should be setting this ourselves.

Adds unit tests.

Props joehoyle.
Fixes #34832.

File:

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

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

@@ -620 +620 @@
         $this->assertContains( 'test/another', $namespaces );
     }
+
+    public function test_nocache_headers_on_authenticated_requests() {
+        $editor = self::factory()->user->create( array( 'role' => 'editor' ) );
+        $request = new WP_REST_Request( 'GET', '/', array() );
+        wp_set_current_user( $editor );
+
+        $result = $this->server->serve_request('/');
+        $headers = $this->server->sent_headers;
+
+        foreach ( wp_get_nocache_headers() as $header => $value ) {
+            $this->assertTrue( isset( $headers[ $header ] ), sprintf( 'Header %s is not present in the response.', $header ) );
+            $this->assertEquals( $value, $headers[ $header ] );
+        }
+    }
+
+    public function test_no_nocache_headers_on_unauthenticated_requests() {
+        $editor = self::factory()->user->create( array( 'role' => 'editor' ) );
+        $request = new WP_REST_Request( 'GET', '/', array() );
+
+        $result = $this->server->serve_request('/');
+        $headers = $this->server->sent_headers;
+
+        foreach ( wp_get_nocache_headers() as $header => $value ) {
+            $this->assertFalse( isset( $headers[ $header ] ) && $headers[ $header ] === $value, sprintf( 'Header %s is set to nocache.', $header ) );
+        }
+    }
 }