Changeset 32173 for trunk/tests/phpunit/tests/user/capabilities.php

Timestamp:

04/20/2015 07:26:05 AM (11 years ago)

Author:

pento

Message:

In Multisite, prevent plugins from unintentionally switching sites.

Props mdawaffe.

File:

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

trunk/tests/phpunit/tests/user/capabilities.php

@@ -696 +696 @@
     }
 
+    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();
+
+        $nullify_current_user = function() {
+            // 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;
+        };
+
+        $nullify_current_user_and_keep_nullifying_user = function() use ( $nullify_current_user ) {
+            $nullify_current_user();
+
+            add_action( 'set_current_user', $nullify_current_user );
+        };
+
+        $nullify_current_user();
+
+        add_action( 'switch_blog', $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() );
+    }
+
     /**
      * @ticket 28374