Chris@0: <?php
Chris@0: 
Chris@0: namespace Drupal\Tests\migrate\Unit;
Chris@0: 
Chris@0: use Drupal\Core\Database\Connection;
Chris@0: use Drupal\migrate\Plugin\MigrationInterface;
Chris@0: use Drupal\migrate\MigrateException;
Chris@0: use Drupal\migrate\Plugin\migrate\id_map\Sql;
Chris@0: use Symfony\Component\EventDispatcher\EventDispatcherInterface;
Chris@0: 
Chris@0: /**
Chris@0:  * Defines a SQL ID map for use in tests.
Chris@0:  */
Chris@0: class TestSqlIdMap extends Sql implements \Iterator {
Chris@0: 
Chris@0:   /**
Chris@0:    * Constructs a TestSqlIdMap object.
Chris@0:    *
Chris@0:    * @param \Drupal\Core\Database\Connection $database
Chris@0:    *   The database.
Chris@0:    * @param array $configuration
Chris@0:    *   The configuration.
Chris@0:    * @param string $plugin_id
Chris@0:    *   The plugin ID for the migration process to do.
Chris@0:    * @param mixed $plugin_definition
Chris@0:    *   The configuration for the plugin.
Chris@0:    * @param \Drupal\migrate\Plugin\MigrationInterface $migration
Chris@0:    *   The migration to do.
Chris@0:    * @param \Symfony\Component\EventDispatcher\EventDispatcherInterface $event_dispatcher
Chris@0:    *   The event dispatcher service.
Chris@0:    */
Chris@0:   public function __construct(Connection $database, array $configuration, $plugin_id, $plugin_definition, MigrationInterface $migration, EventDispatcherInterface $event_dispatcher) {
Chris@0:     $this->database = $database;
Chris@0:     parent::__construct($configuration, $plugin_id, $plugin_definition, $migration, $event_dispatcher);
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * {@inheritdoc}
Chris@0:    */
Chris@0:   public function getDatabase() {
Chris@0:     return parent::getDatabase();
Chris@0:   }
Chris@0: 
Chris@0:   /**
Chris@0:    * Gets the field schema.
Chris@0:    *
Chris@0:    * @param array $id_definition
Chris@0:    *   An array defining the field, with a key 'type'.
Chris@0:    *
Chris@0:    * @return array
Chris@0:    *   A field schema depending on value of key 'type'.  An empty array is
Chris@0:    *   returned if 'type' is not defined.
Chris@0:    *
Chris@0:    * @throws \Drupal\migrate\MigrateException
Chris@0:    */
Chris@0:   protected function getFieldSchema(array $id_definition) {
Chris@0:     if (!isset($id_definition['type'])) {
Chris@0:       return [];
Chris@0:     }
Chris@0:     switch ($id_definition['type']) {
Chris@0:       case 'integer':
Chris@0:         return [
Chris@0:           'type' => 'int',
Chris@0:           'not null' => TRUE,
Chris@0:         ];
Chris@0: 
Chris@0:       case 'string':
Chris@0:         return [
Chris@0:           'type' => 'varchar',
Chris@0:           'length' => 255,
Chris@0:           'not null' => FALSE,
Chris@0:         ];
Chris@0: 
Chris@0:       default:
Chris@0:         throw new MigrateException($id_definition['type'] . ' not supported');
Chris@0:     }
Chris@0:   }
Chris@0: 
Chris@0: }