@Rule
public ArquillianPersistenceRule arquillianPersistenceRule = new ArquillianPersistenceRule();

@DbUnit
@ArquillianResource
RdbmsPopulator rdbmsPopulator;

// To populate data
rdbmsPopulator.forUri(URI.create("jdbc:h2:mem:test;DB_CLOSE_DELAY=-1"))
              .withDriver(Driver.class)
              .withUsername("sa")
              .withPassword("")
              .usingDataSet("heroes.yml")
            .execute();

// To clean data

rdbmsPopulator.forUri(URI.create("jdbc:h2:mem:test;DB_CLOSE_DELAY=-1"))
              .withDriver(Driver.class)
              .withUsername("sa")
              .withPassword("")
              .usingDataSet("heroes.yml")
            .clean();