Ticket #41197: 41197.diff

src/wp-includes/class-wp-site-query.php

@@ -250 +250 @@
                  */
                 do_action_ref_array( 'pre_get_sites', array( &$this ) );
 
+                $_args = wp_array_slice_assoc( $this->query_vars, array_keys( $this->query_var_defaults ) );
+                if ( ! $_args['count'] ) {
+                        unset( $_args['fields'] );
+                }
+
                 // $args can include anything. Only use the args defined in the query_var_defaults to compute the key.
-                $key = md5( serialize( wp_array_slice_assoc( $this->query_vars, array_keys( $this->query_var_defaults ) ) ) );
+                $key = md5( serialize( $_args ) );
                 $last_changed = wp_cache_get_last_changed( 'sites' );
 
                 $cache_key = "get_sites:$key:$last_changed";

tests/phpunit/tests/multisite/siteQuery.php

@@ -163 +163 @@
                 $this->assertEqualSets( $expected, $found );
         }
 
+        public function test_wp_site_query_cache() {
+                global $wpdb;
+                $q                 = new WP_Site_Query();
+                $query_1           = $q->query( array(
+                        'fields'     => 'all',
+                        'network_id' => self::$network_ids['wordpress.org/'],
+                        'number'     => 3,
+                        'order'      => 'ASC',
+                ) );
+                $number_of_queries = $wpdb->num_queries;
+
+                $query_2 = $q->query( array(
+                        'fields'     => 'ids',
+                        'network_id' => self::$network_ids['wordpress.org/'],
+                        'number'     => 3,
+                        'order'      => 'ASC',
+                ) );
+
+                $this->assertEquals( $number_of_queries, $wpdb->num_queries );
+
+                $query_3 = $q->query( array(
+                        'fields'     => 'ids',
+                        'network_id' => self::$network_ids['wordpress.org/'],
+                        'number'     => 3,
+                        'order'      => 'ASC',
+                        'count'      => true,
+                ) );
+                $this->assertEquals( $number_of_queries + 1, $wpdb->num_queries );
+        }
+
         public function test_wp_site_query_by_network_id_with_order() {
                 $q = new WP_Site_Query();
                 $found = $q->query( array(