io.dropwizard.db.DatabaseConfiguration Java Examples

@BeforeEach
public void setUp() throws Exception {
    // Setup necessary mock
    final JarLocation location = mock(JarLocation.class);
    when(location.getVersion()).thenReturn(Optional.of("1.0.0"));

    final DatabaseConfiguration dbConfiguration = mock(DatabaseConfiguration.class);
    when(dbConfiguration.getDataSourceFactory(any())).thenReturn(mock(PooledDataSourceFactory.class));

    final FlywayConfiguration flywayConfiguration = mock(FlywayConfiguration.class);
    mockFlywayFactory = mock(FlywayFactory.class);
    when(flywayConfiguration.getFlywayFactory(any())).thenReturn(mockFlywayFactory);

    mockFlyway = mock(Flyway.class);
    when(mockFlywayFactory.build(any())).thenReturn(mockFlyway);

    // Add commands you want to test
    final Bootstrap<TestConfiguration> bootstrap = new Bootstrap<>(new TestApplication());
    bootstrap.addCommand(new DbCommand<TestConfiguration>("db", dbConfiguration, flywayConfiguration, TestConfiguration.class));

    // Build what'll run the command and interpret arguments
    cli = new Cli(location, bootstrap, System.out, System.err);
}
 

public DbCommand(final String name, final DatabaseConfiguration<T> databaseConfiguration,
                 final FlywayConfiguration<T> flywayConfiguration,
                 final Class<T> configurationClass) {
    super(name, "Run database migration tasks",
            databaseConfiguration, flywayConfiguration, configurationClass);

    addSubCommand(new DbMigrateCommand<>(databaseConfiguration, flywayConfiguration, configurationClass));
    addSubCommand(new DbCleanCommand<>(databaseConfiguration, flywayConfiguration, configurationClass));
    addSubCommand(new DbInitCommand<>(databaseConfiguration, flywayConfiguration, configurationClass));
    addSubCommand(new DbValidateCommand<>(databaseConfiguration, flywayConfiguration, configurationClass));
    addSubCommand(new DbInfoCommand<>(databaseConfiguration, flywayConfiguration, configurationClass));
    addSubCommand(new DbRepairCommand<>(databaseConfiguration, flywayConfiguration, configurationClass));
}
 

AbstractFlywayCommand(final String name,
                      final String description,
                      final DatabaseConfiguration<T> databaseConfiguration,
                      final FlywayConfiguration<T> flywayConfiguration,
                      final Class<T> configurationClass) {
    super(name, description);
    this.databaseConfiguration = databaseConfiguration;
    this.flywayConfiguration = flywayConfiguration;
    this.configurationClass = configurationClass;
}
 

public DbValidateCommand(final DatabaseConfiguration<T> databaseConfiguration,
                         final FlywayConfiguration<T> flywayConfiguration,
                         final Class<T> configurationClass) {
    super("validate", "Validates the applied migrations against the ones available on the classpath.",
            databaseConfiguration, flywayConfiguration, configurationClass);
}
 

public DbRepairCommand(final DatabaseConfiguration<T> databaseConfiguration,
                       final FlywayConfiguration<T> flywayConfiguration,
                       final Class<T> configurationClass) {
    super("repair", "Repairs the metadata table.",
            databaseConfiguration, flywayConfiguration, configurationClass);
}
 

NoOptionsFlywayCommand(String name, String description, DatabaseConfiguration<T> databaseConfiguration,
                       FlywayConfiguration<T> flywayConfiguration, Class<T> configurationClass) {
    super(name, description, databaseConfiguration, flywayConfiguration, configurationClass);
}
 

public DbInitCommand(final DatabaseConfiguration<T> databaseConfiguration,
                     final FlywayConfiguration<T> flywayConfiguration,
                     final Class<T> configurationClass) {
    super("init", "Creates and initializes the metadata table (existing database).",
            databaseConfiguration, flywayConfiguration, configurationClass);
}
 

public DbInfoCommand(final DatabaseConfiguration<T> databaseConfiguration,
                     final FlywayConfiguration<T> flywayConfiguration,
                     final Class<T> configurationClass) {
    super("info", "Prints the details and status information about all the migrations.",
            databaseConfiguration, flywayConfiguration, configurationClass);
}
 

public DbCleanCommand(final DatabaseConfiguration<T> databaseConfiguration,
                      final FlywayConfiguration<T> flywayConfiguration,
                      final Class<T> configurationClass) {
    super("clean", "Drops all objects in the configured schemas.",
            databaseConfiguration, flywayConfiguration, configurationClass);
}
 

public DbMigrateCommand(final DatabaseConfiguration<T> databaseConfiguration,
                        final FlywayConfiguration<T> flywayConfiguration,
                        final Class<T> configurationClass) {
    super("migrate", "Migrates the database.",
            databaseConfiguration, flywayConfiguration, configurationClass);
}