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']);
    }

}