Ticket #21459: 21459.testcase.ms-switch-speed.php

<?php

if ( is_multisite() ) :

/**
 * A set of unit tests for WordPress Multisite
 *
 * @group multisite
 */
class Tests_MS_Switch_Speed extends WP_UnitTestCase {

        protected $num_of_sites = 100;

        function test_speed_new() {
                global $wpdb, $current_site;

                $user1_id = $this->factory->user->create( array( 'role' => 'administrator' ) );
                $blog_ids = array_merge(
                        $this->factory->blog->create_many( $this->num_of_sites, array( 'user_id' => $user1_id ) ),
                        $this->factory->blog->create_many( $this->num_of_sites )
                );
                wp_set_current_user( $user1_id );

                $start = microtime(true);
                foreach ( $blog_ids as $blog_id ) {
                        switch_to_blog( $blog_id );
                        if( current_user_can( 'upload_files' ) ) $sites[] = get_option( 'blogname' );
                        restore_current_blog();
                }
                echo "\n\n" . __FUNCTION__ . " duration: " . ( microtime(true) - $start ) . " (site-count: " . count( $sites ) . " of " . count( $blog_ids ) . ")\n\n";
                
                // cleanup
                foreach ( $blog_ids as $blog_id ) {
                        wpmu_delete_blog( $blog_id, true );
                }
        }

        function test_speed_old() {
                global $wpdb, $current_site;

                $user1_id = $this->factory->user->create( array( 'role' => 'administrator' ) );
                $blog_ids = array_merge(
                        $this->factory->blog->create_many( $this->num_of_sites, array( 'user_id' => $user1_id ) ),
                        $this->factory->blog->create_many( $this->num_of_sites )
                );
                wp_set_current_user( $user1_id );

                $start = microtime(true);
                $sites = array();
                foreach ( $blog_ids as $blog_id ) {
                        if( old_current_user_can_for_blog( $blog_id , 'upload_files') ) $sites[] = get_blog_option( $blog_id, 'blogname' );
                }
                echo "\n\n" . __FUNCTION__ . " duration: " . ( microtime(true) - $start ) . " (site-count: " . count( $sites ) . " of " . count( $blog_ids ) . ")\n\n";
                

                // cleanup
                foreach ( $blog_ids as $blog_id ) {
                        wpmu_delete_blog( $blog_id, true );
                }
        }
}
/**
 * Original 3.4 version
 * 
 * Whether current user has a capability or role for a given blog.
 *
 * @since 3.0.0
 *
 * @param int $blog_id Blog ID
 * @param string $capability Capability or role name.
 * @return bool
 */
function old_current_user_can_for_blog( $blog_id, $capability ) {
        $current_user = wp_get_current_user();

        if ( empty( $current_user ) )
                return false;

        // Create new object to avoid stomping the global current_user.
        $user = new WP_User( $current_user->ID) ;

        // Set the blog id. @todo add blog id arg to WP_User constructor?
        $user->for_blog( $blog_id );

        $args = array_slice( func_get_args(), 2 );
        $args = array_merge( array( $capability ), $args );

        return call_user_func_array( array( &$user, 'has_cap' ), $args );
}
endif;