Changeset 32202 for branches/3.7/tests/phpunit/tests/user/capabilities.php

Timestamp:

04/20/2015 01:28:00 PM (10 years ago)

Author:

pento

Message:

In Multisite, prevent plugins from unintentionally switching sites. Merge of [32173] to the 3.7 branch.

Props mdawaffe, pento.

File:

• branches/3.7/tests/phpunit/tests/user/capabilities.php (modified) (1 diff)

branches/3.7/tests/phpunit/tests/user/capabilities.php

@@ -634 +634 @@
         $author->remove_cap( 'foo' );
         $this->assertFalse ( isset( $author->caps['foo'] ) );
+    }
+
+    function test_borked_current_user_can_for_blog() {
+        if ( ! is_multisite() ) {
+            $this->markTestSkipped( 'Test only runs in multisite' );
+            return;
+        }
+
+        $orig_blog_id = get_current_blog_id();
+        $blog_id = $this->factory->blog->create();
+
+        $this->_nullify_current_user();
+
+        add_action( 'switch_blog', array( $this, '_nullify_current_user_and_keep_nullifying_user' ) );
+
+        current_user_can_for_blog( $blog_id, 'edit_posts' );
+
+        $this->assertEquals( $orig_blog_id, get_current_blog_id() );
+    }
+
+    function _nullify_current_user() {
+        // Prevents fatal errors in ::tearDown()'s and other uses of restore_current_blog()
+        $function_stack = wp_debug_backtrace_summary( null, 0, false );
+        if ( in_array( 'restore_current_blog', $function_stack ) ) {
+            return;
+        }
+        $GLOBALS['current_user'] = null;
+    }
+
+    function _nullify_current_user_and_keep_nullifying_user() {
+        add_action( 'set_current_user', array( $this, '_nullify_current_user' ) );
     }