View source
<?php
namespace Drupal\Tests\Core\Session;
use Drupal\Core\DependencyInjection\ContainerBuilder;
use Drupal\Core\Session\UserSession;
use Drupal\Tests\UnitTestCase;
use Drupal\user\RoleInterface;
class UserSessionTest extends UnitTestCase {
protected $users = array();
public function providerTestHasPermission() {
$data = array();
$data[] = array(
'example permission',
array(
'user_one',
'user_two',
),
array(
'user_last',
),
);
$data[] = array(
'another example permission',
array(
'user_two',
),
array(
'user_one',
'user_last',
),
);
$data[] = array(
'final example permission',
array(),
array(
'user_one',
'user_two',
'user_last',
),
);
return $data;
}
protected function createUserSession(array $rids = array(), $authenticated = FALSE) {
array_unshift($rids, $authenticated ? RoleInterface::AUTHENTICATED_ID : RoleInterface::ANONYMOUS_ID);
return new UserSession(array(
'roles' => $rids,
));
}
protected function setUp() {
parent::setUp();
$roles = array();
$roles['role_one'] = $this
->getMockBuilder('Drupal\\user\\Entity\\Role')
->disableOriginalConstructor()
->setMethods(array(
'hasPermission',
))
->getMock();
$roles['role_one']
->expects($this
->any())
->method('hasPermission')
->will($this
->returnValueMap(array(
array(
'example permission',
TRUE,
),
array(
'another example permission',
FALSE,
),
array(
'last example permission',
FALSE,
),
)));
$roles['role_two'] = $this
->getMockBuilder('Drupal\\user\\Entity\\Role')
->disableOriginalConstructor()
->setMethods(array(
'hasPermission',
))
->getMock();
$roles['role_two']
->expects($this
->any())
->method('hasPermission')
->will($this
->returnValueMap(array(
array(
'example permission',
TRUE,
),
array(
'another example permission',
TRUE,
),
array(
'last example permission',
FALSE,
),
)));
$roles['anonymous'] = $this
->getMockBuilder('Drupal\\user\\Entity\\Role')
->disableOriginalConstructor()
->setMethods(array(
'hasPermission',
))
->getMock();
$roles['anonymous']
->expects($this
->any())
->method('hasPermission')
->will($this
->returnValueMap(array(
array(
'example permission',
FALSE,
),
array(
'another example permission',
FALSE,
),
array(
'last example permission',
FALSE,
),
)));
$role_storage = $this
->getMockBuilder('Drupal\\user\\RoleStorage')
->disableOriginalConstructor()
->setMethods(array(
'loadMultiple',
))
->getMock();
$role_storage
->expects($this
->any())
->method('loadMultiple')
->will($this
->returnValueMap(array(
array(
array(),
array(),
),
array(
NULL,
$roles,
),
array(
array(
'anonymous',
),
array(
$roles['anonymous'],
),
),
array(
array(
'anonymous',
'role_one',
),
array(
$roles['role_one'],
),
),
array(
array(
'anonymous',
'role_two',
),
array(
$roles['role_two'],
),
),
array(
array(
'anonymous',
'role_one',
'role_two',
),
array(
$roles['role_one'],
$roles['role_two'],
),
),
)));
$entity_manager = $this
->getMock('Drupal\\Core\\Entity\\EntityManagerInterface');
$entity_manager
->expects($this
->any())
->method('getStorage')
->with($this
->equalTo('user_role'))
->will($this
->returnValue($role_storage));
$container = new ContainerBuilder();
$container
->set('entity.manager', $entity_manager);
\Drupal::setContainer($container);
$this->users['user_one'] = $this
->createUserSession(array(
'role_one',
));
$this->users['user_two'] = $this
->createUserSession(array(
'role_one',
'role_two',
));
$this->users['user_three'] = $this
->createUserSession(array(
'role_two',
), TRUE);
$this->users['user_last'] = $this
->createUserSession();
}
public function testHasPermission($permission, array $sessions_with_access, array $sessions_without_access) {
foreach ($sessions_with_access as $name) {
$this
->assertTrue($this->users[$name]
->hasPermission($permission));
}
foreach ($sessions_without_access as $name) {
$this
->assertFalse($this->users[$name]
->hasPermission($permission));
}
}
public function testUserGetRoles() {
$this
->assertEquals(array(
RoleInterface::AUTHENTICATED_ID,
'role_two',
), $this->users['user_three']
->getRoles());
$this
->assertEquals(array(
'role_two',
), $this->users['user_three']
->getRoles(TRUE));
}
}