Changeset 42773 for trunk/tests/phpunit/tests/admin/includesFile.php

Timestamp:

03/04/2018 05:13:35 PM (8 years ago)

Author:

SergeyBiryukov

Message:

Filesystem API: Allow download_url() to return the response code and body on error as an additional WP_Error object data.

The error response body size is limited to 1 KB by default to avoid taking up too much memory. The size can be increased using download_url_error_max_body_size filter.

Props soulseekah, campusboy1987, mihdan, SergeyBiryukov.
Fixes #43329.

File:

• trunk/tests/phpunit/tests/admin/includesFile.php (modified) (1 diff)

trunk/tests/phpunit/tests/admin/includesFile.php

@@ -32 +32 @@
                 $_SERVER['SCRIPT_FILENAME'] = $sfn;
         }
+
+        /**
+         * @ticket 43329
+         */
+        public function test_download_url_non_200_response_code() {
+                add_filter( 'pre_http_request', array( $this, '_fake_download_url_non_200_response_code' ), 10, 3 );
+
+                $error = download_url( 'test_download_url_non_200' );
+                $this->assertWPError( $error );
+                $this->assertEquals( array(
+                        'code' => 418,
+                        'body' => 'This is an unexpected error message from your favorite server.',
+                ), $error->get_error_data() );
+
+                add_filter( 'download_url_error_max_body_size', array( $this, '__return_5' ) );
+
+                $error = download_url( 'test_download_url_non_200' );
+                $this->assertWPError( $error );
+                $this->assertEquals( array(
+                        'code' => 418,
+                        'body' => 'This ',
+                ), $error->get_error_data() );
+
+                remove_filter( 'download_url_error_max_body_size', array( $this, '__return_5' ) );
+                remove_filter( 'pre_http_request', array( $this, '_fake_download_url_non_200_response_code' ) );
+        }
+
+        public function _fake_download_url_non_200_response_code( $response, $args, $url ) {
+                file_put_contents( $args['filename'], 'This is an unexpected error message from your favorite server.' );
+                return array(
+                        'response' => array(
+                                'code'    => 418,
+                                'message' => "I'm a teapot!",
+                        ),
+                );
+        }
+
+        public function __return_5() {
+                return 5;
+        }
 }