ShortcutTestBase.php

<?php

namespace Drupal\Tests\shortcut\Functional;

use Drupal\shortcut\Entity\Shortcut;
use Drupal\shortcut\Entity\ShortcutSet;
use Drupal\shortcut\ShortcutSetInterface;
use Drupal\Tests\BrowserTestBase;

/**
 * Defines base class for shortcut test cases.
 */
abstract class ShortcutTestBase extends BrowserTestBase {
    
    /**
     * Modules to enable.
     *
     * @var array
     */
    protected static $modules = [
        'node',
        'toolbar',
        'shortcut',
    ];
    
    /**
     * User with permission to administer shortcuts.
     *
     * @var \Drupal\user\UserInterface
     */
    protected $adminUser;
    
    /**
     * User with permission to use shortcuts, but not administer them.
     *
     * @var \Drupal\user\UserInterface
     */
    protected $shortcutUser;
    
    /**
     * Generic node used for testing.
     *
     * @var \Drupal\node\NodeInterface
     */
    protected $node;
    
    /**
     * Site-wide default shortcut set.
     *
     * @var \Drupal\shortcut\ShortcutSetInterface
     */
    protected $set;
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() {
        parent::setUp();
        if ($this->profile != 'standard') {
            // Create Basic page and Article node types.
            $this->drupalCreateContentType([
                'type' => 'page',
                'name' => 'Basic page',
            ]);
            $this->drupalCreateContentType([
                'type' => 'article',
                'name' => 'Article',
            ]);
            // Populate the default shortcut set.
            $shortcut = Shortcut::create([
                'shortcut_set' => 'default',
                'title' => t('Add content'),
                'weight' => -20,
                'link' => [
                    'uri' => 'internal:/node/add',
                ],
            ]);
            $shortcut->save();
            $shortcut = Shortcut::create([
                'shortcut_set' => 'default',
                'title' => t('All content'),
                'weight' => -19,
                'link' => [
                    'uri' => 'internal:/admin/content',
                ],
            ]);
            $shortcut->save();
        }
        // Create users.
        $this->adminUser = $this->drupalCreateUser([
            'access toolbar',
            'administer shortcuts',
            'view the administration theme',
            'create article content',
            'create page content',
            'access content overview',
            'administer users',
            'link to any page',
            'edit any article content',
        ]);
        $this->shortcutUser = $this->drupalCreateUser([
            'customize shortcut links',
            'switch shortcut sets',
            'access shortcuts',
            'access content',
        ]);
        // Create a node.
        $this->node = $this->drupalCreateNode([
            'type' => 'article',
        ]);
        // Log in as admin and grab the default shortcut set.
        $this->drupalLogin($this->adminUser);
        $this->set = ShortcutSet::load('default');
        \Drupal::entityTypeManager()->getStorage('shortcut_set')
            ->assignUser($this->set, $this->adminUser);
    }
    
    /**
     * Creates a generic shortcut set.
     */
    public function generateShortcutSet($label = '', $id = NULL) {
        $set = ShortcutSet::create([
            'id' => $id ?? strtolower($this->randomMachineName()),
            'label' => empty($label) ? $this->randomString() : $label,
        ]);
        $set->save();
        return $set;
    }
    
    /**
     * Extracts information from shortcut set links.
     *
     * @param \Drupal\shortcut\ShortcutSetInterface $set
     *   The shortcut set object to extract information from.
     * @param string $key
     *   The array key indicating what information to extract from each link:
     *    - 'title': Extract shortcut titles.
     *    - 'link': Extract shortcut paths.
     *    - 'id': Extract the shortcut ID.
     *
     * @return array
     *   Array of the requested information from each link.
     */
    public function getShortcutInformation(ShortcutSetInterface $set, $key) {
        $info = [];
        \Drupal::entityTypeManager()->getStorage('shortcut')
            ->resetCache();
        foreach ($set->getShortcuts() as $shortcut) {
            if ($key == 'link') {
                $info[] = $shortcut->link->uri;
            }
            else {
                $info[] = $shortcut->{$key}->value;
            }
        }
        return $info;
    }

}