class MigrateEmbeddedDataTest

class MigrateEmbeddedDataTest extends KernelTestBase {
    
    /**
     * Modules to enable.
     *
     * @var array
     */
    protected static $modules = [
        'migrate',
    ];
    
    /**
     * Tests the embedded_data source plugin.
     */
    public function testEmbeddedData() {
        $data_rows = [
            [
                'key' => '1',
                'field1' => 'f1value1',
                'field2' => 'f2value1',
            ],
            [
                'key' => '2',
                'field1' => 'f1value2',
                'field2' => 'f2value2',
            ],
        ];
        $ids = [
            'key' => [
                'type' => 'integer',
            ],
        ];
        $definition = [
            'migration_tags' => [
                'Embedded data test',
            ],
            'source' => [
                'plugin' => 'embedded_data',
                'data_rows' => $data_rows,
                'ids' => $ids,
            ],
            'process' => [],
            'destination' => [
                'plugin' => 'null',
            ],
        ];
        $migration = \Drupal::service('plugin.manager.migration')->createStubMigration($definition);
        $source = $migration->getSourcePlugin();
        // Validate the plugin returns the source data that was provided.
        $results = [];
        
        /** @var \Drupal\migrate\Row $row */
        foreach ($source as $row) {
            $this->assertFalse($row->isStub());
            $data_row = $row->getSource();
            // The "data" row returned by getSource() also includes all source
            // configuration - we remove it so we see only the data itself.
            unset($data_row['plugin']);
            unset($data_row['data_rows']);
            unset($data_row['ids']);
            $results[] = $data_row;
        }
        $this->assertSame($data_rows, $results);
        // Validate the public APIs.
        $this->assertSame(count($data_rows), $source->count());
        $this->assertSame($ids, $source->getIds());
        $expected_fields = [
            'key' => 'key',
            'field1' => 'field1',
            'field2' => 'field2',
        ];
        $this->assertSame($expected_fields, $source->fields());
    }

}