MediaMappingUpdateTest.php

<?php

declare (strict_types=1);
namespace Drupal\Tests\media\Functional\Update;

use Drupal\FunctionalTests\Update\UpdatePathTestBase;
use Drupal\media\Entity\MediaType;
use Drupal\media\Plugin\media\Source\File;
use Drupal\Tests\media\Traits\MediaTypeCreationTrait;

/**
 * Tests update functions for the Media module.
 *
 * @group media
 */
class MediaMappingUpdateTest extends UpdatePathTestBase {
    use MediaTypeCreationTrait;
    
    /**
     * {@inheritdoc}
     */
    protected function setDatabaseDumpFiles() : void {
        $this->databaseDumpFiles = [
            $this->getDrupalRoot() . '/core/modules/system/tests/fixtures/update/drupal-9.4.0.bare.standard.php.gz',
            __DIR__ . '/../../../fixtures/update/media.php',
        ];
    }
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        
        /** @var \Drupal\Core\Entity\EntityTypeManagerInterface $entity_type_manager */
        $entity_type_manager = $this->container
            ->get('entity_type.manager');
        
        /** @var \Drupal\Core\Entity\EntityDefinitionUpdateManagerInterface $update_manager */
        $update_manager = $this->container
            ->get('entity.definition_update_manager');
        $entity_type = $entity_type_manager->getDefinition('media_type');
        $update_manager->installEntityType($entity_type);
        $entity_type = $entity_type_manager->getDefinition('media');
        $update_manager->installEntityType($entity_type);
    }
    
    /**
     * Tests updating Media types using source field in meta mappings.
     *
     * @see media_post_update_remove_mappings_targeting_source_field()
     */
    public function testMediaMappingUpdate() : void {
        $media_type = $this->createMediaType('image', [
            'id' => 'invalid_mapping',
        ]);
        $source_field_name = $media_type->getSource()
            ->getSourceFieldDefinition($media_type)
            ->getName();
        $field_map = $media_type->getFieldMap();
        $field_map[File::METADATA_ATTRIBUTE_MIME] = $source_field_name;
        $this->config($media_type->getConfigDependencyName())
            ->set('field_map', $field_map)
            ->save();
        $this->runUpdates();
        $this->container
            ->get('entity_type.manager')
            ->getStorage('media_type')
            ->resetCache([
            'invalid_mapping',
        ]);
        $field_map = MediaType::load('invalid_mapping')?->getFieldMap();
        $this->assertIsArray($field_map);
        $this->assertNotContains($source_field_name, $field_map);
    }

}