Changeset 46086 for trunk/tests/phpunit/tests/comment/query.php

Timestamp:

09/10/2019 06:41:03 PM (4 years ago)

Author:

adamsilverstein

Message:

Comments: add a new comments_pre_query filter to short circuit WP_Comment_Query 'get_comments' queries.

Return a non-null value to bypass WordPress's default comment queries.

Props felipeelia, spacedmonkey.
Fixes #45800.

File:

• trunk/tests/phpunit/tests/comment/query.php (modified) (1 diff)

trunk/tests/phpunit/tests/comment/query.php

@@ -4883 +4883 @@
         $this->assertEqualSets( $c1, $found );
     }
+
+    /**
+     * @ticket 45800
+     */
+    public function test_comments_pre_query_filter_should_bypass_database_query() {
+        global $wpdb;
+
+        add_filter( 'comments_pre_query', array( __CLASS__, 'filter_comments_pre_query' ), 10, 2 );
+
+        $num_queries = $wpdb->num_queries;
+
+        $q       = new WP_Comment_Query();
+        $results = $q->query( array() );
+
+        remove_filter( 'comments_pre_query', array( __CLASS__, 'filter_comments_pre_query' ), 10, 2 );
+
+        // Make sure no queries were executed.
+        $this->assertSame( $num_queries, $wpdb->num_queries );
+
+        // We manually inserted a non-existing site and overrode the results with it.
+        $this->assertSame( array( 555 ), $results );
+
+        // Make sure manually setting total_users doesn't get overwritten.
+        $this->assertEquals( 1, $q->found_comments );
+    }
+
+    public static function filter_comments_pre_query( $comments, $query ) {
+        $query->found_comments = 1;
+
+        return array( 555 );
+    }
 }