trait FieldDiscoveryTestTrait in Drupal 9

trait FieldDiscoveryTestTrait {

  /**
   * Asserts the field discovery returns the expected processes.
   *
   * @param \Drupal\migrate_drupal\FieldDiscoveryInterface $field_discovery
   *   The Field Discovery service.
   * @param \Drupal\migrate\Plugin\MigrationPluginManagerInterface $migration_plugin_manager
   *   The migration plugin manager service.
   * @param string $core
   *   The Drupal core version, either '6', or '7'.
   * @param string $field_plugin_method
   *   (optional) The field plugin method to use.
   * @param array $expected_process
   *   (optional) The expected resulting process.
   * @param string $entity_type_id
   *   (optional) The entity type id.
   * @param string $bundle
   *   (optional) The bundle.
   */
  public function assertFieldProcess(FieldDiscoveryInterface $field_discovery, MigrationPluginManagerInterface $migration_plugin_manager, $core, $field_plugin_method = NULL, array $expected_process = [], $entity_type_id = NULL, $bundle = NULL) {
    $definition = [
      'migration_tags' => [
        'Drupal ' . $core,
      ],
      'field_plugin_method' => $field_plugin_method,
    ];
    $migration = $migration_plugin_manager
      ->createStubMigration($definition);
    if ($bundle) {
      $field_discovery
        ->addBundleFieldProcesses($migration, $entity_type_id, $bundle);
    }
    elseif ($entity_type_id) {
      $field_discovery
        ->addEntityFieldProcesses($migration, $entity_type_id);
    }
    else {
      $field_discovery
        ->addAllFieldProcesses($migration);
    }
    $actual_process = $migration
      ->getProcess();
    $this
      ->assertSame($expected_process, $actual_process);
  }

  /**
   * Asserts the field discovery returns the expected processes.
   *
   * @param \Drupal\migrate_drupal\FieldDiscoveryInterface $field_discovery
   *   The Field Discovery service.
   * @param \Drupal\migrate\Plugin\MigrationPluginManagerInterface $migration_plugin_manager
   *   The migration plugin manager service.
   * @param string $core
   *   The Drupal core version, either '6', or '7'.
   * @param array $expected_process_keys
   *   (optional) The expected resulting process_keys.
   * @param string $entity_type_id
   *   (optional) The entity type id.
   * @param string $bundle
   *   (optional) The bundle.
   */
  public function assertFieldProcessKeys(FieldDiscoveryInterface $field_discovery, MigrationPluginManagerInterface $migration_plugin_manager, $core, array $expected_process_keys, $entity_type_id = NULL, $bundle = NULL) {
    $definition = [
      'migration_tags' => [
        'Drupal ' . $core,
      ],
    ];
    $migration = $migration_plugin_manager
      ->createStubMigration($definition);
    if ($bundle) {
      $field_discovery
        ->addBundleFieldProcesses($migration, $entity_type_id, $bundle);
    }
    elseif ($entity_type_id) {
      $field_discovery
        ->addEntityFieldProcesses($migration, $entity_type_id);
    }
    else {
      $field_discovery
        ->addAllFieldProcesses($migration);
    }
    $actual_process = $migration
      ->getProcess();
    $actual = array_keys($actual_process);
    $this
      ->assertSame(sort($expected_process_keys), sort($actual));
  }

  /**
   * Asserts a migrate source plugin.
   *
   * @param string $core
   *   The Drupal core version.
   * @param string $class
   *   The expected class of the source plugin.
   * @param array $expected_definition
   *   The expected source plugin definition.
   */
  public function assertSourcePlugin($core, $class, array $expected_definition) {
    $field_discovery = new FieldDiscoveryTestClass($this->fieldPluginManager, $this->migrationPluginManager, $this->logger);
    $source = $field_discovery
      ->getSourcePlugin($core);
    $this
      ->assertInstanceOf($class, $source);
    $this
      ->assertSame($expected_definition, $source
      ->getPluginDefinition());
  }

}