Ticket #21426: wp-user-query-tests.diff

tests/user/query.php

@@ +1 @@
+<?php
+/**
+ * Test WP_User Query, in wp-includes/user.php
+ *
+ * @group user
+ */
+class Tests_WP_User_Query extends WP_UnitTestCase {
+        
+        protected $user_id;
+        
+        function setUp() {
+                parent::setUp();
+
+                $this->user_id = $this->factory->user->create( array(
+                        'role' => 'author'
+                ) );
+        }
+        
+        function test_get_and_set() {
+                $users = new WP_User_Query();
+                
+                $users->set( 'fields', 'all' );
+                $this->assertEquals( $users->get( 'fields' ), 'all' );
+                
+                $users->set( 'fields', '' );
+                $this->assertEquals( $users->get( 'fields' ), '' );
+        }
+        
+        function test_include() {
+                $users = new WP_User_Query();
+                $users->set( 'fields', '' );
+                $users->set( 'include', $this->user_id );
+                $users->prepare_query();
+                $users->query();
+                
+                $ids = $users->get_results();
+                $this->assertTrue( in_array( $this->user_id, $ids ) );
+        }
+        
+        function test_exclude() {
+                $users = new WP_User_Query();
+                $users->set( 'fields', '' );
+                $users->set( 'exclude', $this->user_id );
+                $users->prepare_query();
+                $users->query();
+                
+                $ids = $users->get_results();
+                $this->assertTrue( empty( $ids ) || ! in_array( $this->user_id, $ids ) );
+        }
+}