trait FieldDiscoveryTestTrait

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());
    }

}