Same name in this branch
  1. 10 core/modules/migrate/tests/src/Kernel/SqlBaseTest.php \Drupal\Tests\migrate\Kernel\SqlBaseTest
  2. 10 core/modules/migrate/tests/src/Unit/SqlBaseTest.php \Drupal\Tests\migrate\Unit\SqlBaseTest
  3. 10 core/modules/views/tests/src/Unit/Plugin/pager/SqlBaseTest.php \Drupal\Tests\views\Unit\Plugin\pager\SqlBaseTest
Same name and namespace in other branches
  1. 8.9.x core/modules/migrate/tests/src/Kernel/SqlBaseTest.php \Drupal\Tests\migrate\Kernel\SqlBaseTest
  2. 9 core/modules/migrate/tests/src/Kernel/SqlBaseTest.php \Drupal\Tests\migrate\Kernel\SqlBaseTest

Tests the functionality of SqlBase.

@group migrate

Hierarchy

Expanded class hierarchy of SqlBaseTest

File

core/modules/migrate/tests/src/Kernel/SqlBaseTest.php, line 18

Namespace

Drupal\Tests\migrate\Kernel
View source
class SqlBaseTest extends MigrateTestBase {

  /**
   * The (probably mocked) migration under test.
   *
   * @var \Drupal\migrate\Plugin\MigrationInterface
   */
  protected $migration;

  /**
   * {@inheritdoc}
   */
  protected function setUp() : void {
    parent::setUp();
    $this->migration = $this
      ->createMock(MigrationInterface::class);
    $this->migration
      ->method('id')
      ->willReturn('foo');
  }

  /**
   * Tests different connection types.
   */
  public function testConnectionTypes() {
    $sql_base = new TestSqlBase([], $this->migration);

    // Verify that falling back to the default 'migrate' connection (defined in
    // the base class) works.
    $this
      ->assertSame('default', $sql_base
      ->getDatabase()
      ->getTarget());
    $this
      ->assertSame('migrate', $sql_base
      ->getDatabase()
      ->getKey());

    // Verify the fallback state key overrides the 'migrate' connection.
    $target = 'test_fallback_target';
    $key = 'test_fallback_key';
    $config = [
      'target' => $target,
      'key' => $key,
    ];
    $database_state_key = 'test_fallback_state';
    \Drupal::state()
      ->set($database_state_key, $config);
    \Drupal::state()
      ->set('migrate.fallback_state_key', $database_state_key);

    // Create a test connection using the default database configuration.
    Database::addConnectionInfo($key, $target, Database::getConnectionInfo('default')['default']);
    $this
      ->assertSame($sql_base
      ->getDatabase()
      ->getTarget(), $target);
    $this
      ->assertSame($sql_base
      ->getDatabase()
      ->getKey(), $key);

    // Verify that setting explicit connection information overrides fallbacks.
    $target = 'test_db_target';
    $key = 'test_migrate_connection';
    $config = [
      'target' => $target,
      'key' => $key,
    ];
    $sql_base
      ->setConfiguration($config);
    Database::addConnectionInfo($key, $target, Database::getConnectionInfo('default')['default']);

    // Validate we have injected our custom key and target.
    $this
      ->assertSame($sql_base
      ->getDatabase()
      ->getTarget(), $target);
    $this
      ->assertSame($sql_base
      ->getDatabase()
      ->getKey(), $key);

    // Now test we can have SqlBase create the connection from an info array.
    $sql_base = new TestSqlBase([], $this->migration);
    $target = 'test_db_target2';
    $key = 'test_migrate_connection2';
    $database = Database::getConnectionInfo('default')['default'];
    $config = [
      'target' => $target,
      'key' => $key,
      'database' => $database,
    ];
    $sql_base
      ->setConfiguration($config);

    // Call getDatabase() to get the connection defined.
    $sql_base
      ->getDatabase();

    // Validate the connection has been created with the right values.
    $this
      ->assertSame(Database::getConnectionInfo($key)[$target], $database);

    // Now, test this all works when using state to store db info.
    $target = 'test_state_db_target';
    $key = 'test_state_migrate_connection';
    $config = [
      'target' => $target,
      'key' => $key,
    ];
    $database_state_key = 'migrate_sql_base_test';
    \Drupal::state()
      ->set($database_state_key, $config);
    $sql_base
      ->setConfiguration([
      'database_state_key' => $database_state_key,
    ]);
    Database::addConnectionInfo($key, $target, Database::getConnectionInfo('default')['default']);

    // Validate we have injected our custom key and target.
    $this
      ->assertSame($sql_base
      ->getDatabase()
      ->getTarget(), $target);
    $this
      ->assertSame($sql_base
      ->getDatabase()
      ->getKey(), $key);

    // Now test we can have SqlBase create the connection from an info array.
    $sql_base = new TestSqlBase([], $this->migration);
    $target = 'test_state_db_target2';
    $key = 'test_state_migrate_connection2';
    $database = Database::getConnectionInfo('default')['default'];
    $config = [
      'target' => $target,
      'key' => $key,
      'database' => $database,
    ];
    $database_state_key = 'migrate_sql_base_test2';
    \Drupal::state()
      ->set($database_state_key, $config);
    $sql_base
      ->setConfiguration([
      'database_state_key' => $database_state_key,
    ]);

    // Call getDatabase() to get the connection defined.
    $sql_base
      ->getDatabase();

    // Validate the connection has been created with the right values.
    $this
      ->assertSame(Database::getConnectionInfo($key)[$target], $database);

    // Verify that falling back to 'migrate' when the connection is not defined
    // throws a RequirementsException.
    \Drupal::state()
      ->delete('migrate.fallback_state_key');
    $sql_base
      ->setConfiguration([]);
    Database::renameConnection('migrate', 'fallback_connection');
    $this
      ->expectException(RequirementsException::class);
    $this
      ->expectExceptionMessage('No database connection configured for source plugin');
    $sql_base
      ->getDatabase();
  }

  /**
   * Tests the exception when a connection is defined but not available.
   */
  public function testBrokenConnection() : void {
    if (Database::getConnection()
      ->driver() === 'sqlite') {
      $this
        ->markTestSkipped('Not compatible with sqlite');
    }
    $sql_base = new TestSqlBase([], $this->migration);
    $target = 'test_state_db_target2';
    $key = 'test_state_migrate_connection2';
    $database = Database::getConnectionInfo('default')['default'];
    $database['database'] = 'godot';
    $config = [
      'target' => $target,
      'key' => $key,
      'database' => $database,
    ];
    $database_state_key = 'migrate_sql_base_test2';
    \Drupal::state()
      ->set($database_state_key, $config);
    $sql_base
      ->setConfiguration([
      'database_state_key' => $database_state_key,
    ]);

    // Call checkRequirements(): it will call getDatabase() and convert the
    // exception to a RequirementsException.
    $this
      ->expectException(RequirementsException::class);
    $this
      ->expectExceptionMessage('No database connection available for source plugin sql_base');
    $sql_base
      ->checkRequirements();
  }

  /**
   * Tests that SqlBase respects high-water values.
   *
   * @param mixed $high_water
   *   (optional) The high-water value to set.
   * @param array $query_result
   *   (optional) The expected query results.
   *
   * @dataProvider highWaterDataProvider
   */
  public function testHighWater($high_water = NULL, array $query_result = []) {
    $configuration = [
      'high_water_property' => [
        'name' => 'order',
      ],
    ];
    $source = new TestSqlBase($configuration, $this->migration);
    if ($high_water) {
      \Drupal::keyValue('migrate:high_water')
        ->set($this->migration
        ->id(), $high_water);
    }
    $statement = $this
      ->createMock(StatementInterface::class);
    $statement
      ->expects($this
      ->atLeastOnce())
      ->method('setFetchMode')
      ->with(\PDO::FETCH_ASSOC);
    $query = $this
      ->createMock(SelectInterface::class);
    $query
      ->method('execute')
      ->willReturn($statement);
    $query
      ->expects($this
      ->atLeastOnce())
      ->method('orderBy')
      ->with('order', 'ASC');
    $condition_group = $this
      ->createMock(ConditionInterface::class);
    $query
      ->method('orConditionGroup')
      ->willReturn($condition_group);
    $source
      ->setQuery($query);
    $source
      ->rewind();
  }

  /**
   * Data provider for ::testHighWater().
   *
   * @return array
   *   The scenarios to test.
   */
  public static function highWaterDataProvider() {
    return [
      'no high-water value set' => [],
      'high-water value set' => [
        33,
      ],
    ];
  }

}

Members

Namesort descending Modifiers Type Description Overrides
MigrateTestBase::$collectMessages protected property TRUE to collect messages instead of displaying them.
MigrateTestBase::$logger protected property A logger prophecy object. 2
MigrateTestBase::$migrateMessages protected property A two dimensional array of messages.
MigrateTestBase::$modules protected static property 16
MigrateTestBase::$sourceDatabase protected property The source database connection.
MigrateTestBase::cleanupMigrateConnection private function Cleans up the test migrate connection.
MigrateTestBase::createMigrationConnection private function Changes the database connection to the prefixed one.
MigrateTestBase::display public function Displays a migrate message. Overrides MigrateMessageInterface::display
MigrateTestBase::executeMigration protected function Executes a single migration.
MigrateTestBase::executeMigrations protected function Executes a set of migrations in dependency order.
MigrateTestBase::getMigration protected function Gets the migration plugin.
MigrateTestBase::mockFailure protected function Records a failure in the map table of a specific migration.
MigrateTestBase::prepareMigration protected function Modify a migration's configuration before executing it.
MigrateTestBase::prepareMigrations protected function Prepare any dependent migrations.
MigrateTestBase::setTestLogger protected function Injects the test logger into the container.
MigrateTestBase::startCollectingMessages public function Start collecting messages and erase previous messages.
MigrateTestBase::stopCollectingMessages public function Stop collecting messages.
MigrateTestBase::tearDown protected function
SqlBaseTest::$migration protected property The (probably mocked) migration under test. Overrides MigrateTestBase::$migration
SqlBaseTest::highWaterDataProvider public static function Data provider for ::testHighWater().
SqlBaseTest::setUp protected function Overrides MigrateTestBase::setUp
SqlBaseTest::testBrokenConnection public function Tests the exception when a connection is defined but not available.
SqlBaseTest::testConnectionTypes public function Tests different connection types.
SqlBaseTest::testHighWater public function Tests that SqlBase respects high-water values.