Changeset 35483 for trunk/tests/phpunit/tests/customize/manager.php

Timestamp:

11/01/2015 06:39:50 AM (9 years ago)

Author:

westonruter

Message:

Customize: Return user to referring URL when leaving Customizer in absence of return query param.

When referring URL is not available, default returning user to frontend URL instead of admin URL. Themes page is updated to include the return path in Customizer links.

Props McGuive7, westonruter.
Fixes #32637.

File:

• trunk/tests/phpunit/tests/customize/manager.php (modified) (1 diff)

trunk/tests/phpunit/tests/customize/manager.php

@@ -253 +253 @@
     function test_return_url() {
         wp_set_current_user( self::factory()->user->create( array( 'role' => 'author' ) ) );
-        $this->assertEquals( get_admin_url(), $this->manager->get_return_url() );
+        $this->assertEquals( home_url( '/' ), $this->manager->get_return_url() );
 
         wp_set_current_user( self::factory()->user->create( array( 'role' => 'administrator' ) ) );
         $this->assertTrue( current_user_can( 'edit_theme_options' ) );
-        $this->assertEquals( admin_url( 'themes.php' ), $this->manager->get_return_url() );
+        $this->assertEquals( home_url( '/' ), $this->manager->get_return_url() );
 
         $preview_url = home_url( '/foo/' );
         $this->manager->set_preview_url( $preview_url );
+        $this->assertEquals( $preview_url, $this->manager->get_return_url() );
+
+        $url = home_url( '/referred/' );
+        $_SERVER['HTTP_REFERER'] = wp_slash( $url );
+        $this->assertEquals( $url, $this->manager->get_return_url() );
+
+        $url = 'http://badreferer.example.com/';
+        $_SERVER['HTTP_REFERER'] = wp_slash( $url );
+        $this->assertNotEquals( $url, $this->manager->get_return_url() );
         $this->assertEquals( $preview_url, $this->manager->get_return_url() );