How to use

PersistenceExtensionFeatureResolver

in

org.jboss.arquillian.persistence.core.metadata

public boolean shouldCleanupAfter()
{
 return shouldCleanup() && TestExecutionPhase.AFTER.equals(getCleanupTestPhase());
}

private void createDataSource()
{
 String dataSourceName = persistenceExtensionFeatureResolverProvider.get().getDataSourceName();
 dataSourceProducer.set(loadDataSource(dataSourceName));
}

public void verifyDatabase(@Observes(precedence = 20) AfterPersistenceTest afterPersistenceTest)
{
 if (persistenceExtensionFeatureResolverInstance.get().shouldCleanupAfter())
 {
   cleanUpDataEvent.fire(new CleanupData(afterPersistenceTest, persistenceExtensionFeatureResolverInstance.get().getCleanupStrategy()));
 }
 if (persistenceExtensionScriptingFeatureResolverInstance.get().shouldCleanupUsingScriptAfter())
 {
   final SqlScriptProvider<CleanupUsingScript> scriptsProvider = SqlScriptProvider.createProviderForCleanupScripts(afterPersistenceTest.getTestClass(), scriptingConfigurationInstance.get());
   cleanUpDataUsingScriptEvent.fire(new CleanupDataUsingScript(scriptsProvider.getDescriptorsDefinedFor(afterPersistenceTest.getTestMethod())));
 }
}

public void prepareDatabase(@Observes(precedence = 40) BeforePersistenceTest beforePersistenceTest)
{
 if (persistenceExtensionFeatureResolverInstance.get().shouldCleanupBefore())
 {
   cleanUpDataEvent.fire(new CleanupData(beforePersistenceTest, persistenceExtensionFeatureResolverInstance.get().getCleanupStrategy()));
 }
 if (persistenceExtensionScriptingFeatureResolverInstance.get().shouldCleanupUsingScriptBefore())
 {
   final SqlScriptProvider<CleanupUsingScript> scriptsProvider = SqlScriptProvider.createProviderForCleanupScripts(beforePersistenceTest.getTestClass(), scriptingConfigurationInstance.get());
   cleanUpDataUsingScriptEvent.fire(new CleanupDataUsingScript(scriptsProvider.getDescriptorsDefinedFor(beforePersistenceTest.getTestMethod())));
 }
}

public void before(@Observes final BeforeStep event) {
  final MetadataExtractor extractor = new MetadataExtractor(event.getTestClass());
  final boolean isActive = new PersistenceExtensionEnabler(extractor).shouldPersistenceExtensionBeActivated();
  active.set(isActive);
  if (isActive) {
    final PersistenceConfiguration persistenceConfiguration = configurationInstance.get();
    persistenceExtensionFeatureResolverProvider.set(new PersistenceExtensionFeatureResolver(event.getTestMethod(), extractor, persistenceConfiguration));
    persistenceExtensionScriptingFeatureResolverProvider.set(new PersistenceExtensionScriptingFeatureResolver(event.getTestMethod(), extractor, scriptingConfigurationInstance.get()));
    beforePersistenceTestEvent.fire(new BeforePersistenceTest(event));
  }
}

  private void executeCustomScriptsAfter(AfterPersistenceTest afterPersistenceTest)
  {
   if (!persistenceExtensionFeatureResolver.get().shouldCustomScriptBeAppliedAfterTestRequested())
   {
     return;
   }

   SqlScriptProvider<ApplyScriptAfter> scriptsProvider = SqlScriptProvider.createProviderForScriptsToBeAppliedAfterTest(afterPersistenceTest.getTestClass(), configuration.get());

   executeScriptsEvent.fire(new ExecuteScripts(afterPersistenceTest, scriptsProvider.getDescriptorsDefinedFor(afterPersistenceTest.getTestMethod())));
  }
}

private DatabaseOperation seedingStrategy() {
  final DBUnitConfiguration dbUnitConfiguration = dbunitConfigurationInstance.get();
  final DataSeedStrategy dataSeedStrategy = persistenceExtensionFeatureResolverInstance.get().getDataSeedStrategy();
  final boolean useIdentityInsert = dbUnitConfiguration.isUseIdentityInsert();
  return dataSeedStrategy.provide(new DBUnitDataSeedStrategyProvider(useIdentityInsert));
}

public void createSchema(@Observes(precedence = 10) EventContext<BeforePersistenceTest> context) {
  final BeforePersistenceTest beforePersistenceTest = context.getEvent();
  if (persistenceExtensionFeatureResolver.get().shouldCreateSchema() && !schemaCreated(beforePersistenceTest)) {
    final Collection<SqlScriptResourceDescriptor> schemaDescriptors =
      SqlScriptProvider.createProviderForCreateSchemaScripts(beforePersistenceTest.getTestClass(),
        configuration.get()).getDescriptors(beforePersistenceTest.getTestClass());
    if (!schemaDescriptors.isEmpty()) {
      executeScriptsEvent.fire(new ExecuteScripts(beforePersistenceTest, schemaDescriptors));
    }
  }
  context.proceed();
}

public void verifyDatabase(@Observes(precedence = 20) AfterPersistenceTest afterPersistenceTest)
{
 if (persistenceExtensionFeatureResolverInstance.get().shouldCleanupAfter())
 {
   cleanUpDataEvent.fire(new CleanupData(afterPersistenceTest, persistenceExtensionFeatureResolverInstance.get().getCleanupStrategy()));
 }
 if (persistenceExtensionScriptingFeatureResolverInstance.get().shouldCleanupUsingScriptAfter())
 {
   final SqlScriptProvider<CleanupUsingScript> scriptsProvider = SqlScriptProvider.createProviderForCleanupScripts(afterPersistenceTest.getTestClass(), scriptingConfigurationInstance.get());
   cleanUpDataUsingScriptEvent.fire(new CleanupDataUsingScript(scriptsProvider.getDescriptorsDefinedFor(afterPersistenceTest.getTestMethod())));
 }
}

public void prepareDatabase(@Observes(precedence = 40) BeforePersistenceTest beforePersistenceTest)
{
 if (persistenceExtensionFeatureResolverInstance.get().shouldCleanupBefore())
 {
   cleanUpDataEvent.fire(new CleanupData(beforePersistenceTest, persistenceExtensionFeatureResolverInstance.get().getCleanupStrategy()));
 }
 if (persistenceExtensionScriptingFeatureResolverInstance.get().shouldCleanupUsingScriptBefore())
 {
   final SqlScriptProvider<CleanupUsingScript> scriptsProvider = SqlScriptProvider.createProviderForCleanupScripts(beforePersistenceTest.getTestClass(), scriptingConfigurationInstance.get());
   cleanUpDataUsingScriptEvent.fire(new CleanupDataUsingScript(scriptsProvider.getDescriptorsDefinedFor(beforePersistenceTest.getTestMethod())));
 }
}

public void beforeTest(@Observes(precedence = 25) Before beforeTestEvent)
{
 PersistenceConfiguration persistenceConfiguration = configurationInstance.get();
 persistenceExtensionFeatureResolverProvider.set(new PersistenceExtensionFeatureResolver(beforeTestEvent.getTestMethod(), metadataExtractorProducer.get(), persistenceConfiguration));
 persistenceExtensionScriptingFeatureResolverProvider.set(new PersistenceExtensionScriptingFeatureResolver(beforeTestEvent.getTestMethod(), metadataExtractorProducer.get(), scriptingConfigurationInstance.get()));
 if (persistenceExtensionEnabler.get().shouldPersistenceExtensionBeActivated())
 {
   createDataSource();
   beforePersistenceTestEvent.fire(new BeforePersistenceTest(beforeTestEvent));
 }
}

  private void executeCustomScriptsAfter(AfterPersistenceTest afterPersistenceTest) {
    if (!persistenceExtensionFeatureResolver.get().shouldCustomScriptBeAppliedAfterTestRequested()) {
      return;
    }

    SqlScriptProvider<ApplyScriptAfter> scriptsProvider =
      SqlScriptProvider.createProviderForScriptsToBeAppliedAfterTest(afterPersistenceTest.getTestClass(),
        configuration.get());

    executeScriptsEvent.fire(new ExecuteScripts(afterPersistenceTest,
      scriptsProvider.getDescriptorsDefinedFor(afterPersistenceTest.getTestMethod())));
  }
}

private DatabaseOperation seedingStrategy()
{
 final DBUnitConfiguration dbUnitConfiguration = dbunitConfigurationInstance.get();
 final DataSeedStrategy dataSeedStrategy = persistenceExtensionFeatureResolverInstance.get().getDataSeedStrategy();
 final boolean useIdentityInsert = dbUnitConfiguration.isUseIdentityInsert();
 return dataSeedStrategy.provide(new DBUnitDataSeedStrategyProvider(useIdentityInsert));
}

public void createSchema(@Observes(precedence = 10) EventContext<BeforePersistenceTest> context)
{
 final BeforePersistenceTest beforePersistenceTest = context.getEvent();
 if (persistenceExtensionFeatureResolver.get().shouldCreateSchema() && !schemaCreated(beforePersistenceTest))
 {
   final Collection<SqlScriptResourceDescriptor> schemaDescriptors = SqlScriptProvider.createProviderForCreateSchemaScripts(beforePersistenceTest.getTestClass(), configuration.get()).getDescriptors(beforePersistenceTest.getTestClass());
   if (!schemaDescriptors.isEmpty())
   {
    executeScriptsEvent.fire(new ExecuteScripts(beforePersistenceTest, schemaDescriptors));
   }
 }
 context.proceed();
}

public boolean shouldCleanupBefore()
{
 return shouldCleanup() && TestExecutionPhase.BEFORE.equals(getCleanupTestPhase());
}

  public void verifyDatabase(@Observes(precedence = 20) AfterPersistenceTest afterPersistenceTest) {

    if (persistenceExtensionFeatureResolverInstance.get().shouldCleanupAfter()) {
      cleanUpDataEvent.fire(new CleanupData(afterPersistenceTest,
        persistenceExtensionFeatureResolverInstance.get().getCleanupStrategy()));
    }

    if (persistenceExtensionScriptingFeatureResolverInstance.get().shouldCleanupUsingScriptAfter()) {
      final SqlScriptProvider<CleanupUsingScript> scriptsProvider =
        SqlScriptProvider.createProviderForCleanupScripts(afterPersistenceTest.getTestClass(),
          scriptingConfigurationInstance.get());
      cleanUpDataUsingScriptEvent.fire(new CleanupDataUsingScript(
        scriptsProvider.getDescriptorsDefinedFor(afterPersistenceTest.getTestMethod())));
    }
  }
}

public void prepareDatabase(@Observes(precedence = 40) BeforePersistenceTest beforePersistenceTest) {
  if (persistenceExtensionFeatureResolverInstance.get().shouldCleanupBefore()) {
    cleanUpDataEvent.fire(new CleanupData(beforePersistenceTest,
      persistenceExtensionFeatureResolverInstance.get().getCleanupStrategy()));
  }
  if (persistenceExtensionScriptingFeatureResolverInstance.get().shouldCleanupUsingScriptBefore()) {
    final SqlScriptProvider<CleanupUsingScript> scriptsProvider =
      SqlScriptProvider.createProviderForCleanupScripts(beforePersistenceTest.getTestClass(),
        scriptingConfigurationInstance.get());
    cleanUpDataUsingScriptEvent.fire(new CleanupDataUsingScript(
      scriptsProvider.getDescriptorsDefinedFor(beforePersistenceTest.getTestMethod())));
  }
}

public void beforeTest(@Observes(precedence = 25) Before beforeTestEvent)
{
 PersistenceConfiguration persistenceConfiguration = configurationInstance.get();
 persistenceExtensionFeatureResolverProvider.set(new PersistenceExtensionFeatureResolver(beforeTestEvent.getTestMethod(), metadataExtractorProducer.get(), persistenceConfiguration));
 persistenceExtensionScriptingFeatureResolverProvider.set(new PersistenceExtensionScriptingFeatureResolver(beforeTestEvent.getTestMethod(), metadataExtractorProducer.get(), scriptingConfigurationInstance.get()));
 if (persistenceExtensionEnabler.get().shouldPersistenceExtensionBeActivated())
 {
   createDataSource();
   beforePersistenceTestEvent.fire(new BeforePersistenceTest(beforeTestEvent));
 }
}

  private void executeCustomScriptsAfter(AfterPersistenceTest afterPersistenceTest)
  {
   if (!persistenceExtensionFeatureResolver.get().shouldCustomScriptBeAppliedAfterTestRequested())
   {
     return;
   }

   SqlScriptProvider<ApplyScriptAfter> scriptsProvider = SqlScriptProvider.createProviderForScriptsToBeAppliedAfterTest(afterPersistenceTest.getTestClass(), configuration.get());

   executeScriptsEvent.fire(new ExecuteScripts(afterPersistenceTest, scriptsProvider.getDescriptorsDefinedFor(afterPersistenceTest.getTestMethod())));
  }
}

private void createDataSource()
{
 String dataSourceName = persistenceExtensionFeatureResolverProvider.get().getDataSourceName();
 dataSourceProducer.set(loadDataSource(dataSourceName));
}