View source
<?php
namespace Drupal\Tests\user\Unit;
use Drupal\Core\Extension\Extension;
use Drupal\Core\StringTranslation\PluralTranslatableMarkup;
use Drupal\Core\StringTranslation\TranslatableMarkup;
use Drupal\Core\StringTranslation\TranslationInterface;
use Drupal\Tests\UnitTestCase;
use Drupal\user\PermissionHandler;
use org\bovigo\vfs\vfsStream;
use org\bovigo\vfs\vfsStreamDirectory;
use org\bovigo\vfs\vfsStreamWrapper;
class PermissionHandlerTest extends UnitTestCase {
protected $permissionHandler;
protected $moduleHandler;
protected $stringTranslation;
protected $controllerResolver;
protected function setUp() : void {
parent::setUp();
$this->stringTranslation = new TestTranslationManager();
$this->controllerResolver = $this
->createMock('Drupal\\Core\\Controller\\ControllerResolverInterface');
}
protected function mockModuleExtension($module, $name) {
$extension = new Extension('vfs:/', $module, "modules/{$module}");
$extension->info['name'] = $name;
return $extension;
}
public function testBuildPermissionsYaml() {
vfsStreamWrapper::register();
$root = new vfsStreamDirectory('modules');
vfsStreamWrapper::setRoot($root);
$this->moduleHandler = $this
->createMock('Drupal\\Core\\Extension\\ModuleHandlerInterface');
$this->moduleHandler
->expects($this
->once())
->method('getModuleDirectories')
->willReturn([
'module_a' => vfsStream::url('modules/module_a'),
'module_b' => vfsStream::url('modules/module_b'),
'module_c' => vfsStream::url('modules/module_c'),
]);
$url = vfsStream::url('modules');
mkdir($url . '/module_a');
file_put_contents($url . '/module_a/module_a.permissions.yml', "access_module_a: single_description");
mkdir($url . '/module_b');
file_put_contents($url . '/module_b/module_b.permissions.yml', <<<EOF
'access module b':
title: 'Access B'
description: 'bla bla'
'access module a via module b':
title: 'Access A via B'
provider: 'module_a'
EOF
);
mkdir($url . '/module_c');
file_put_contents($url . '/module_c/module_c.permissions.yml', <<<EOF
'access_module_c':
title: 'Access C'
description: 'bla bla'
'restrict access': TRUE
EOF
);
$modules = [
'module_a',
'module_b',
'module_c',
];
$this->moduleHandler
->expects($this
->any())
->method('getModuleList')
->willReturn(array_flip($modules));
$this->controllerResolver
->expects($this
->never())
->method('getControllerFromDefinition');
$this->permissionHandler = new PermissionHandler($this->moduleHandler, $this->stringTranslation, $this->controllerResolver);
$actual_permissions = $this->permissionHandler
->getPermissions();
$this
->assertPermissions($actual_permissions);
$this
->assertTrue($this->permissionHandler
->moduleProvidesPermissions('module_a'));
$this
->assertTrue($this->permissionHandler
->moduleProvidesPermissions('module_b'));
$this
->assertTrue($this->permissionHandler
->moduleProvidesPermissions('module_c'));
$this
->assertFalse($this->permissionHandler
->moduleProvidesPermissions('module_d'));
}
public function testBuildPermissionsSortPerModule() {
vfsStreamWrapper::register();
$root = new vfsStreamDirectory('modules');
vfsStreamWrapper::setRoot($root);
$this->moduleHandler = $this
->createMock('Drupal\\Core\\Extension\\ModuleHandlerInterface');
$this->moduleHandler
->expects($this
->once())
->method('getModuleDirectories')
->willReturn([
'module_a' => vfsStream::url('modules/module_a'),
'module_b' => vfsStream::url('modules/module_b'),
'module_c' => vfsStream::url('modules/module_c'),
]);
$this->moduleHandler
->expects($this
->exactly(3))
->method('getName')
->willReturnMap([
[
'module_a',
'Module a',
],
[
'module_b',
'Module b',
],
[
'module_c',
'A Module',
],
]);
$url = vfsStream::url('modules');
mkdir($url . '/module_a');
file_put_contents($url . '/module_a/module_a.permissions.yml', <<<EOF
access_module_a2: single_description2
access_module_a1: single_description1
EOF
);
mkdir($url . '/module_b');
file_put_contents($url . '/module_b/module_b.permissions.yml', "access_module_a3: single_description");
mkdir($url . '/module_c');
file_put_contents($url . '/module_c/module_c.permissions.yml', "access_module_a4: single_description");
$modules = [
'module_a',
'module_b',
'module_c',
];
$this->moduleHandler
->expects($this
->once())
->method('getModuleList')
->willReturn(array_flip($modules));
$permissionHandler = new PermissionHandler($this->moduleHandler, $this->stringTranslation, $this->controllerResolver);
$actual_permissions = $permissionHandler
->getPermissions();
$this
->assertEquals([
'access_module_a4',
'access_module_a1',
'access_module_a2',
'access_module_a3',
], array_keys($actual_permissions));
}
public function testBuildPermissionsYamlCallback() {
vfsStreamWrapper::register();
$root = new vfsStreamDirectory('modules');
vfsStreamWrapper::setRoot($root);
$this->moduleHandler = $this
->createMock('Drupal\\Core\\Extension\\ModuleHandlerInterface');
$this->moduleHandler
->expects($this
->once())
->method('getModuleDirectories')
->willReturn([
'module_a' => vfsStream::url('modules/module_a'),
'module_b' => vfsStream::url('modules/module_b'),
'module_c' => vfsStream::url('modules/module_c'),
]);
$url = vfsStream::url('modules');
mkdir($url . '/module_a');
file_put_contents($url . '/module_a/module_a.permissions.yml', <<<EOF
permission_callbacks:
- 'Drupal\\user\\Tests\\TestPermissionCallbacks::singleDescription'
EOF
);
mkdir($url . '/module_b');
file_put_contents($url . '/module_b/module_b.permissions.yml', <<<EOF
permission_callbacks:
- 'Drupal\\user\\Tests\\TestPermissionCallbacks::titleDescription'
- 'Drupal\\user\\Tests\\TestPermissionCallbacks::titleProvider'
EOF
);
mkdir($url . '/module_c');
file_put_contents($url . '/module_c/module_c.permissions.yml', <<<EOF
permission_callbacks:
- 'Drupal\\user\\Tests\\TestPermissionCallbacks::titleDescriptionRestrictAccess'
EOF
);
$modules = [
'module_a',
'module_b',
'module_c',
];
$this->moduleHandler
->expects($this
->any())
->method('getModuleList')
->willReturn(array_flip($modules));
$this->controllerResolver
->expects($this
->exactly(4))
->method('getControllerFromDefinition')
->willReturnMap([
[
'Drupal\\user\\Tests\\TestPermissionCallbacks::singleDescription',
[
new TestPermissionCallbacks(),
'singleDescription',
],
],
[
'Drupal\\user\\Tests\\TestPermissionCallbacks::titleDescription',
[
new TestPermissionCallbacks(),
'titleDescription',
],
],
[
'Drupal\\user\\Tests\\TestPermissionCallbacks::titleProvider',
[
new TestPermissionCallbacks(),
'titleProvider',
],
],
[
'Drupal\\user\\Tests\\TestPermissionCallbacks::titleDescriptionRestrictAccess',
[
new TestPermissionCallbacks(),
'titleDescriptionRestrictAccess',
],
],
]);
$this->permissionHandler = new PermissionHandler($this->moduleHandler, $this->stringTranslation, $this->controllerResolver);
$actual_permissions = $this->permissionHandler
->getPermissions();
$this
->assertPermissions($actual_permissions);
}
public function testPermissionsYamlStaticAndCallback() {
vfsStreamWrapper::register();
$root = new vfsStreamDirectory('modules');
vfsStreamWrapper::setRoot($root);
$this->moduleHandler = $this
->createMock('Drupal\\Core\\Extension\\ModuleHandlerInterface');
$this->moduleHandler
->expects($this
->once())
->method('getModuleDirectories')
->willReturn([
'module_a' => vfsStream::url('modules/module_a'),
]);
$url = vfsStream::url('modules');
mkdir($url . '/module_a');
file_put_contents($url . '/module_a/module_a.permissions.yml', <<<EOF
'access module a':
title: 'Access A'
description: 'bla bla'
permission_callbacks:
- 'Drupal\\user\\Tests\\TestPermissionCallbacks::titleDescription'
EOF
);
$modules = [
'module_a',
];
$this->moduleHandler
->expects($this
->any())
->method('getModuleList')
->willReturn(array_flip($modules));
$this->controllerResolver
->expects($this
->once())
->method('getControllerFromDefinition')
->with('Drupal\\user\\Tests\\TestPermissionCallbacks::titleDescription')
->willReturn([
new TestPermissionCallbacks(),
'titleDescription',
]);
$this->permissionHandler = new PermissionHandler($this->moduleHandler, $this->stringTranslation, $this->controllerResolver);
$actual_permissions = $this->permissionHandler
->getPermissions();
$this
->assertCount(2, $actual_permissions);
$this
->assertEquals('Access A', $actual_permissions['access module a']['title']);
$this
->assertEquals('module_a', $actual_permissions['access module a']['provider']);
$this
->assertEquals('bla bla', $actual_permissions['access module a']['description']);
$this
->assertEquals('Access B', $actual_permissions['access module b']['title']);
$this
->assertEquals('module_a', $actual_permissions['access module b']['provider']);
$this
->assertEquals('bla bla', $actual_permissions['access module b']['description']);
}
protected function assertPermissions(array $actual_permissions) : void {
$this
->assertCount(4, $actual_permissions);
$this
->assertEquals('single_description', $actual_permissions['access_module_a']['title']);
$this
->assertEquals('module_a', $actual_permissions['access_module_a']['provider']);
$this
->assertEquals('Access B', $actual_permissions['access module b']['title']);
$this
->assertEquals('module_b', $actual_permissions['access module b']['provider']);
$this
->assertEquals('Access C', $actual_permissions['access_module_c']['title']);
$this
->assertEquals('module_c', $actual_permissions['access_module_c']['provider']);
$this
->assertTrue($actual_permissions['access_module_c']['restrict access']);
$this
->assertEquals('module_a', $actual_permissions['access module a via module b']['provider']);
}
}
class TestPermissionCallbacks {
public function singleDescription() {
return [
'access_module_a' => 'single_description',
];
}
public function titleDescription() {
return [
'access module b' => [
'title' => 'Access B',
'description' => 'bla bla',
],
];
}
public function titleDescriptionRestrictAccess() {
return [
'access_module_c' => [
'title' => 'Access C',
'description' => 'bla bla',
'restrict access' => TRUE,
],
];
}
public function titleProvider() {
return [
'access module a via module b' => [
'title' => 'Access A via B',
'provider' => 'module_a',
],
];
}
}
class TestTranslationManager implements TranslationInterface {
public function translate($string, array $args = [], array $options = []) {
return new TranslatableMarkup($string, $args, $options, $this);
}
public function translateString(TranslatableMarkup $translated_string) {
return $translated_string
->getUntranslatedString();
}
public function formatPlural($count, $singular, $plural, array $args = [], array $options = []) {
return new PluralTranslatableMarkup($count, $singular, $plural, $args, $options, $this);
}
}