Changeset 36451 for branches/4.0/tests/phpunit/tests/formatting/redirect.php

Timestamp:

02/02/2016 04:59:23 PM (11 years ago)

Author:

ocean90

Message:

Better validation of the URL used in HTTP redirects.

Merges [36444] to the 4.0 branch.

File:

• branches/4.0/tests/phpunit/tests/formatting/redirect.php (modified) (2 diffs)

branches/4.0/tests/phpunit/tests/formatting/redirect.php

@@ -4 +4 @@
  * @group pluggable
  * @group formatting
+ * @group redirect
  */
 class Tests_Formatting_Redirect extends WP_UnitTestCase {
+        function setUp() {
+                add_filter( 'home_url', array( $this, 'home_url' ) );
+        }
+
+        function tearDown() {
+                remove_filter( 'home_url', array( $this, 'home_url' ) );
+        }
+
+        function home_url() {
+                return 'http://example.com/';
+        }
+
         function test_wp_sanitize_redirect() {
                 $this->assertEquals('http://example.com/watchthelinefeedgo', wp_sanitize_redirect('http://example.com/watchthelinefeed%0Ago'));
@@ -16 +29 @@
                 $this->assertEquals('http://example.com/watchthecarriagereturngo', wp_sanitize_redirect('http://example.com/watchthecarriagereturn%0%0DDgo'));
         }
+
+        /**
+         * @dataProvider valid_url_provider
+         */
+        function test_wp_validate_redirect_valid_url( $url, $expected ) {
+                $this->assertEquals( $expected, wp_validate_redirect( $url ) );
+        }
+
+        /**
+         * @dataProvider invalid_url_provider
+         */
+        function test_wp_validate_redirect_invalid_url( $url ) {
+                $this->assertEquals( false, wp_validate_redirect( $url, false ) );
+        }
+
+        function valid_url_provider() {
+                return array(
+                        array( 'http://example.com', 'http://example.com' ),
+                        array( 'http://example.com/', 'http://example.com/' ),
+                        array( 'https://example.com/', 'https://example.com/' ),
+                        array( '//example.com', 'http://example.com' ),
+                        array( '//example.com/', 'http://example.com/' ),
+                        array( 'http://example.com/?foo=http://example.com/', 'http://example.com/?foo=http://example.com/' ),
+                        array( 'http://user@example.com/', 'http://user@example.com/' ),
+                        array( 'http://user:@example.com/', 'http://user:@example.com/' ),
+                        array( 'http://user:pass@example.com/', 'http://user:pass@example.com/' ),
+                );
+        }
+
+        function invalid_url_provider() {
+                return array(
+                        // parse_url() fails
+                        array( '' ),
+                        array( 'http://:' ),
+
+                        // non-safelisted domain
+                        array( 'http://non-safelisted.example/' ),
+
+                        // unsupported schemes
+                        array( 'data:text/plain;charset=utf-8,Hello%20World!' ),
+                        array( 'file:///etc/passwd' ),
+                        array( 'ftp://example.com/' ),
+
+                        // malformed input
+                        array( 'http:example.com' ),
+                        array( 'http:80' ),
+                        array( 'http://example.com:1234:5678/' ),
+                        array( 'http://user:pa:ss@example.com/' ),
+
+                        array( 'http://user@@example.com' ),
+                        array( 'http://user@:example.com' ),
+                        array( 'http://user?@example.com' ),
+                        array( 'http://user@?example.com' ),
+                        array( 'http://user#@example.com' ),
+                        array( 'http://user@#example.com' ),
+
+                        array( 'http://user@@example.com/' ),
+                        array( 'http://user@:example.com/' ),
+                        array( 'http://user?@example.com/' ),
+                        array( 'http://user@?example.com/' ),
+                        array( 'http://user#@example.com/' ),
+                        array( 'http://user@#example.com/' ),
+
+                        array( 'http://user:pass@@example.com' ),
+                        array( 'http://user:pass@:example.com' ),
+                        array( 'http://user:pass?@example.com' ),
+                        array( 'http://user:pass@?example.com' ),
+                        array( 'http://user:pass#@example.com' ),
+                        array( 'http://user:pass@#example.com' ),
+
+                        array( 'http://user:pass@@example.com/' ),
+                        array( 'http://user:pass@:example.com/' ),
+                        array( 'http://user:pass?@example.com/' ),
+                        array( 'http://user:pass@?example.com/' ),
+                        array( 'http://user:pass#@example.com/' ),
+                        array( 'http://user:pass@#example.com/' ),
+
+                        array( 'http://user.pass@@example.com' ),
+                        array( 'http://user.pass@:example.com' ),
+                        array( 'http://user.pass?@example.com' ),
+                        array( 'http://user.pass@?example.com' ),
+                        array( 'http://user.pass#@example.com' ),
+                        array( 'http://user.pass@#example.com' ),
+
+                        array( 'http://user.pass@@example.com/' ),
+                        array( 'http://user.pass@:example.com/' ),
+                        array( 'http://user.pass?@example.com/' ),
+                        array( 'http://user.pass@?example.com/' ),
+                        array( 'http://user.pass#@example.com/' ),
+                        array( 'http://user.pass@#example.com/' ),
+                );
+        }
 }