Athou_commafeed/commafeed-server/src/test/java/com/commafeed/CommaFeedDropwizardAppExtension.java

package com.commafeed;

import java.sql.Connection;
import java.sql.SQLException;

import javax.sql.DataSource;

import org.mockserver.socket.PortFactory;

import com.codahale.metrics.MetricRegistry;

import io.dropwizard.testing.ConfigOverride;
import io.dropwizard.testing.ResourceHelpers;
import io.dropwizard.testing.junit5.DropwizardAppExtension;

public class CommaFeedDropwizardAppExtension extends DropwizardAppExtension<CommaFeedConfiguration> {

	public CommaFeedDropwizardAppExtension() {
		super(CommaFeedApplication.class, ResourceHelpers.resourceFilePath("config.test.yml"),
				ConfigOverride.config("server.applicationConnectors[0].port", String.valueOf(PortFactory.findFreePort())));
	}

	@Override
	public void after() {
		super.after();

		// clean database after each test
		DataSource dataSource = getConfiguration().getDataSourceFactory().build(new MetricRegistry(), "cleanup");
		try (Connection connection = dataSource.getConnection()) {
			connection.prepareStatement("DROP ALL OBJECTS").executeUpdate();
		} catch (SQLException e) {
			throw new RuntimeException("could not cleanup database", e);
		}
	}

}