class YamlDiscoveryTest

class YamlDiscoveryTest extends UnitTestCase {
    
    /**
     * The YamlDiscovery instance to test.
     *
     * @var \Drupal\Core\Plugin\Discovery\YamlDiscovery
     */
    protected $discovery;
    
    /**
     * Expected provider => key mappings for testing.
     *
     * @var array
     */
    protected $expectedKeys = [
        'test_1' => 'test_1_a',
        'another_provider_1' => 'test_1_b',
        'another_provider_2' => 'test_2_a',
        'test_2' => 'test_2_b',
    ];
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        $base_path = __DIR__ . '/Fixtures';
        // Set up the directories to search.
        $directories = [
            'test_1' => $base_path . '/test_1',
            'test_2' => $base_path . '/test_2',
        ];
        $this->discovery = new YamlDiscovery('test', $directories);
    }
    
    /**
     * Tests the getDefinitions() method.
     */
    public function testGetDefinitions() {
        $definitions = $this->discovery
            ->getDefinitions();
        $this->assertIsArray($definitions);
        $this->assertCount(4, $definitions);
        foreach ($this->expectedKeys as $expected_key) {
            $this->assertArrayHasKey($expected_key, $definitions);
        }
        foreach ($definitions as $id => $definition) {
            foreach ([
                'name',
                'id',
                'provider',
            ] as $key) {
                $this->assertArrayHasKey($key, $definition);
            }
            $this->assertEquals($id, $definition['id']);
            $this->assertEquals(array_search($id, $this->expectedKeys), $definition['provider']);
        }
    }
    
    /**
     * @covers ::getDefinitions
     */
    public function testGetDefinitionsWithTranslatableDefinitions() {
        vfsStream::setup('root');
        $file_1 = <<<'EOS'
test_plugin:
  title: test title
EOS;
        $file_2 = <<<'EOS'
test_plugin2:
  title: test title2
  title_context: 'test-context'
EOS;
        vfsStream::create([
            'test_1' => [
                'test_1.test.yml' => $file_1,
            ],
            'test_2' => [
                'test_2.test.yml' => $file_2,
            ],
        ]);
        $discovery = new YamlDiscovery('test', [
            'test_1' => vfsStream::url('root/test_1'),
            'test_2' => vfsStream::url('root/test_2'),
        ]);
        $discovery->addTranslatableProperty('title', 'title_context');
        $definitions = $discovery->getDefinitions();
        $this->assertCount(2, $definitions);
        $plugin_1 = $definitions['test_plugin'];
        $plugin_2 = $definitions['test_plugin2'];
        $this->assertInstanceOf(TranslatableMarkup::class, $plugin_1['title']);
        $this->assertEquals([], $plugin_1['title']->getOptions());
        $this->assertInstanceOf(TranslatableMarkup::class, $plugin_2['title']);
        $this->assertEquals([
            'context' => 'test-context',
        ], $plugin_2['title']->getOptions());
    }
    
    /**
     * Tests the getDefinition() method.
     */
    public function testGetDefinition() {
        $definitions = $this->discovery
            ->getDefinitions();
        // Test the getDefinition() method.
        foreach ($this->expectedKeys as $expected_key) {
            $this->assertEquals($definitions[$expected_key], $this->discovery
                ->getDefinition($expected_key));
        }
    }

}