MigrateEntityContentBaseTest.php

<?php

declare (strict_types=1);
namespace Drupal\Tests\migrate\Kernel;

use Drupal\Core\Entity\EntityFieldManager;
use Drupal\entity_test\Entity\EntityTestMul;
use Drupal\KernelTests\KernelTestBase;
use Drupal\language\Entity\ConfigurableLanguage;
use Drupal\migrate\MigrateExecutable;
use Drupal\migrate\Plugin\migrate\destination\EntityContentBase;
use Drupal\migrate\Plugin\MigrateIdMapInterface;
use Drupal\migrate\Plugin\MigrationInterface;
use Drupal\migrate\Row;
use Drupal\migrate_drupal\Tests\StubTestTrait;
use Drupal\migrate_entity_test\Entity\StringIdEntityTest;

/**
 * Tests the EntityContentBase destination.
 *
 * @group migrate
 */
class MigrateEntityContentBaseTest extends KernelTestBase {
    use StubTestTrait;
    
    /**
     * {@inheritdoc}
     */
    protected static $modules = [
        'migrate',
        'user',
        'language',
        'entity_test',
    ];
    
    /**
     * The storage for entity_test_mul.
     *
     * @var \Drupal\Core\Entity\ContentEntityStorageInterface
     */
    protected $storage;
    
    /**
     * A content migrate destination.
     *
     * @var \Drupal\migrate\Plugin\MigrateDestinationInterface
     */
    protected $destination;
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        // Enable two required fields with default values: a single-value field and
        // a multi-value field.
        \Drupal::state()->set('entity_test.required_default_field', TRUE);
        \Drupal::state()->set('entity_test.required_multi_default_field', TRUE);
        $this->installEntitySchema('entity_test_mul');
        ConfigurableLanguage::createFromLangcode('en')->save();
        ConfigurableLanguage::createFromLangcode('fr')->save();
        $this->storage = $this->container
            ->get('entity_type.manager')
            ->getStorage('entity_test_mul');
    }
    
    /**
     * Check the existing translations of an entity.
     *
     * @param int $id
     *   The entity ID.
     * @param string $default
     *   The expected default translation language code.
     * @param string[] $others
     *   The expected other translation language codes.
     *
     * @internal
     */
    protected function assertTranslations(int $id, string $default, array $others = []) : void {
        $entity = $this->storage
            ->load($id);
        $this->assertNotEmpty($entity, "Entity exists");
        $this->assertEquals($default, $entity->language()
            ->getId(), "Entity default translation");
        $translations = array_keys($entity->getTranslationLanguages(FALSE));
        sort($others);
        sort($translations);
        $this->assertEquals($others, $translations, "Entity translations");
    }
    
    /**
     * Create the destination plugin to test.
     *
     * @param array $configuration
     *   The plugin configuration.
     */
    protected function createDestination(array $configuration) {
        $this->destination = new EntityContentBase($configuration, 'fake_plugin_id', [], $this->createMock(MigrationInterface::class), $this->storage, [], $this->container
            ->get('entity_field.manager'), $this->container
            ->get('plugin.manager.field.field_type'), $this->container
            ->get('account_switcher'));
    }
    
    /**
     * Tests importing and rolling back translated entities.
     */
    public function testTranslated() : void {
        // Create a destination.
        $this->createDestination([
            'translations' => TRUE,
        ]);
        // Create some pre-existing entities.
        $this->storage
            ->create([
            'id' => 1,
            'langcode' => 'en',
        ])
            ->save();
        $this->storage
            ->create([
            'id' => 2,
            'langcode' => 'fr',
        ])
            ->save();
        $translated = $this->storage
            ->create([
            'id' => 3,
            'langcode' => 'en',
        ]);
        $translated->save();
        $translated->addTranslation('fr')
            ->save();
        // Pre-assert that things are as expected.
        $this->assertTranslations(1, 'en');
        $this->assertTranslations(2, 'fr');
        $this->assertTranslations(3, 'en', [
            'fr',
        ]);
        $this->assertNull($this->storage
            ->load(4));
        $destination_rows = [
            // Existing default translation.
[
                'id' => 1,
                'langcode' => 'en',
                'action' => MigrateIdMapInterface::ROLLBACK_PRESERVE,
            ],
            // New translation.
[
                'id' => 2,
                'langcode' => 'en',
                'action' => MigrateIdMapInterface::ROLLBACK_DELETE,
            ],
            // Existing non-default translation.
[
                'id' => 3,
                'langcode' => 'fr',
                'action' => MigrateIdMapInterface::ROLLBACK_PRESERVE,
            ],
            // Brand new row.
[
                'id' => 4,
                'langcode' => 'fr',
                'action' => MigrateIdMapInterface::ROLLBACK_DELETE,
            ],
        ];
        $rollback_actions = [];
        // Import some rows.
        foreach ($destination_rows as $idx => $destination_row) {
            $row = new Row();
            foreach ($destination_row as $key => $value) {
                $row->setDestinationProperty($key, $value);
            }
            $this->destination
                ->import($row);
            // Check that the rollback action is correct, and save it.
            $this->assertEquals($destination_row['action'], $this->destination
                ->rollbackAction());
            $rollback_actions[$idx] = $this->destination
                ->rollbackAction();
        }
        $this->assertTranslations(1, 'en');
        $this->assertTranslations(2, 'fr', [
            'en',
        ]);
        $this->assertTranslations(3, 'en', [
            'fr',
        ]);
        $this->assertTranslations(4, 'fr');
        // Rollback the rows.
        foreach ($destination_rows as $idx => $destination_row) {
            if ($rollback_actions[$idx] == MigrateIdMapInterface::ROLLBACK_DELETE) {
                $this->destination
                    ->rollback($destination_row);
            }
        }
        // No change, update of existing translation.
        $this->assertTranslations(1, 'en');
        // Remove added translation.
        $this->assertTranslations(2, 'fr');
        // No change, update of existing translation.
        $this->assertTranslations(3, 'en', [
            'fr',
        ]);
        // No change, can't remove default translation.
        $this->assertTranslations(4, 'fr');
    }
    
    /**
     * Tests creation of ID columns table with definitions taken from entity type.
     */
    public function testEntityWithStringId() : void {
        $this->enableModules([
            'migrate_entity_test',
        ]);
        $this->installEntitySchema('migrate_string_id_entity_test');
        $definition = [
            'source' => [
                'plugin' => 'embedded_data',
                'data_rows' => [
                    [
                        'id' => 123,
                        'version' => 'foo',
                    ],
                    // This integer needs an 'int' schema with 'big' size. If 'destid1'
                    // is not correctly taking the definition from the destination entity
                    // type, the import will fail with a SQL exception.
[
                        'id' => 123456789012,
                        'version' => 'bar',
                    ],
                ],
                'ids' => [
                    'id' => [
                        'type' => 'integer',
                        'size' => 'big',
                    ],
                    'version' => [
                        'type' => 'string',
                    ],
                ],
            ],
            'process' => [
                'id' => 'id',
                'version' => 'version',
            ],
            'destination' => [
                'plugin' => 'entity:migrate_string_id_entity_test',
            ],
        ];
        $migration = \Drupal::service('plugin.manager.migration')->createStubMigration($definition);
        $executable = new MigrateExecutable($migration);
        $result = $executable->import();
        $this->assertEquals(MigrationInterface::RESULT_COMPLETED, $result);
        
        /** @var \Drupal\migrate\Plugin\MigrateIdMapInterface $id_map_plugin */
        $id_map_plugin = $migration->getIdMap();
        // Check that the destination has been stored.
        $map_row = $id_map_plugin->getRowBySource([
            'id' => 123,
            'version' => 'foo',
        ]);
        $this->assertEquals(123, $map_row['destid1']);
        $map_row = $id_map_plugin->getRowBySource([
            'id' => 123456789012,
            'version' => 'bar',
        ]);
        $this->assertEquals(123456789012, $map_row['destid1']);
    }
    
    /**
     * Tests empty destinations.
     */
    public function testEmptyDestinations() : void {
        $this->enableModules([
            'migrate_entity_test',
        ]);
        $this->installEntitySchema('migrate_string_id_entity_test');
        $definition = [
            'source' => [
                'plugin' => 'embedded_data',
                'data_rows' => [
                    [
                        'id' => 123,
                        'version' => 'foo',
                    ],
                    // This integer needs an 'int' schema with 'big' size. If 'destid1'
                    // is not correctly taking the definition from the destination entity
                    // type, the import will fail with an SQL exception.
[
                        'id' => 123456789012,
                        'version' => 'bar',
                    ],
                ],
                'ids' => [
                    'id' => [
                        'type' => 'integer',
                        'size' => 'big',
                    ],
                    'version' => [
                        'type' => 'string',
                    ],
                ],
                'constants' => [
                    'null' => NULL,
                ],
            ],
            'process' => [
                'id' => 'id',
                'version' => 'version',
            ],
            'destination' => [
                'plugin' => 'entity:migrate_string_id_entity_test',
            ],
        ];
        $migration = \Drupal::service('plugin.manager.migration')->createStubMigration($definition);
        $executable = new MigrateExecutable($migration);
        $executable->import();
        
        /** @var \Drupal\migrate_entity_test\Entity\StringIdEntityTest $entity */
        $entity = StringIdEntityTest::load('123');
        $this->assertSame('foo', $entity->version->value);
        $entity = StringIdEntityTest::load('123456789012');
        $this->assertSame('bar', $entity->version->value);
        // Rerun the migration forcing the version to NULL.
        $definition['process'] = [
            'id' => 'id',
            'version' => 'constants/null',
        ];
        $migration = \Drupal::service('plugin.manager.migration')->createStubMigration($definition);
        $executable = new MigrateExecutable($migration);
        $executable->import();
        
        /** @var \Drupal\migrate_entity_test\Entity\StringIdEntityTest $entity */
        $entity = StringIdEntityTest::load('123');
        $this->assertNull($entity->version->value);
        $entity = StringIdEntityTest::load('123456789012');
        $this->assertNull($entity->version->value);
    }
    
    /**
     * Tests stub rows.
     */
    public function testStubRows() : void {
        // Create a destination.
        $this->createDestination([]);
        // Import a stub row.
        $row = new Row([], [], TRUE);
        $row->setDestinationProperty('type', 'test');
        $ids = $this->destination
            ->import($row);
        $this->assertCount(1, $ids);
        // Make sure the entity was saved.
        $entity = EntityTestMul::load(reset($ids));
        $this->assertInstanceOf(EntityTestMul::class, $entity);
        // Make sure the default value was applied to the required fields.
        $single_field_name = 'required_default_field';
        $single_default_value = $entity->getFieldDefinition($single_field_name)
            ->getDefaultValueLiteral();
        $this->assertSame($single_default_value, $entity->get($single_field_name)
            ->getValue());
        $multi_field_name = 'required_multi_default_field';
        $multi_default_value = $entity->getFieldDefinition($multi_field_name)
            ->getDefaultValueLiteral();
        $count = 3;
        $this->assertCount($count, $multi_default_value);
        for ($i = 0; $i < $count; ++$i) {
            $this->assertSame($multi_default_value[$i], $entity->get($multi_field_name)
                ->get($i)
                ->getValue());
        }
    }
    
    /**
     * Tests bundle is properly provided for stubs without bundle support.
     *
     * @todo Remove this test in when native PHP type-hints will be added for
     *   EntityFieldManagerInterface::getFieldDefinitions(). See
     *   https://www.drupal.org/project/drupal/issues/3050720.
     */
    public function testBundleFallbackForStub() : void {
        $this->enableModules([
            'migrate_entity_test',
        ]);
        $this->installEntitySchema('migrate_string_id_entity_test');
        $entity_type_manager = $this->container
            ->get('entity_type.manager');
        $entity_type_bundle_info = $this->container
            ->get('entity_type.bundle.info');
        $entity_display_repository = $this->container
            ->get('entity_display.repository');
        $typed_data_manager = $this->container
            ->get('typed_data_manager');
        $language_manager = $this->container
            ->get('language_manager');
        $keyvalue = $this->container
            ->get('keyvalue');
        $module_handler = $this->container
            ->get('module_handler');
        $cache_discovery = $this->container
            ->get('cache.discovery');
        $entity_last_installed_schema_repository = $this->container
            ->get('entity.last_installed_schema.repository');
        $decorated_entity_field_manager = new class ($entity_type_manager, $entity_type_bundle_info, $entity_display_repository, $typed_data_manager, $language_manager, $keyvalue, $module_handler, $cache_discovery, $entity_last_installed_schema_repository) extends EntityFieldManager {
            
            /**
             * {@inheritdoc}
             */
            public function getFieldDefinitions($entity_type_id, $bundle) {
                if (\is_null($bundle)) {
                    throw new \Exception("Bundle value shouldn't be NULL.");
                }
                return parent::getFieldDefinitions($entity_type_id, $bundle);
            }

};
        $this->container
            ->set('entity_field.manager', $decorated_entity_field_manager);
        $this->createEntityStub('migrate_string_id_entity_test');
    }

}