Ticket #38071: countSites.php

<?php

if ( is_multisite() ) :
/**
 * Test wp_count_sites() in multisite.
 *
 * @group ms-site
 * @group multisite
 * @ticket 38071
 */
class Tests_Multisite_Count_Sites extends WP_UnitTestCase {
        protected static $site_ids;
        protected static $expected;

        protected static $different_network_id;
        protected static $different_site_ids;
        protected static $different_expected;

        public static function wpSetUpBeforeClass( $factory ) {
                self::$site_ids = array(
                        'wordpress.org/'         => array( 'domain' => 'wordpress.org', 'path' => '/'        ,
                                                                                                'meta' => array( 'public' => 1, 'archived' => 0, 'mature' => 0, 'spam' => 0, 'deleted' => 0 ) ),
                        'wordpress.org/foo/'     => array( 'domain' => 'wordpress.org', 'path' => '/foo/'    ,
                                                                                                'meta' => array( 'public' => 0, 'archived' => 0, 'mature' => 0, 'spam' => 0, 'deleted' => 0 ) ),
                        'wordpress.org/foo/bar/' => array( 'domain' => 'wordpress.org', 'path' => '/foo/bar/',
                                                                                                'meta' => array( 'public' => 1, 'archived' => 1, 'mature' => 1, 'spam' => 0, 'deleted' => 0 ) ),
                        'wordpress.org/bif/'     => array( 'domain' => 'wordpress.org', 'path' => '/bif/'    ,
                                                                                                'meta' => array( 'public' => 1, 'archived' => 0, 'mature' => 1, 'spam' => 1, 'deleted' => 0 ) ),
                        'wordpress.org/baz/'     => array( 'domain' => 'wordpress.org', 'path' => '/baz/'    ,
                                                                                                'meta' => array( 'public' => 1, 'archived' => 0, 'mature' => 0, 'spam' => 1, 'deleted' => 1 ) ),
                        'wordpress.org/boo/'     => array( 'domain' => 'wordpress.org', 'path' => '/boo/'    ,
                                                                                                'meta' => array( 'public' => 1, 'archived' => 1, 'mature' => 1, 'spam' => 1, 'deleted' => 1 ) ),
                );

                foreach ( self::$site_ids as &$id ) {
                        $id = $factory->blog->create( $id );
                }
                unset( $id );

                self::$expected = (object) array (
                        'public' => 6, // +1 because of example.org/ site
                        'archived' => 2,
                        'mature' => 3,
                        'spam' => 3,
                        'deleted' => 2,
                        );

                self::$different_network_id = $factory->network->create( array( 'domain' => 'diff_wordpress.org', 'path' => '/' ) );

                self::$different_site_ids = array(
                        'diff.wordpress.org/'         => array( 'domain' => 'diff.wordpress.org', 'path' => '/'        , 'site_id' => self::$different_network_id,
                                                                                                'meta' => array( 'public' => 1, 'archived' => 0, 'mature' => 0, 'spam' => 0, 'deleted' => 0 ) ),
                        'diff.wordpress.org/foo/'     => array( 'domain' => 'diff.wordpress.org', 'path' => '/foo/'    , 'site_id' => self::$different_network_id,
                                                                                                'meta' => array( 'public' => 1, 'archived' => 1, 'mature' => 0, 'spam' => 0, 'deleted' => 0 ) ),
                        'diff.wordpress.org/foo/bar/' => array( 'domain' => 'diff.wordpress.org', 'path' => '/foo/bar/', 'site_id' => self::$different_network_id,
                                                                                                'meta' => array( 'public' => 0, 'archived' => 0, 'mature' => 1, 'spam' => 0, 'deleted' => 0 ) ),
                        'diff.wordpress.org/bif/'     => array( 'domain' => 'diff.wordpress.org', 'path' => '/bif/'    , 'site_id' => self::$different_network_id,
                                                                                                'meta' => array( 'public' => 1, 'archived' => 0, 'mature' => 1, 'spam' => 0, 'deleted' => 1 ) ),
                        'diff.wordpress.org/baz/'     => array( 'domain' => 'diff.wordpress.org', 'path' => '/baz/'    , 'site_id' => self::$different_network_id,
                                                                                                'meta' => array( 'public' => 1, 'archived' => 0, 'mature' => 0, 'spam' => 1, 'deleted' => 1 ) ),
                        'diff.wordpress.org/boo/'     => array( 'domain' => 'diff.wordpress.org', 'path' => '/boo/'    , 'site_id' => self::$different_network_id,
                                                                                                'meta' => array( 'public' => 1, 'archived' => 1, 'mature' => 0, 'spam' => 0, 'deleted' => 0 ) ),
                );

                foreach ( self::$different_site_ids as &$id ) {
                        $id = $factory->blog->create( $id );
                }
                unset( $id );

                self::$different_expected = (object) array (
                        'public' => 5,
                        'archived' => 2,
                        'mature' => 2,
                        'spam' => 1,
                        'deleted' => 2,
                        );
        }

        public static function wpTearDownAfterClass() {
                // just in case
                self::switch_network( 1 );

                foreach( self::$site_ids as $id ) {
                        wpmu_delete_blog( $id, true );
                }

                foreach( self::$different_site_ids as $id ) {
                        wpmu_delete_blog( $id, true );
                }

                self::delete_network ( self::$different_network_id );

                wp_update_network_site_counts();
        }

        public function test_count_sites() {
                $counts = wp_count_sites();

                $this->assertEquals( self::$expected, $counts );
        }

        public function test_filtered_count_sites() {
                add_filter( 'wp_count_sites', array( __CLASS__, 'modify_counts' ), 10, 2 );

                $counts = wp_count_sites();

                $this->assertEquals( self::modify_counts( self::$expected, get_current_network_id() ), $counts );
        }

        public function test_different_network_count_sites() {
                $counts = wp_count_sites( self::$different_network_id );

                $this->assertEquals( self::$different_expected, $counts );
        }

        public function test_filtered_different_network_count_sites() {
                add_filter( 'wp_count_sites', array( __CLASS__, 'modify_counts' ), 10, 2 );

                $counts = wp_count_sites( self::$different_network_id );

                $this->assertEquals( self::modify_counts( self::$different_expected, self::$different_network_id ), $counts );
        }

        public function test_switched_network_count_sites() {
                self::switch_network( self::$different_network_id );

                $counts = wp_count_sites();

                $this->assertEquals( self::$different_expected, $counts );

                self::switch_network( 1 );
        }

        public function test_filtered_switched_network_count_sites() {
                self::switch_network( self::$different_network_id );

                add_filter( 'wp_count_sites', array( __CLASS__, 'modify_counts' ), 10, 2 );

                $counts = wp_count_sites();

                $this->assertEquals( self::modify_counts( self::$different_expected, self::$different_network_id ), $counts );

                self::switch_network( 1 );
        }

        /**
         * Filter the site counts
         *
         *      if counts are for the default network then subtract 1 from all counts
         *      if counts are for a different network then add 1 to all counts
         *
         * @param array $counts
         * @param int $network_id
         * @return array
         *
         * @filter wp_count_sites
         */
        public static function modify_counts( $counts, $network_id ) {
                $diff = ( 1 === $network_id ? -1 : +1 );

                $mod_counts = clone $counts ;
                foreach ( get_object_vars( $counts ) as $status => $count ) {
                        $mod_counts->{$status} = max ( $count + $diff, 0 );
                }

                return $mod_counts;
        }
        
        /*
         * I've never worked with multi-network setups before and was shocked
         * to find that there is no real API support in core for multi-networks
         * (@link https://core.trac.wordpress.org/ticket/29411).
         *
         * The functions below appear to be the BARE MINIMUM multi-network support
         * necessary for the tests in this file to run correctly and leave the
         * environment "clean" after they run.
         */

        /**
         * minimalist network switching
         *
         * @param int $network_id
         */
        public static function switch_network( $network_id ) {
                global $current_site, $wpdb;

                // so that get_current_network_id() will do the right thing
                $current_site->id = $network_id;
                
                // so that wp_update_network_site_counts() will do the right thing
                $wpdb->siteid = $network_id;
        }

        /**
         * minimalist network deletion
         *
         * @param int $network_id
         */
        public static function delete_network( $network_id ) {
                global $wpdb;

                $sql  = "DELETE FROM {$wpdb->site} WHERE id = %d";
                $sql = $wpdb->prepare( $sql, $network_id );
                $wpdb->query( $sql );

                $sql  = "DELETE FROM {$wpdb->sitemeta} WHERE site_id = %d";
                $sql = $wpdb->prepare( $sql, $network_id );
                $wpdb->query( $sql );
        }
}

endif;