trait NodeMigrateTypeTestTrait in Drupal 9

trait NodeMigrateTypeTestTrait {

  /**
   * The migrate_map table name.
   *
   * @var string
   */
  public $tableName = NULL;

  /**
   * Gets the numbers of complete and classic node migrate_map tables.
   *
   * @param string $version
   *   The source database version.
   *
   * @return array
   *   An associative array with the total number of complete and classic
   *   node migrate_map tables.
   */
  protected function nodeMigrateMapTableCount($version) {
    $results = [];
    $bases = [
      'node',
      'node_complete',
    ];
    $tables = \Drupal::database()
      ->schema()
      ->findTables('migrate_map_d' . $version . '_node%');
    foreach ($bases as $base) {
      $base_tables = preg_grep('/^migrate_map_d' . $version . '_' . $base . '_{2}.*$/', $tables);
      $results[$base] = count($base_tables);
    }
    return $results;
  }

  /**
   * Remove the node migrate map table.
   *
   * @param string $type
   *   The type of node migration, 'complete' or 'classic'.
   * @param string $version
   *   The source database version.
   *
   * @throws \Exception
   */
  protected function removeNodeMigrateMapTable($type, $version) {
    $name = $this
      ->getTableName($type, $version);
    \Drupal::database()
      ->schema()
      ->dropTable($name);
  }

  /**
   * Gets the migrate_map table name.
   *
   * @param string $type
   *   The type of node migration, 'complete' or 'classic'.
   * @param string $version
   *   The source database version.
   *
   * @return string
   *   The migrate_map table name.
   */
  protected function getTableName($type, $version) {
    if (!$this->tableName) {
      $content_type = $this
        ->randomMachineName();
      $this->tableName = 'migrate_map_d' . $version . '_node_complete__' . $content_type;
      if ($type == NodeMigrateType::NODE_MIGRATE_TYPE_CLASSIC) {
        $this->tableName = 'migrate_map_d' . $version . '_node__' . $content_type;
      }
    }
    return $this->tableName;
  }

  /**
   * Create a node migrate_map table.
   *
   * @param string $type
   *   The type of node migration, 'complete' or 'classic'.
   * @param string $version
   *   The source database version.
   *
   * @throws \Exception
   */
  protected function makeNodeMigrateMapTable($type, $version) {
    $name = $this
      ->getTableName($type, $version);
    $fields = [
      'source_ids_hash' => [
        'type' => 'varchar',
        'not null' => TRUE,
        'length' => '64',
      ],
      'sourceid1' => [
        'type' => 'int',
        'not null' => TRUE,
        'size' => 'normal',
      ],
      'sourceid2' => [
        'type' => 'int',
        'not null' => TRUE,
        'size' => 'normal',
      ],
      'sourceid3' => [
        'type' => 'varchar',
        'not null' => TRUE,
        'length' => '255',
      ],
      'destid1' => [
        'type' => 'int',
        'not null' => FALSE,
        'size' => 'normal',
        'unsigned' => TRUE,
      ],
      'destid2' => [
        'type' => 'int',
        'not null' => FALSE,
        'size' => 'normal',
        'unsigned' => TRUE,
      ],
      'destid3' => [
        'type' => 'varchar_ascii',
        'not null' => FALSE,
        'length' => '12',
      ],
      'source_row_status' => [
        'type' => 'int',
        'not null' => TRUE,
        'size' => 'tiny',
        'default' => '0',
        'unsigned' => TRUE,
      ],
      'rollback_action' => [
        'type' => 'int',
        'not null' => TRUE,
        'size' => 'tiny',
        'default' => '0',
        'unsigned' => TRUE,
      ],
      'last_imported' => [
        'type' => 'int',
        'not null' => TRUE,
        'size' => 'normal',
        'default' => '0',
        'unsigned' => TRUE,
      ],
      'hash' => [
        'type' => 'varchar',
        'not null' => FALSE,
        'length' => '64',
      ],
    ];
    $values = [
      'source_ids_hash' => '123',
      'sourceid1' => '4242',
      'sourceid2' => '4242',
      'sourceid3' => 'en',
      'destid1' => '4242',
      'destid2' => '4242',
      'destid3' => 'en',
      'source_row_status' => '1',
      'rollback_action' => '1',
      'last_imported' => time(),
      'hash' => 'abc',
    ];
    $indexes = [
      'source' => [
        'sourceid1',
        'sourceid2',
        'sourceid3',
      ],
    ];

    // Remove keys not used.
    if ($type == NodeMigrateType::NODE_MIGRATE_TYPE_CLASSIC) {
      $keys = [
        'sourceid2',
        'sourceid3',
        'destid2',
        'destid3',
      ];
      foreach ($keys as $key) {
        unset($fields[$key]);
        unset($values[$key]);
        if (strstr($key, 'sourceid')) {
          $index_key = substr($key, -1) - 1;
          unset($indexes['source'][$index_key]);
        }
      }
    }
    $connection = \Drupal::database();
    $connection
      ->schema()
      ->createTable($name, [
      'fields' => $fields,
      'primary key' => [
        'source_ids_hash',
      ],
      'indexes' => $indexes,
      'mysql_character_set' => 'utf8mb4',
    ]);
    $field_names = array_keys($fields);
    $connection
      ->insert($name)
      ->fields($field_names)
      ->values($values)
      ->execute();
  }

}