MigrateSqlIdMapEnsureTablesTest.php in Zircon Profile 8

Contains \Drupal\Tests\migrate\Unit\MigrateSqlIdMapEnsureTablesTest.

Namespace

Drupal\Tests\migrate\Unit

File

core/modules/migrate/tests/src/Unit/MigrateSqlIdMapEnsureTablesTest.php

<?php

/**
 * @file
 * Contains \Drupal\Tests\migrate\Unit\MigrateSqlIdMapEnsureTablesTest.
 */
namespace Drupal\Tests\migrate\Unit;

use Drupal\migrate\Plugin\MigrateIdMapInterface;

/**
 * Tests the SQL ID map plugin ensureTables() method.
 *
 * @group migrate
 */
class MigrateSqlIdMapEnsureTablesTest extends MigrateTestCase {

  /**
   * The migration configuration, initialized to set the ID and destination IDs.
   *
   * @var array
   */
  protected $migrationConfiguration = array(
    'id' => 'sql_idmap_test',
  );

  /**
   * Tests the ensureTables method when the tables do not exist.
   */
  public function testEnsureTablesNotExist() {
    $fields['source_row_status'] = array(
      'type' => 'int',
      'size' => 'tiny',
      'unsigned' => TRUE,
      'not null' => TRUE,
      'default' => MigrateIdMapInterface::STATUS_IMPORTED,
      'description' => 'Indicates current status of the source row',
    );
    $fields['rollback_action'] = array(
      'type' => 'int',
      'size' => 'tiny',
      'unsigned' => TRUE,
      'not null' => TRUE,
      'default' => MigrateIdMapInterface::ROLLBACK_DELETE,
      'description' => 'Flag indicating what to do for this item on rollback',
    );
    $fields['last_imported'] = array(
      'type' => 'int',
      'unsigned' => TRUE,
      'not null' => TRUE,
      'default' => 0,
      'description' => 'UNIX timestamp of the last time this row was imported',
    );
    $fields['hash'] = array(
      'type' => 'varchar',
      'length' => '64',
      'not null' => FALSE,
      'description' => 'Hash of source row data, for detecting changes',
    );
    $fields['sourceid1'] = array(
      'type' => 'int',
      'not null' => TRUE,
    );
    $fields['destid1'] = array(
      'type' => 'varchar',
      'length' => 255,
      'not null' => FALSE,
    );
    $map_table_schema = array(
      'description' => 'Mappings from source identifier value(s) to destination identifier value(s).',
      'fields' => $fields,
      'primary key' => array(
        'sourceid1',
      ),
    );
    $schema = $this
      ->getMockBuilder('Drupal\\Core\\Database\\Schema')
      ->disableOriginalConstructor()
      ->getMock();
    $schema
      ->expects($this
      ->at(0))
      ->method('tableExists')
      ->with('migrate_map_sql_idmap_test')
      ->will($this
      ->returnValue(FALSE));
    $schema
      ->expects($this
      ->at(1))
      ->method('createTable')
      ->with('migrate_map_sql_idmap_test', $map_table_schema);

    // Now do the message table.
    $fields = array();
    $fields['msgid'] = array(
      'type' => 'serial',
      'unsigned' => TRUE,
      'not null' => TRUE,
    );
    $fields['sourceid1'] = array(
      'type' => 'int',
      'not null' => TRUE,
    );
    $fields['level'] = array(
      'type' => 'int',
      'unsigned' => TRUE,
      'not null' => TRUE,
      'default' => 1,
    );
    $fields['message'] = array(
      'type' => 'text',
      'size' => 'medium',
      'not null' => TRUE,
    );
    $table_schema = array(
      'description' => 'Messages generated during a migration process',
      'fields' => $fields,
      'primary key' => array(
        'msgid',
      ),
    );
    $table_schema['indexes']['sourcekey'] = array(
      'sourceid1',
    );
    $schema
      ->expects($this
      ->at(2))
      ->method('tableExists')
      ->with('migrate_message_sql_idmap_test')
      ->will($this
      ->returnValue(FALSE));
    $schema
      ->expects($this
      ->at(3))
      ->method('createTable')
      ->with('migrate_message_sql_idmap_test', $table_schema);
    $schema
      ->expects($this
      ->any())
      ->method($this
      ->anything());
    $this
      ->runEnsureTablesTest($schema);
  }

  /**
   * Tests the ensureTables method when the tables exist.
   */
  public function testEnsureTablesExist() {
    $schema = $this
      ->getMockBuilder('Drupal\\Core\\Database\\Schema')
      ->disableOriginalConstructor()
      ->getMock();
    $schema
      ->expects($this
      ->at(0))
      ->method('tableExists')
      ->with('migrate_map_sql_idmap_test')
      ->will($this
      ->returnValue(TRUE));
    $schema
      ->expects($this
      ->at(1))
      ->method('fieldExists')
      ->with('migrate_map_sql_idmap_test', 'rollback_action')
      ->will($this
      ->returnValue(FALSE));
    $field_schema = array(
      'type' => 'int',
      'size' => 'tiny',
      'unsigned' => TRUE,
      'not null' => TRUE,
      'default' => 0,
      'description' => 'Flag indicating what to do for this item on rollback',
    );
    $schema
      ->expects($this
      ->at(2))
      ->method('addField')
      ->with('migrate_map_sql_idmap_test', 'rollback_action', $field_schema);
    $schema
      ->expects($this
      ->at(3))
      ->method('fieldExists')
      ->with('migrate_map_sql_idmap_test', 'hash')
      ->will($this
      ->returnValue(FALSE));
    $field_schema = array(
      'type' => 'varchar',
      'length' => '64',
      'not null' => FALSE,
      'description' => 'Hash of source row data, for detecting changes',
    );
    $schema
      ->expects($this
      ->at(4))
      ->method('addField')
      ->with('migrate_map_sql_idmap_test', 'hash', $field_schema);
    $schema
      ->expects($this
      ->exactly(5))
      ->method($this
      ->anything());
    $this
      ->runEnsureTablesTest($schema);
  }

  /**
   * Actually run the test.
   *
   * @param array $schema
   *   The mock schema object with expectations set. The Sql constructor calls
   *   ensureTables() which in turn calls this object and the expectations on
   *   it are the actual test and there are no additional asserts added.
   */
  protected function runEnsureTablesTest($schema) {
    $database = $this
      ->getMockBuilder('Drupal\\Core\\Database\\Connection')
      ->disableOriginalConstructor()
      ->getMock();
    $database
      ->expects($this
      ->any())
      ->method('schema')
      ->willReturn($schema);
    $migration = $this
      ->getMigration();
    $plugin = $this
      ->getMock('Drupal\\migrate\\Plugin\\MigrateSourceInterface');
    $plugin
      ->expects($this
      ->any())
      ->method('getIds')
      ->willReturn(array(
      'source_id_property' => array(
        'type' => 'integer',
      ),
    ));
    $migration
      ->expects($this
      ->any())
      ->method('getSourcePlugin')
      ->willReturn($plugin);
    $plugin = $this
      ->getMock('Drupal\\migrate\\Plugin\\MigrateSourceInterface');
    $plugin
      ->expects($this
      ->any())
      ->method('getIds')
      ->willReturn(array(
      'destination_id_property' => array(
        'type' => 'string',
      ),
    ));
    $migration
      ->expects($this
      ->any())
      ->method('getDestinationPlugin')
      ->willReturn($plugin);

    /** @var \Symfony\Component\EventDispatcher\EventDispatcherInterface $event_dispatcher */
    $event_dispatcher = $this
      ->getMock('Symfony\\Component\\EventDispatcher\\EventDispatcherInterface');
    $map = new TestSqlIdMap($database, array(), 'sql', array(), $migration, $event_dispatcher);
    $map
      ->getDatabase();
  }

}

Classes

Name	Description
MigrateSqlIdMapEnsureTablesTest	Tests the SQL ID map plugin ensureTables() method.