class CategorizingPluginManagerTraitTest

class CategorizingPluginManagerTraitTest extends UnitTestCase {
  
  /**
   * The plugin manager to test.
   *
   * @var \Drupal\Component\Plugin\CategorizingPluginManagerInterface|\PHPUnit\Framework\MockObject\MockObject
   */
  protected $pluginManager;
  
  /**
   * {@inheritdoc}
   */
  protected function setUp() : void {
    parent::setUp();
    $module_handler = $this->createMock('Drupal\\Core\\Extension\\ModuleHandlerInterface');
    $module_handler->expects($this->any())
      ->method('getModuleList')
      ->willReturn([
      'node' => [],
    ]);
    $module_extension_list = $this->createMock(ModuleExtensionList::class);
    $module_extension_list->expects($this->any())
      ->method('getName')
      ->willReturnCallback(function ($argument) {
      if ($argument == 'node') {
        return 'Node';
      }
      throw new UnknownExtensionException();
    });
    $this->pluginManager = new CategorizingPluginManager($module_handler, $module_extension_list);
    $this->pluginManager
      ->setStringTranslation($this->getStringTranslationStub());
  }
  
  /**
   * @covers ::getCategories
   */
  public function testGetCategories() : void {
    $this->assertSame([
      'fruits',
      'vegetables',
    ], array_values($this->pluginManager
      ->getCategories()));
  }
  
  /**
   * @covers ::getSortedDefinitions
   */
  public function testGetSortedDefinitions() : void {
    $sorted = $this->pluginManager
      ->getSortedDefinitions();
    $this->assertSame([
      'apple',
      'mango',
      'cucumber',
    ], array_keys($sorted));
  }
  
  /**
   * @covers ::getGroupedDefinitions
   */
  public function testGetGroupedDefinitions() : void {
    $grouped = $this->pluginManager
      ->getGroupedDefinitions();
    $this->assertSame([
      'fruits',
      'vegetables',
    ], array_keys($grouped));
    $this->assertSame([
      'apple',
      'mango',
    ], array_keys($grouped['fruits']));
    $this->assertSame([
      'cucumber',
    ], array_keys($grouped['vegetables']));
  }
  
  /**
   * @covers ::processDefinitionCategory
   */
  public function testProcessDefinitionCategory() : void {
    // Existing category.
    $definition = [
      'label' => 'some',
      'provider' => 'core',
      'category' => 'bag',
    ];
    $this->pluginManager
      ->processDefinition($definition, 'some');
    $this->assertSame('bag', $definition['category']);
    // No category, provider without label.
    $definition = [
      'label' => 'some',
      'provider' => 'core',
    ];
    $this->pluginManager
      ->processDefinition($definition, 'some');
    $this->assertSame('core', $definition['category']);
    // No category, provider is module with label.
    $definition = [
      'label' => 'some',
      'provider' => 'node',
    ];
    $this->pluginManager
      ->processDefinition($definition, 'some');
    $this->assertSame('Node', $definition['category']);
  }

}