HighWaterTest.php in Drupal 9

Namespace

Drupal\Tests\migrate\Kernel

File

core/modules/migrate/tests/src/Kernel/HighWaterTest.php

<?php

namespace Drupal\Tests\migrate\Kernel;


/**
 * Tests migration high water property.
 *
 * @group migrate
 */
class HighWaterTest extends MigrateTestBase {

  /**
   * {@inheritdoc}
   */
  protected static $modules = [
    'system',
    'user',
    'node',
    'migrate',
    'migrate_high_water_test',
    'field',
  ];

  /**
   * {@inheritdoc}
   */
  protected function setUp() : void {
    parent::setUp();

    // Create source test table.
    $this->sourceDatabase
      ->schema()
      ->createTable('high_water_node', [
      'fields' => [
        'id' => [
          'description' => 'Serial',
          'type' => 'serial',
          'unsigned' => TRUE,
          'not null' => TRUE,
        ],
        'changed' => [
          'description' => 'Highwater',
          'type' => 'int',
          'unsigned' => TRUE,
        ],
        'title' => [
          'description' => 'Title',
          'type' => 'varchar',
          'length' => 128,
          'not null' => TRUE,
          'default' => '',
        ],
      ],
      'primary key' => [
        'id',
      ],
      'description' => 'Contains nodes to import',
    ]);

    // Add 3 items to source table.
    $this->sourceDatabase
      ->insert('high_water_node')
      ->fields([
      'title',
      'changed',
    ])
      ->values([
      'title' => 'Item 1',
      'changed' => 1,
    ])
      ->values([
      'title' => 'Item 2',
      'changed' => 2,
    ])
      ->values([
      'title' => 'Item 3',
      'changed' => 3,
    ])
      ->execute();
    $this
      ->installEntitySchema('node');
    $this
      ->installEntitySchema('user');
    $this
      ->installSchema('node', 'node_access');
    $this
      ->executeMigration('high_water_test');
  }

  /**
   * Tests high water property of SqlBase.
   */
  public function testHighWater() {

    // Assert all of the nodes have been imported.
    $this
      ->assertNodeExists('Item 1');
    $this
      ->assertNodeExists('Item 2');
    $this
      ->assertNodeExists('Item 3');

    // Update Item 1 setting its high_water_property to value that is below
    // current high water mark.
    $this->sourceDatabase
      ->update('high_water_node')
      ->fields([
      'title' => 'Item 1 updated',
      'changed' => 2,
    ])
      ->condition('title', 'Item 1')
      ->execute();

    // Update Item 2 setting its high_water_property to value equal to
    // current high water mark.
    $this->sourceDatabase
      ->update('high_water_node')
      ->fields([
      'title' => 'Item 2 updated',
      'changed' => 3,
    ])
      ->condition('title', 'Item 2')
      ->execute();

    // Update Item 3 setting its high_water_property to value that is above
    // current high water mark.
    $this->sourceDatabase
      ->update('high_water_node')
      ->fields([
      'title' => 'Item 3 updated',
      'changed' => 4,
    ])
      ->condition('title', 'Item 3')
      ->execute();

    // Execute migration again.
    $this
      ->executeMigration('high_water_test');

    // Item with lower highwater should not be updated.
    $this
      ->assertNodeExists('Item 1');
    $this
      ->assertNodeDoesNotExist('Item 1 updated');

    // Item with equal highwater should not be updated.
    $this
      ->assertNodeExists('Item 2');
    $this
      ->assertNodeDoesNotExist('Item 2 updated');

    // Item with greater highwater should be updated.
    $this
      ->assertNodeExists('Item 3 updated');
    $this
      ->assertNodeDoesNotExist('Item 3');
  }

  /**
   * Tests that the high water value can be 0.
   */
  public function testZeroHighwater() {

    // Assert all of the nodes have been imported.
    $this
      ->assertNodeExists('Item 1');
    $this
      ->assertNodeExists('Item 2');
    $this
      ->assertNodeExists('Item 3');
    $migration = $this->container
      ->get('plugin.manager.migration')
      ->CreateInstance('high_water_test', []);
    $source = $migration
      ->getSourcePlugin();
    $source
      ->rewind();
    $count = 0;
    while ($source
      ->valid()) {
      $count++;
      $source
        ->next();
    }

    // Expect no rows as everything is below the high water mark.
    $this
      ->assertSame(0, $count);

    // Test resetting the high water mark to 0.
    $this->container
      ->get('keyvalue')
      ->get('migrate:high_water')
      ->set('high_water_test', 0);
    $migration = $this->container
      ->get('plugin.manager.migration')
      ->CreateInstance('high_water_test', []);
    $source = $migration
      ->getSourcePlugin();
    $source
      ->rewind();
    $count = 0;
    while ($source
      ->valid()) {
      $count++;
      $source
        ->next();
    }
    $this
      ->assertSame(3, $count);
  }

  /**
   * Tests that deleting the high water value causes all rows to be reimported.
   */
  public function testNullHighwater() {

    // Assert all of the nodes have been imported.
    $this
      ->assertNodeExists('Item 1');
    $this
      ->assertNodeExists('Item 2');
    $this
      ->assertNodeExists('Item 3');
    $migration = $this->container
      ->get('plugin.manager.migration')
      ->CreateInstance('high_water_test', []);
    $source = $migration
      ->getSourcePlugin();
    $source
      ->rewind();
    $count = 0;
    while ($source
      ->valid()) {
      $count++;
      $source
        ->next();
    }

    // Expect no rows as everything is below the high water mark.
    $this
      ->assertSame(0, $count);

    // Test resetting the high water mark.
    $this->container
      ->get('keyvalue')
      ->get('migrate:high_water')
      ->delete('high_water_test');
    $migration = $this->container
      ->get('plugin.manager.migration')
      ->CreateInstance('high_water_test', []);
    $source = $migration
      ->getSourcePlugin();
    $source
      ->rewind();
    $count = 0;
    while ($source
      ->valid()) {
      $count++;
      $source
        ->next();
    }
    $this
      ->assertSame(3, $count);
  }

  /**
   * Tests high water property of SqlBase when rows marked for update.
   */
  public function testHighWaterUpdate() {

    // Assert all of the nodes have been imported.
    $this
      ->assertNodeExists('Item 1');
    $this
      ->assertNodeExists('Item 2');
    $this
      ->assertNodeExists('Item 3');

    // Update Item 1 setting its high_water_property to value that is below
    // current high water mark.
    $this->sourceDatabase
      ->update('high_water_node')
      ->fields([
      'title' => 'Item 1 updated',
      'changed' => 2,
    ])
      ->condition('title', 'Item 1')
      ->execute();

    // Update Item 2 setting its high_water_property to value equal to
    // current high water mark.
    $this->sourceDatabase
      ->update('high_water_node')
      ->fields([
      'title' => 'Item 2 updated',
      'changed' => 3,
    ])
      ->condition('title', 'Item 2')
      ->execute();

    // Update Item 3 setting its high_water_property to value that is above
    // current high water mark.
    $this->sourceDatabase
      ->update('high_water_node')
      ->fields([
      'title' => 'Item 3 updated',
      'changed' => 4,
    ])
      ->condition('title', 'Item 3')
      ->execute();

    // Set all rows as needing an update.
    $id_map = $this
      ->getMigration('high_water_test')
      ->getIdMap();
    $id_map
      ->prepareUpdate();
    $this
      ->executeMigration('high_water_test');

    // Item with lower highwater should be updated.
    $this
      ->assertNodeExists('Item 1 updated');
    $this
      ->assertNodeDoesNotExist('Item 1');

    // Item with equal highwater should be updated.
    $this
      ->assertNodeExists('Item 2 updated');
    $this
      ->assertNodeDoesNotExist('Item 2');

    // Item with greater highwater should be updated.
    $this
      ->assertNodeExists('Item 3 updated');
    $this
      ->assertNodeDoesNotExist('Item 3');
  }

  /**
   * Assert that node with given title exists.
   *
   * @param string $title
   *   Title of the node.
   */
  protected function assertNodeExists($title) {
    self::assertTrue($this
      ->nodeExists($title));
  }

  /**
   * Assert that node with given title does not exist.
   *
   * @param string $title
   *   Title of the node.
   */
  protected function assertNodeDoesNotExist($title) {
    self::assertFalse($this
      ->nodeExists($title));
  }

  /**
   * Checks if node with given title exists.
   *
   * @param string $title
   *   Title of the node.
   *
   * @return bool
   */
  protected function nodeExists($title) {
    $query = \Drupal::entityQuery('node')
      ->accessCheck(FALSE);
    $result = $query
      ->condition('title', $title)
      ->range(0, 1)
      ->execute();
    return !empty($result);
  }

}

Classes

Name	Description
HighWaterTest	Tests migration high water property.