Changeset 41225 for trunk/tests/phpunit/tests/user/multisite.php

Timestamp:

08/03/2017 09:40:02 PM (8 years ago)

Author:

flixos90

Message:

Multisite: Introduce a can_add_user_to_blog filter to prevent adding a user to a site.

Under certain circumstances, it can be necessary that a user should not be added to a site, beyond the restrictions that WordPress core applies. With the new can_add_user_to_blog filter, plugin developers can run custom checks and return an error in case of a failure, that will prevent the user from being added.

The user-facing parts and the REST API route that interact with add_user_to_blog() have been adjusted accordingly to provide appropriate error feedback when a user could not be added to a site. Furthermore, two existing error feedback messages in the site admin's "New User" screen have been adjusted to properly show inside an error notice instead of a success notice.

Props jmdodd.
Fixes #41101.

File:

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

trunk/tests/phpunit/tests/user/multisite.php

@@ -396 +396 @@
 
         $this->assertWPError( $result );
+    }
+
+    /**
+     * @ticket 41101
+     */
+    public function test_should_fail_can_add_user_to_blog_filter() {
+        $site_id = self::factory()->blog->create();
+        $user_id = self::factory()->user->create();
+
+        add_filter( 'can_add_user_to_blog', '__return_false' );
+        $result = add_user_to_blog( $site_id, $user_id, 'subscriber' );
+
+        $this->assertWPError( $result );
+    }
+
+    /**
+     * @ticket 41101
+     */
+    public function test_should_succeed_can_add_user_to_blog_filter() {
+        $site_id = self::factory()->blog->create();
+        $user_id = self::factory()->user->create();
+
+        add_filter( 'can_add_user_to_blog', '__return_true' );
+        $result = add_user_to_blog( $site_id, $user_id, 'subscriber' );
+
+        $this->assertTrue( $result );
     }