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

Timestamp:

10/24/2014 12:18:44 PM (10 years ago)

Author:

boonebgorges

Message:

Improve unit tests for WP_User_Query include param.

File:

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

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

@@ -35 +35 @@
     }
 
-    function test_include() {
-        $users = new WP_User_Query();
-        $users->set( 'fields', '' );
-        $users->set( 'include', $this->user_id );
-        $users->prepare_query();
-        $users->query();
+    public function test_include_single() {
+        $users = $this->factory->user->create_many( 2 );
+        $q = new WP_User_Query( array(
+            'fields' => '',
+            'include' => $users[0],
+        ) );
+        $ids = $q->get_results();
 
-        $ids = $users->get_results();
-        $this->assertEquals( array( $this->user_id ), $ids );
+        $this->assertEquals( array( $users[0] ), $ids );
+    }
+
+    public function test_include_comma_separated() {
+        $users = $this->factory->user->create_many( 3 );
+        $q = new WP_User_Query( array(
+            'fields' => '',
+            'include' => $users[0] . ', ' . $users[2],
+        ) );
+        $ids = $q->get_results();
+
+        $this->assertEqualSets( array( $users[0], $users[2] ), $ids );
+    }
+
+    public function test_include_array() {
+        $users = $this->factory->user->create_many( 3 );
+        $q = new WP_User_Query( array(
+            'fields' => '',
+            'include' => array( $users[0], $users[2] ),
+        ) );
+        $ids = $q->get_results();
+
+        $this->assertEqualSets( array( $users[0], $users[2] ), $ids );
+    }
+
+    public function test_include_array_bad_values() {
+        $users = $this->factory->user->create_many( 3 );
+        $q = new WP_User_Query( array(
+            'fields' => '',
+            'include' => array( $users[0], 'foo', $users[2] ),
+        ) );
+        $ids = $q->get_results();
+
+        $this->assertEqualSets( array( $users[0], $users[2] ), $ids );
     }