Changeset 34804 for trunk/tests/phpunit/tests/user/query.php

Timestamp:

10/03/2015 06:44:40 PM (11 years ago)

Author:

boonebgorges

Message:

Ensure that WP_User_Query vars are filled after 'pre_get_users'.

This prevents notices from being thrown when a 'pre_get_users' callback
removes required values from the list of query_vars.

For backward compatibility with previous uses of 'pre_get_users', default
values are parsed both before and after the action is fired.

Fixes #33449.

File:

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

trunk/tests/phpunit/tests/user/query.php

@@ -875 +875 @@
                 $this->assertEquals( array( $users[2], $users[1] ), $q->results );
         }
+
+        /**
+         * @ticket 33449
+         */
+        public function test_query_vars_should_be_filled_in_after_pre_get_users() {
+                $query_vars = array( 'blog_id', 'role', 'meta_key', 'meta_value', 'meta_compare', 'include', 'exclude', 'search', 'search_columns', 'orderby', 'order', 'offset', 'number', 'paged', 'count_total', 'fields', 'who', 'has_published_posts' );
+
+                add_action( 'pre_get_users', array( $this, 'filter_pre_get_users_args' ) );
+                $q = new WP_User_Query( array_fill_keys( $query_vars, '1' ) );
+                remove_action( 'pre_get_users', array( $this, 'filter_pre_get_users_args' ) );
+
+                foreach ( $query_vars as $query_var ) {
+                        $this->assertTrue( array_key_exists( $query_var, $q->query_vars ), "$query_var does not exist." );
+                }
+
+        }
+
+        public function filter_pre_get_users_args( $q ) {
+                foreach ( $q->query_vars as $k => $v ) {
+                        unset( $q->query_vars[ $k ] );
+                }
+        }
 }