@SuppressWarnings("unchecked")
@Test
public void testCleanupFeatureExecution() throws DbFeatureException {

    // GIVEN
    when(cleanupStrategy.provide(any(CleanupStrategy.StrategyProvider.class))).thenReturn(cleanupStrategyExecutor);
    final List<IDataSet> initialDataSets = Arrays.asList(mock(IDataSet.class));
    // WHEN
    final DbFeature<IDatabaseConnection> feature = featureExecutor.createCleanupFeature(cleanupStrategy, initialDataSets);
    assertThat(feature, notNullValue());
    feature.execute(connection);
    // THEN
    verify(cleanupStrategyExecutor).execute(eq(connection), eq(initialDataSets));
}