WordPress.org

Make WordPress Core

Ticket #36624: 36624.patch

File 36624.patch, 12.0 KB (added by ryanplas, 5 years ago)
  • src/wp-includes/class-wp-user-query.php

     
    124124                        'fields' => 'all',
    125125                        'who' => '',
    126126                        'has_published_posts' => null,
     127                        'nicename' => '',
     128                        'nicename__in' => array(),
     129                        'nicename__not_in' => array(),
     130                        'login' => '',
     131                        'login__in' => array(),
     132                        'login__not_in' => array()
    127133                );
    128134
    129135                return wp_parse_args( $args, $defaults );
     
    173179         *                                             an array of values, or a multi-dimensional array with fields as
    174180         *                                             keys and orders ('ASC' or 'DESC') as values. Accepted values are
    175181         *                                             'ID', 'display_name' (or 'name'), 'include', 'user_login'
    176          *                                             (or 'login'), 'user_nicename' (or 'nicename'), 'user_email'
    177          *                                             (or 'email'), 'user_url' (or 'url'), 'user_registered'
    178          *                                             or 'registered'), 'post_count', 'meta_value', 'meta_value_num',
    179          *                                             the value of `$meta_key`, or an array key of `$meta_query`. To use
    180          *                                             'meta_value' or 'meta_value_num', `$meta_key` must be also be
    181          *                                             defined. Default 'user_login'.
     182         *                                             (or 'login'), 'login__in', 'user_nicename' (or 'nicename'),
     183         *                                             'nicename__in', 'user_email (or 'email'), 'user_url' (or 'url'),
     184         *                                             'user_registered' (or 'registered'), 'post_count', 'meta_value',
     185         *                                             'meta_value_num', the value of `$meta_key`, or an array key of
     186         *                                             `$meta_query`. To use 'meta_value' or 'meta_value_num', `$meta_key`
     187         *                                             must be also be defined. Default 'user_login'.
    182188         *     @type string       $order               Designates ascending or descending order of users. Order values
    183189         *                                             passed as part of an `$orderby` array take precedence over this
    184190         *                                             parameter. Accepts 'ASC', 'DESC'. Default 'ASC'.
     
    203209         *     @type bool|array   $has_published_posts Pass an array of post types to filter results to users who have
    204210         *                                             published posts in those post types. `true` is an alias for all
    205211         *                                             public post types.
     212         *     @type string       $nicename            The user nicename. Default empty.
     213         *     @type array        $nicename__in        An array of nicenames. Matched users must have at least one of these
     214         *                                             nicenames. Default empty array.
     215         *     @type array        $nicename__not_in    An array of nicenames to exclude. Users matching one or more of these
     216         *                                             nicenames will not be included in results. Default empty array.
     217         *     @type string       $login               The user login. Default empty.
     218         *     @type array        $login__in           An array of logins. Matched users must have at least one of these
     219         *                                             logins. Default empty array.
     220         *     @type array        $login__not_in       An array of logins to exclude. Users matching one or more of these
     221         *                                             logins will not be included in results. Default empty array.
    206222         * }
    207223         */
    208224        public function prepare_query( $query = array() ) {
     
    276292                        $this->query_where .= " AND {$this->db->users}.ID IN ( SELECT DISTINCT $posts_table.post_author FROM $posts_table WHERE $posts_table.post_status = 'publish' AND $posts_table.post_type IN ( " . join( ", ", $post_types ) . " ) )";
    277293                }
    278294
     295                // nicename
     296                if ( '' !== $qv['nicename']) {
     297                        $this->query_where .= $this->db->prepare( ' AND user_nicename = %s', $qv['nicename'] );
     298                }
     299
     300                if ( ! empty( $qv['nicename__in'] ) ) {
     301                        $nicename__in = implode( "','", $qv['nicename__in'] );
     302                        $this->query_where .= " AND user_nicename IN ( '$nicename__in' )";
     303                }
     304
     305                if ( ! empty( $qv['nicename__not_in'] ) ) {
     306                        $nicename__not_in = implode( "','", $qv['nicename__not_in'] );
     307                        $this->query_where .= " AND user_nicename NOT IN ( '$nicename__not_in' )";
     308                }
     309
     310                // login
     311                if ( '' !== $qv['login']) {
     312                        $this->query_where .= $this->db->prepare( ' AND user_login = %s', $qv['login'] );
     313                }
     314
     315                if ( ! empty( $qv['login__in'] ) ) {
     316                        $login__in = implode( "','", $qv['login__in'] );
     317                        $this->query_where .= " AND user_login IN ( '$login__in' )";
     318                }
     319
     320                if ( ! empty( $qv['login__not_in'] ) ) {
     321                        $login__not_in = implode( "','", $qv['login__not_in'] );
     322                        $this->query_where .= " AND user_login NOT IN ( '$login__not_in' )";
     323                }
     324
    279325                // Meta query.
    280326                $this->meta_query = new WP_Meta_Query();
    281327                $this->meta_query->parse_query_vars( $qv );
     
    434480                                continue;
    435481                        }
    436482
     483                        if ( 'nicename__in' === $_orderby || 'login__in' === $_orderby) {
     484                                $orderby_array[] = $parsed;
     485                                continue;
     486                        }
     487
    437488                        $orderby_array[] = $parsed . ' ' . $this->parse_order( $_order );
    438489                }
    439490
     
    700751                        $include = wp_parse_id_list( $this->query_vars['include'] );
    701752                        $include_sql = implode( ',', $include );
    702753                        $_orderby = "FIELD( {$this->db->users}.ID, $include_sql )";
    703                 } elseif ( isset( $meta_query_clauses[ $orderby ] ) ) {
     754                } elseif ( 'nicename__in' === $orderby ) {
     755                        $nicename__in = implode( "','", $this->query_vars['nicename__in']);
     756                        $_orderby = "FIELD( user_nicename, '$nicename__in' )";
     757                } elseif ( 'login__in' === $orderby ) {
     758                        $login__in = implode( "','", $this->query_vars['login__in']);
     759                        $_orderby = "FIELD( user_login, '$login__in' )";
     760                } elseif  ( isset( $meta_query_clauses[ $orderby ] ) ) {
    704761                        $meta_clause = $meta_query_clauses[ $orderby ];
    705762                        $_orderby = sprintf( "CAST(%s.meta_value AS %s)", esc_sql( $meta_clause['alias'] ), esc_sql( $meta_clause['cast'] ) );
    706763                }
  • tests/phpunit/tests/user/query.php

     
    878878        }
    879879
    880880        /**
     881         * @ticket 36624
     882         */
     883        public function test_nicename_returns_user_with_nicename() {
     884                wp_update_user( array(
     885                        'ID' => self::$author_ids[0],
     886                        'user_nicename' => 'peter'
     887                ) );
     888
     889                $q = new WP_User_Query( array (
     890                        'nicename' => 'peter'
     891                ) );
     892
     893                $found = wp_list_pluck( $q->get_results(), 'ID' );
     894                $expected = array( self::$author_ids[0] );
     895
     896                $this->assertContains( "AND user_nicename = 'peter'", $q->query_where);
     897                $this->assertEqualSets( $expected, $found);
     898        }
     899
     900        /**
     901         * @ticket 36624
     902         */
     903        public function test_nicename__in_returns_users_with_included_nicenames() {
     904                wp_update_user( array(
     905                        'ID' => self::$author_ids[0],
     906                        'user_nicename' => 'peter'
     907                ) );
     908
     909                wp_update_user( array(
     910                        'ID' => self::$author_ids[1],
     911                        'user_nicename' => 'paul'
     912                ) );
     913
     914                wp_update_user( array(
     915                        'ID' => self::$author_ids[2],
     916                        'user_nicename' => 'mary'
     917                ) );
     918
     919                $q = new WP_User_Query( array (
     920                        'nicename__in' => array( 'peter', 'paul', 'mary' )
     921                ) );
     922
     923                $found = wp_list_pluck( $q->get_results(), 'ID' );
     924                $expected = array( self::$author_ids[0], self::$author_ids[1], self::$author_ids[2] );
     925
     926                $this->assertContains( "AND user_nicename IN ( 'peter','paul','mary' )", $q->query_where);
     927                $this->assertEqualSets( $expected, $found );
     928        }
     929
     930        /**
     931         * @ticket 36624
     932         */
     933        public function test_nicename__not_in_returns_users_without_included_nicenames() {
     934                wp_update_user( array(
     935                        'ID' => self::$author_ids[0],
     936                        'user_nicename' => 'peter'
     937                ) );
     938
     939                wp_update_user( array(
     940                        'ID' => self::$author_ids[1],
     941                        'user_nicename' => 'paul'
     942                ) );
     943
     944                wp_update_user( array(
     945                        'ID' => self::$author_ids[2],
     946                        'user_nicename' => 'mary'
     947                ) );
     948
     949                $q = new WP_User_Query( array (
     950                        'nicename__not_in' => array( 'peter', 'paul', 'mary' )
     951                ) );
     952
     953                $foundCount = count($q->get_results());
     954                $expectedCount = 10; // 13 total users minus 3 from query
     955
     956                $this->assertContains( "AND user_nicename NOT IN ( 'peter','paul','mary' )", $q->query_where);
     957                $this->assertEquals( $expectedCount, $foundCount );
     958        }
     959
     960        /**
     961         * @ticket 36624
     962         */
     963        public function test_orderby_nicename__in() {
     964                wp_update_user( array(
     965                        'ID' => self::$author_ids[0],
     966                        'user_nicename' => 'peter'
     967                ) );
     968
     969                wp_update_user( array(
     970                        'ID' => self::$author_ids[1],
     971                        'user_nicename' => 'paul'
     972                ) );
     973
     974                wp_update_user( array(
     975                        'ID' => self::$author_ids[2],
     976                        'user_nicename' => 'mary'
     977                ) );
     978
     979                $q = new WP_User_Query( array (
     980                        'nicename__in' => array( 'mary', 'peter', 'paul' ),
     981                        'orderby' => 'nicename__in'
     982                ) );
     983
     984                $found = wp_list_pluck( $q->get_results(), 'ID' );
     985                $expected = array( self::$author_ids[2], self::$author_ids[0], self::$author_ids[1] );
     986
     987                $this->assertContains( "FIELD( user_nicename, 'mary','peter','paul' )", $q->query_orderby);
     988                $this->assertSame( $expected, $found );
     989        }
     990
     991        /**
     992         * @ticket 36624
     993         */
     994        public function test_login_returns_user_with_login() {
     995
     996                $user_login = get_userdata( self::$author_ids[0] )->user_login;
     997
     998                $q = new WP_User_Query( array (
     999                        'login' => $user_login
     1000                ) );
     1001
     1002                $found = wp_list_pluck( $q->get_results(), 'ID' );
     1003                $expected = array( self::$author_ids[0] );
     1004
     1005                $this->assertContains( "AND user_login = '$user_login'", $q->query_where);
     1006                $this->assertEqualSets( $expected, $found);
     1007        }
     1008
     1009        /**
     1010         * @ticket 36624
     1011         */
     1012        public function test_login__in_returns_users_with_included_logins() {
     1013                $user_login1 = get_userdata( self::$author_ids[0] )->user_login;
     1014                $user_login2 = get_userdata( self::$author_ids[1] )->user_login;
     1015                $user_login3 = get_userdata( self::$author_ids[2] )->user_login;
     1016
     1017                $q = new WP_User_Query( array (
     1018                        'login__in' => array( $user_login1, $user_login2, $user_login3 )
     1019                ) );
     1020
     1021                $found = wp_list_pluck( $q->get_results(), 'ID' );
     1022                $expected = array( self::$author_ids[0], self::$author_ids[1], self::$author_ids[2] );
     1023
     1024                $this->assertContains( "AND user_login IN ( '$user_login1','$user_login2','$user_login3' )", $q->query_where);
     1025                $this->assertEqualSets( $expected, $found );
     1026        }
     1027
     1028        /**
     1029         * @ticket 36624
     1030         */
     1031        public function test_login__not_in_returns_users_without_included_logins() {
     1032                $user_login1 = get_userdata( self::$author_ids[0] )->user_login;
     1033                $user_login2 = get_userdata( self::$author_ids[1] )->user_login;
     1034                $user_login3 = get_userdata( self::$author_ids[2] )->user_login;
     1035
     1036                $q = new WP_User_Query( array (
     1037                        'login__not_in' => array( $user_login1, $user_login2, $user_login3 )
     1038                ) );
     1039
     1040                $foundCount = count($q->get_results());
     1041                $expectedCount = 10; // 13 total users minus 3 from query
     1042
     1043                $this->assertContains( "AND user_login NOT IN ( '$user_login1','$user_login2','$user_login3' )", $q->query_where);
     1044                $this->assertEquals( $expectedCount, $foundCount );
     1045        }
     1046
     1047        /**
     1048         * @ticket 36624
     1049         */
     1050        public function test_orderby_login__in() {
     1051                $user_login1 = get_userdata( self::$author_ids[0] )->user_login;
     1052                $user_login2 = get_userdata( self::$author_ids[1] )->user_login;
     1053                $user_login3 = get_userdata( self::$author_ids[2] )->user_login;
     1054
     1055                $q = new WP_User_Query( array (
     1056                        'login__in' => array( $user_login2, $user_login3, $user_login1 ),
     1057                        'orderby' => 'login__in'
     1058                ) );
     1059
     1060                $found = wp_list_pluck( $q->get_results(), 'ID' );
     1061                $expected = array( self::$author_ids[1], self::$author_ids[2], self::$author_ids[0] );
     1062
     1063                $this->assertContains( "FIELD( user_login, '$user_login2','$user_login3','$user_login1' )", $q->query_orderby);
     1064                $this->assertSame( $expected, $found );
     1065        }
     1066
     1067        /**
    8811068         * @ticket 25145
    8821069         */
    8831070        public function test_paged() {