wp/tests/phpunit/tests/user/multisite.php

<?php

if ( is_multisite() ) :

	/**
	 * Tests specific to users in multisite.
	 *
	 * @group user
	 * @group ms-user
	 * @group multisite
	 */
	class Tests_Multisite_User extends WP_UnitTestCase {
		protected $suppress = false;

		function setUp() {
			global $wpdb;
			parent::setUp();
			$this->suppress = $wpdb->suppress_errors();
		}

		function tearDown() {
			global $wpdb;
			$wpdb->suppress_errors( $this->suppress );
			parent::tearDown();
		}

		function test_remove_user_from_blog() {
			$user1 = self::factory()->user->create_and_get();
			$user2 = self::factory()->user->create_and_get();

			$post_id = self::factory()->post->create( array( 'post_author' => $user1->ID ) );

			remove_user_from_blog( $user1->ID, 1, $user2->ID );

			$post = get_post( $post_id );

			$this->assertNotEquals( $user1->ID, $post->post_author );
			$this->assertEquals( $user2->ID, $post->post_author );
		}

		/**
		 * Test the returned data from get_blogs_of_user()
		 */
		function test_get_blogs_of_user() {
			$user1_id = self::factory()->user->create( array( 'role' => 'administrator' ) );

			// Maintain a list of 6 total sites and include the primary network site.
			$blog_ids = self::factory()->blog->create_many( 5, array( 'user_id' => $user1_id ) );
			$blog_ids = array_merge( array( 1 ), $blog_ids );

			// All sites are new and not marked as spam, archived, or deleted.
			$blog_ids_of_user = array_keys( get_blogs_of_user( $user1_id ) );

			// User should be a member of the created sites and the network's initial site.
			$this->assertSame( $blog_ids, $blog_ids_of_user );

			$this->assertTrue( remove_user_from_blog( $user1_id, $blog_ids[0] ) );
			$this->assertTrue( remove_user_from_blog( $user1_id, $blog_ids[2] ) );
			$this->assertTrue( remove_user_from_blog( $user1_id, $blog_ids[4] ) );

			unset( $blog_ids[0] );
			unset( $blog_ids[2] );
			unset( $blog_ids[4] );
			sort( $blog_ids );

			$blogs_of_user = get_blogs_of_user( $user1_id, false );

			// The user should still be a member of all remaining sites.
			$blog_ids_of_user = array_keys( $blogs_of_user );
			$this->assertSame( $blog_ids, $blog_ids_of_user );

			// Each site retrieved should match the expected structure.
			foreach ( $blogs_of_user as $blog_id => $blog ) {
				$this->assertSame( $blog_id, $blog->userblog_id );
				$this->assertTrue( isset( $blog->userblog_id ) );
				$this->assertTrue( isset( $blog->blogname ) );
				$this->assertTrue( isset( $blog->domain ) );
				$this->assertTrue( isset( $blog->path ) );
				$this->assertTrue( isset( $blog->site_id ) );
				$this->assertTrue( isset( $blog->siteurl ) );
				$this->assertTrue( isset( $blog->archived ) );
				$this->assertTrue( isset( $blog->spam ) );
				$this->assertTrue( isset( $blog->deleted ) );
			}

			// Mark each remaining site as spam, archived, and deleted.
			update_blog_details( $blog_ids[0], array( 'spam' => 1 ) );
			update_blog_details( $blog_ids[1], array( 'archived' => 1 ) );
			update_blog_details( $blog_ids[2], array( 'deleted' => 1 ) );

			// Passing true as the second parameter should retrieve ALL sites, even if marked.
			$blogs_of_user    = get_blogs_of_user( $user1_id, true );
			$blog_ids_of_user = array_keys( $blogs_of_user );
			$this->assertSame( $blog_ids, $blog_ids_of_user );

			// Check if sites are flagged as expected.
			$this->assertEquals( 1, $blogs_of_user[ $blog_ids[0] ]->spam );
			$this->assertEquals( 1, $blogs_of_user[ $blog_ids[1] ]->archived );
			$this->assertEquals( 1, $blogs_of_user[ $blog_ids[2] ]->deleted );

			unset( $blog_ids[0] );
			unset( $blog_ids[1] );
			unset( $blog_ids[2] );
			sort( $blog_ids );

			// Passing false (the default) as the second parameter should retrieve only good sites.
			$blog_ids_of_user = array_keys( get_blogs_of_user( $user1_id, false ) );
			$this->assertSame( $blog_ids, $blog_ids_of_user );
		}

		/**
		 * @expectedDeprecated is_blog_user
		 */
		function test_is_blog_user() {
			global $wpdb;

			$user1_id = self::factory()->user->create( array( 'role' => 'administrator' ) );

			$old_current = get_current_user_id();
			wp_set_current_user( $user1_id );

			$this->assertTrue( is_blog_user() );
			$this->assertTrue( is_blog_user( get_current_blog_id() ) );

			$blog_id = self::factory()->blog->create( array( 'user_id' => get_current_user_id() ) );

			$this->assertInternalType( 'int', $blog_id );
			$this->assertTrue( is_blog_user( $blog_id ) );
			$this->assertTrue( remove_user_from_blog( $user1_id, $blog_id ) );
			$this->assertFalse( is_blog_user( $blog_id ) );

			wp_set_current_user( $old_current );
		}

		function test_is_user_member_of_blog() {
			global $wpdb;

			$user1_id = self::factory()->user->create( array( 'role' => 'administrator' ) );
			$user2_id = self::factory()->user->create( array( 'role' => 'administrator' ) );

			$old_current = get_current_user_id();

			$this->assertSame( 0, $old_current );

			// Test for "get current user" when not logged in.
			$this->assertFalse( is_user_member_of_blog() );

			wp_set_current_user( $user1_id );
			$site_id = get_current_blog_id();

			$this->assertTrue( is_user_member_of_blog() );
			$this->assertTrue( is_user_member_of_blog( 0, 0 ) );
			$this->assertTrue( is_user_member_of_blog( 0, $site_id ) );
			$this->assertTrue( is_user_member_of_blog( $user1_id ) );
			$this->assertTrue( is_user_member_of_blog( $user1_id, $site_id ) );

			$blog_id = self::factory()->blog->create( array( 'user_id' => get_current_user_id() ) );

			$this->assertInternalType( 'int', $blog_id );

			// Current user gets added to new blogs.
			$this->assertTrue( is_user_member_of_blog( $user1_id, $blog_id ) );
			// Other users should not.
			$this->assertFalse( is_user_member_of_blog( $user2_id, $blog_id ) );

			switch_to_blog( $blog_id );

			$this->assertTrue( is_user_member_of_blog( $user1_id ) );
			$this->assertFalse( is_user_member_of_blog( $user2_id ) );

			// Remove user 1 from blog.
			$this->assertTrue( remove_user_from_blog( $user1_id, $blog_id ) );

			// Add user 2 to blog.
			$this->assertTrue( add_user_to_blog( $blog_id, $user2_id, 'subscriber' ) );

			$this->assertFalse( is_user_member_of_blog( $user1_id ) );
			$this->assertTrue( is_user_member_of_blog( $user2_id ) );

			restore_current_blog();

			$this->assertFalse( is_user_member_of_blog( $user1_id, $blog_id ) );
			$this->assertTrue( is_user_member_of_blog( $user2_id, $blog_id ) );

			wpmu_delete_user( $user1_id );
			$user = new WP_User( $user1_id );
			$this->assertFalse( $user->exists() );
			$this->assertFalse( is_user_member_of_blog( $user1_id ) );

			wp_set_current_user( $old_current );
		}

		/**
		 * @ticket 23192
		 */
		function test_is_user_spammy() {
			$user_id = self::factory()->user->create(
				array(
					'role'       => 'author',
					'user_login' => 'testuser1',
				)
			);

			$spam_username = (string) $user_id;
			$spam_user_id  = self::factory()->user->create(
				array(
					'role'       => 'author',
					'user_login' => $spam_username,
				)
			);
			wp_update_user(
				array(
					'ID'   => $spam_user_id,
					'spam' => '1',
				)
			);

			$this->assertTrue( is_user_spammy( $spam_username ) );
			$this->assertFalse( is_user_spammy( 'testuser1' ) );
		}

		/**
		 * @ticket 20601
		 */
		function test_user_member_of_blog() {
			global $wp_rewrite;

			self::factory()->blog->create();
			$user_id = self::factory()->user->create();
			self::factory()->blog->create( array( 'user_id' => $user_id ) );

			$blogs = get_blogs_of_user( $user_id );
			$this->assertCount( 2, $blogs );
			$first = reset( $blogs )->userblog_id;
			remove_user_from_blog( $user_id, $first );

			$blogs  = get_blogs_of_user( $user_id );
			$second = reset( $blogs )->userblog_id;
			$this->assertCount( 1, $blogs );

			switch_to_blog( $first );
			$wp_rewrite->init();

			$this->go_to( get_author_posts_url( $user_id ) );
			$this->assertQueryTrue( 'is_404' );

			switch_to_blog( $second );
			$wp_rewrite->init();

			$this->go_to( get_author_posts_url( $user_id ) );
			$this->assertQueryTrue( 'is_author', 'is_archive' );

			add_user_to_blog( $first, $user_id, 'administrator' );
			$blogs = get_blogs_of_user( $user_id );
			$this->assertCount( 2, $blogs );

			switch_to_blog( $first );
			$wp_rewrite->init();

			$this->go_to( get_author_posts_url( $user_id ) );
			$this->assertQueryTrue( 'is_author', 'is_archive' );
		}

		function test_revoked_super_admin_can_be_deleted() {
			if ( isset( $GLOBALS['super_admins'] ) ) {
				$old_global = $GLOBALS['super_admins'];
				unset( $GLOBALS['super_admins'] );
			}

			$user_id = self::factory()->user->create();
			grant_super_admin( $user_id );
			revoke_super_admin( $user_id );

			$this->assertTrue( wpmu_delete_user( $user_id ) );

			if ( isset( $old_global ) ) {
				$GLOBALS['super_admins'] = $old_global;
			}
		}

		function test_revoked_super_admin_is_deleted() {
			if ( isset( $GLOBALS['super_admins'] ) ) {
				$old_global = $GLOBALS['super_admins'];
				unset( $GLOBALS['super_admins'] );
			}

			$user_id = self::factory()->user->create();
			grant_super_admin( $user_id );
			revoke_super_admin( $user_id );
			wpmu_delete_user( $user_id );
			$user = new WP_User( $user_id );

			$this->assertFalse( $user->exists(), 'WP_User->exists' );

			if ( isset( $old_global ) ) {
				$GLOBALS['super_admins'] = $old_global;
			}
		}

		function test_super_admin_cannot_be_deleted() {
			if ( isset( $GLOBALS['super_admins'] ) ) {
				$old_global = $GLOBALS['super_admins'];
				unset( $GLOBALS['super_admins'] );
			}

			$user_id = self::factory()->user->create();
			grant_super_admin( $user_id );

			$this->assertFalse( wpmu_delete_user( $user_id ) );

			if ( isset( $old_global ) ) {
				$GLOBALS['super_admins'] = $old_global;
			}
		}

		/**
		 * @ticket 27205
		 */
		function test_granting_super_admins() {
			if ( isset( $GLOBALS['super_admins'] ) ) {
				$old_global = $GLOBALS['super_admins'];
				unset( $GLOBALS['super_admins'] );
			}

			$user_id = self::factory()->user->create();

			$this->assertFalse( is_super_admin( $user_id ) );
			$this->assertFalse( revoke_super_admin( $user_id ) );
			$this->assertTrue( grant_super_admin( $user_id ) );
			$this->assertTrue( is_super_admin( $user_id ) );
			$this->assertFalse( grant_super_admin( $user_id ) );
			$this->assertTrue( revoke_super_admin( $user_id ) );

			// None of these operations should set the $super_admins global.
			$this->assertFalse( isset( $GLOBALS['super_admins'] ) );

			// Try with two users.
			$second_user = self::factory()->user->create();
			$this->assertTrue( grant_super_admin( $user_id ) );
			$this->assertTrue( grant_super_admin( $second_user ) );
			$this->assertTrue( is_super_admin( $second_user ) );
			$this->assertTrue( is_super_admin( $user_id ) );
			$this->assertTrue( revoke_super_admin( $user_id ) );
			$this->assertTrue( revoke_super_admin( $second_user ) );

			if ( isset( $old_global ) ) {
				$GLOBALS['super_admins'] = $old_global;
			}
		}

		public function test_numeric_string_user_id() {
			$u = self::factory()->user->create();

			$u_string = (string) $u;
			$this->assertTrue( wpmu_delete_user( $u_string ) );
			$this->assertFalse( get_user_by( 'id', $u ) );
		}

		/**
		 * @ticket 33800
		 */
		public function test_should_return_false_for_non_numeric_string_user_id() {
			$this->assertFalse( wpmu_delete_user( 'abcde' ) );
		}

		/**
		 * @ticket 33800
		 */
		public function test_should_return_false_for_object_user_id() {
			$u_obj = self::factory()->user->create_and_get();
			$this->assertFalse( wpmu_delete_user( $u_obj ) );
			$this->assertSame( $u_obj->ID, username_exists( $u_obj->user_login ) );
		}

		/**
		 * @ticket 38356
		 */
		public function test_add_user_to_blog_subscriber() {
			$site_id = self::factory()->blog->create();
			$user_id = self::factory()->user->create();

			add_user_to_blog( $site_id, $user_id, 'subscriber' );

			switch_to_blog( $site_id );
			$user = get_user_by( 'id', $user_id );
			restore_current_blog();

			wp_delete_site( $site_id );
			wpmu_delete_user( $user_id );

			$this->assertContains( 'subscriber', $user->roles );
		}

		/**
		 * @ticket 38356
		 */
		public function test_add_user_to_blog_invalid_user() {
			$site_id = self::factory()->blog->create();

			$result = add_user_to_blog( 73622, $site_id, 'subscriber' );

			wp_delete_site( $site_id );

			$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 );
		}

		/**
		 * @ticket 23016
		 */
		public function test_wp_roles_global_is_reset() {
			global $wp_roles;
			$role      = 'test_global_is_reset';
			$role_name = 'Test Global Is Reset';
			$blog_id   = self::factory()->blog->create();

			$wp_roles->add_role( $role, $role_name, array() );

			$this->assertNotEmpty( $wp_roles->get_role( $role ) );

			switch_to_blog( $blog_id );

			$this->assertEmpty( $wp_roles->get_role( $role ) );

			$wp_roles->add_role( $role, $role_name, array() );

			$this->assertNotEmpty( $wp_roles->get_role( $role ) );

			restore_current_blog();

			$this->assertNotEmpty( $wp_roles->get_role( $role ) );

			$wp_roles->remove_role( $role );
		}

	}

endif;