patilswapnilv / shortcodely

        $caps = $this->getAllCapsAndRoles();

        foreach ( $caps as $cap => $roles ) {

            if (array_intersect($user->roles, $roles) ) {

                $this->assertTrue($user->has_cap($cap), "User should have the {$cap} capability");

                $this->assertTrue(user_can($user, $cap), "User should have the {$cap} capability");

            } else {

                $this->assertFalse($user->has_cap($cap), "User should not have the {$cap} capability");

                $this->assertFalse(user_can($user, $cap), "User should not have the {$cap} capability");

            }

        }

        $user->remove_role('contributor');

        // user should have one role now

        $caps = $this->getPrimitiveCapsAndRoles();

        foreach ( $caps as $cap => $roles ) {

               // the user should have all the above caps

            if (isset($expected_caps[ $cap ]) ) {

                $this->assertTrue($user->has_cap($cap), "User should have the {$cap} capability");

                $this->assertTrue(user_can($user, $cap), "User should have the {$cap} capability");

            } else {

                 $this->assertFalse($user->has_cap($cap), "User should not have the {$cap} capability");

                 $this->assertFalse(user_can($user, $cap), "User should not have the {$cap} capability");

            }

        }

        // clean up

        remove_role($role_name);