diff --git a/src/wp-includes/functions.php b/src/wp-includes/functions.php index 946853c361..9b999e656b 100644 --- a/src/wp-includes/functions.php +++ b/src/wp-includes/functions.php @@ -6033,7 +6033,7 @@ function wp_guess_url() { // The request is for the admin. if ( strpos( $_SERVER['REQUEST_URI'], 'wp-admin' ) !== false || strpos( $_SERVER['REQUEST_URI'], 'wp-login.php' ) !== false ) { - $path = preg_replace( '#/(wp-admin/.*|wp-login.php)#i', '', $_SERVER['REQUEST_URI'] ); + $path = preg_replace( '#/(wp-admin/?.*|wp-login\.php.*)#i', '', $_SERVER['REQUEST_URI'] ); // The request is for a file in ABSPATH. } elseif ( $script_filename_dir . '/' === $abspath_fix ) { diff --git a/tests/phpunit/tests/functions/wpGuessUrl.php b/tests/phpunit/tests/functions/wpGuessUrl.php new file mode 100644 index 0000000000..fc9f57c718 --- /dev/null +++ b/tests/phpunit/tests/functions/wpGuessUrl.php @@ -0,0 +1,38 @@ +go_to( site_url( $url ) ); + $this->assertSame( $siteurl, wp_guess_url() ); + } + + /** + * Data provider. + * + * @return array + */ + function data_guess_url_should_return_site_url() { + return array( + 'no trailing slash' => array( 'url' => 'wp-admin' ), + 'trailing slash' => array( 'url' => 'wp-admin/' ), + 'trailing slash, query var' => array( 'url' => 'wp-admin/?foo=bar' ), + 'file extension, no trailing slash' => array( 'url' => 'wp-login.php' ), + 'file extension, query var, no trailing slash' => array( 'url' => 'wp-login.php?foo=bar' ), + ); + } +}