FollowUpMigrationsTest.php

<?php

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

use Drupal\node\Entity\Node;
use Drupal\Tests\file\Kernel\Migrate\d7\FileMigrationSetupTrait;
use Drupal\user\Entity\User;

/**
 * Tests follow-up migrations.
 *
 * @group migrate_drupal
 */
class FollowUpMigrationsTest extends MigrateDrupal7TestBase {
    use FileMigrationSetupTrait;
    
    /**
     * {@inheritdoc}
     */
    protected static $modules = [
        'content_translation',
        'comment',
        'datetime',
        'datetime_range',
        'image',
        'language',
        'link',
        'menu_ui',
        'node',
        'taxonomy',
        'telephone',
        'text',
    ];
    
    /**
     * {@inheritdoc}
     */
    protected function setUp() : void {
        parent::setUp();
        $this->fileMigrationSetup();
        $this->installEntitySchema('comment');
        $this->installSchema('node', [
            'node_access',
        ]);
        $this->migrateFields();
        $this->migrateUsers();
        $this->executeMigrations([
            'language',
            'd7_language_content_settings',
            'd7_taxonomy_vocabulary',
        ]);
    }
    
    /**
     * {@inheritdoc}
     */
    protected function getFileMigrationInfo() : array {
        return [
            'path' => 'public://sites/default/files/cube.jpeg',
            'size' => 3620,
            'base_path' => 'public://',
            'plugin_id' => 'd7_file',
        ];
    }
    
    /**
     * Tests entity reference translations.
     *
     * @dataProvider providerTestEntityReferenceTranslations
     */
    public function testEntityReferenceTranslations($node_migrations) : void {
        $this->executeMigrations($node_migrations);
        // Test the entity reference field before the follow-up migrations.
        $node = Node::load(2);
        $this->assertSame('5', $node->get('field_reference')->target_id);
        $this->assertSame('6', $node->get('field_reference_2')->target_id);
        $translation = $node->getTranslation('is');
        $this->assertSame('4', $translation->get('field_reference')->target_id);
        $this->assertSame('4', $translation->get('field_reference_2')->target_id);
        $node = Node::load(4);
        $this->assertSame('3', $node->get('field_reference')->target_id);
        $this->assertSame('3', $node->get('field_reference_2')->target_id);
        $translation = $node->getTranslation('en');
        $this->assertSame('2', $translation->get('field_reference')->target_id);
        $this->assertSame('2', $translation->get('field_reference_2')->target_id);
        $user = User::load(2);
        $this->assertSame('3', $user->get('field_reference')->target_id);
        // Run the follow-up migrations.
        $migration_plugin_manager = $this->container
            ->get('plugin.manager.migration');
        $migration_plugin_manager->clearCachedDefinitions();
        $follow_up_migrations = $migration_plugin_manager->createInstances('d7_entity_reference_translation');
        $this->executeMigrations(array_keys($follow_up_migrations));
        // Test the entity reference field after the follow-up migrations.
        $node = Node::load(2);
        $this->assertSame('4', $node->get('field_reference')->target_id);
        $this->assertSame('6', $node->get('field_reference_2')->target_id);
        $translation = $node->getTranslation('is');
        $this->assertSame('4', $translation->get('field_reference')->target_id);
        $this->assertSame('4', $translation->get('field_reference_2')->target_id);
        $node = Node::load(4);
        $this->assertSame('2', $node->get('field_reference')->target_id);
        $this->assertSame('2', $node->get('field_reference_2')->target_id);
        $translation = $node->getTranslation('en');
        $this->assertSame('2', $translation->get('field_reference')->target_id);
        $this->assertSame('2', $translation->get('field_reference_2')->target_id);
        $user = User::load(2);
        $this->assertSame('2', $user->get('field_reference')->target_id);
    }
    
    /**
     * Data provider for testEntityReferenceTranslations().
     */
    public static function providerTestEntityReferenceTranslations() {
        return [
            [
                [
                    'd7_node',
                    'd7_node_translation',
                ],
            ],
            [
                [
                    'd7_node_complete',
                ],
            ],
        ];
    }

}