Exemplo n.º 1
0
  @Test
  public void info() throws Exception {
    DriverDataSource dataSource =
        new DriverDataSource(null, "jdbc:h2:mem:flyway_db_info;DB_CLOSE_DELAY=-1", "sa", null);

    Flyway flyway = new Flyway();
    flyway.setDataSource(dataSource);

    flyway.setLocations("migration/sql");
    assertEquals(4, flyway.info().all().length);
    assertEquals(4, flyway.info().pending().length);

    flyway.setTarget(new MigrationVersion("1.1"));
    assertEquals(4, flyway.info().all().length);
    assertEquals(2, flyway.info().pending().length);
    assertEquals(MigrationState.ABOVE_TARGET, flyway.info().all()[2].getState());
    assertEquals(MigrationState.ABOVE_TARGET, flyway.info().all()[3].getState());

    flyway.migrate();
    assertEquals("1.1", flyway.info().current().getVersion().toString());
    assertEquals(MigrationState.SUCCESS, flyway.info().current().getState());
    assertEquals(4, flyway.info().all().length);
    assertEquals(0, flyway.info().pending().length);

    flyway.setTarget(MigrationVersion.LATEST);
    assertEquals(4, flyway.info().all().length);
    assertEquals(2, flyway.info().pending().length);

    flyway.migrate();
    assertEquals("2.0", flyway.info().current().getVersion().toString());
    assertEquals(MigrationState.SUCCESS, flyway.info().current().getState());
    assertEquals(4, flyway.info().all().length);
    assertEquals(0, flyway.info().pending().length);
  }
Exemplo n.º 2
0
  @Test
  public void outOfOrder() {
    DriverDataSource dataSource =
        new DriverDataSource(null, "jdbc:h2:mem:flyway_out_of_order;DB_CLOSE_DELAY=-1", "sa", "");

    Flyway flyway = new Flyway();
    flyway.setDataSource(dataSource);
    flyway.setLocations("migration/sql");
    flyway.setTarget(new MigrationVersion("1.2"));
    assertEquals(4, flyway.info().all().length);
    assertEquals(3, flyway.info().pending().length);

    flyway.clean();
    assertEquals(3, flyway.migrate());
    assertEquals(0, flyway.info().pending().length);

    flyway.setLocations("migration/sql", "migration/outoforder");
    assertEquals(5, flyway.info().all().length);
    assertEquals(MigrationState.IGNORED, flyway.info().all()[2].getState());
    assertEquals(0, flyway.migrate());

    flyway.setTarget(MigrationVersion.LATEST);
    flyway.setOutOfOrder(true);
    assertEquals(MigrationState.PENDING, flyway.info().all()[2].getState());
    assertEquals(2, flyway.migrate());

    assertEquals(MigrationState.OUT_OF_ORDER, flyway.info().all()[2].getState());
    assertEquals(MigrationState.SUCCESS, flyway.info().all()[4].getState());
  }
Exemplo n.º 3
0
 @Test
 public void emptyLocations() {
   Flyway flyway = new Flyway();
   flyway.setDataSource("jdbc:h2:mem:flyway_empty;DB_CLOSE_DELAY=-1", "sa", "");
   flyway.setLocations("migration/empty");
   assertEquals(0, flyway.migrate());
   // Used to fail with exception due to non-empty schema and empty metadata table.
   assertEquals(0, flyway.migrate());
 }
Exemplo n.º 4
0
  @Test
  public void multipleSetDataSourceCalls() throws Exception {
    DriverDataSource dataSource1 =
        new DriverDataSource(null, "jdbc:h2:mem:flyway_db_1;DB_CLOSE_DELAY=-1", "sa", "");

    DriverDataSource dataSource2 =
        new DriverDataSource(null, "jdbc:h2:mem:flyway_db_2;DB_CLOSE_DELAY=-1", "sa", "");

    Connection connection1 = dataSource1.getConnection();
    Connection connection2 = dataSource2.getConnection();

    Schema schema1 = new H2DbSupport(connection1).getSchema("PUBLIC");
    Schema schema2 = new H2DbSupport(connection2).getSchema("PUBLIC");

    assertTrue(schema1.empty());
    assertTrue(schema2.empty());

    Flyway flyway = new Flyway();

    flyway.setDataSource(dataSource1);
    flyway.setDataSource(dataSource2);

    flyway.setLocations("migration/sql");
    flyway.migrate();

    assertTrue(schema1.empty());
    assertFalse(schema2.empty());

    connection1.close();
    connection2.close();
  }
Exemplo n.º 5
0
  @Test
  public void futureMigrations() {
    Flyway flyway = new Flyway();
    flyway.setDataSource("jdbc:h2:mem:flyway_future;DB_CLOSE_DELAY=-1", "sa", "");
    flyway.setLocations("migration/sql");
    flyway.migrate();

    flyway.setLocations("migration/empty");
    assertEquals(MigrationState.FUTURE_SUCCESS, flyway.info().applied()[0].getState());
  }
Exemplo n.º 6
0
  @Test
  public void initOnMigrateOnCleanOnValidate() throws Exception {
    DriverDataSource dataSource =
        new DriverDataSource(
            null, "jdbc:h2:mem:flyway_db_init_validate;DB_CLOSE_DELAY=-1", "sa", "");

    Flyway flyway = new Flyway();
    flyway.setDataSource(dataSource);
    flyway.setSchemas("new1");
    flyway.setLocations("migration/validate");
    flyway.migrate();

    flyway.setCleanOnValidationError(true);
    flyway.setValidateOnMigrate(true);
    flyway.setSqlMigrationPrefix("CheckValidate");
    flyway.migrate();

    assertEquals("1", flyway.info().current().getVersion().toString());
  }
Exemplo n.º 7
0
  @Test
  public void noConnectionLeak() {
    OpenConnectionCountDriverDataSource dataSource = new OpenConnectionCountDriverDataSource();

    assertEquals(0, dataSource.getOpenConnectionCount());
    Flyway flyway = new Flyway();
    flyway.setDataSource(dataSource);
    flyway.setLocations("migration/sql");
    flyway.clean();
    assertEquals(0, dataSource.getOpenConnectionCount());
    assertEquals(4, flyway.migrate());
    assertEquals(0, dataSource.getOpenConnectionCount());
  }
Exemplo n.º 8
0
  @Test
  public void failed() {
    StringLogCreator logCreator = new StringLogCreator();
    LogFactory.setLogCreator(logCreator);

    try {
      Flyway flyway = new Flyway();
      flyway.setDataSource("jdbc:h2:mem:flyway_failed;DB_CLOSE_DELAY=-1", "sa", "");
      flyway.setLocations("migration/failed");
      flyway.migrate();
      fail();
    } catch (FlywayException e) {
      System.out.println(logCreator.getOutput());
    } finally {
      LogFactory.setLogCreator(null);
    }
  }
Exemplo n.º 9
0
  @Test
  public void noConnectionLeakWithException() {
    OpenConnectionCountDriverDataSource dataSource = new OpenConnectionCountDriverDataSource();

    assertEquals(0, dataSource.getOpenConnectionCount());
    Flyway flyway = new Flyway();
    flyway.setDataSource(dataSource);
    flyway.setLocations("migration/failed");
    try {
      flyway.clean();
      assertEquals(0, dataSource.getOpenConnectionCount());
      flyway.migrate();
      fail();
    } catch (FlywayException e) {
      // Expected -> Ignore
    }
    assertEquals(0, dataSource.getOpenConnectionCount());
  }
Exemplo n.º 10
0
  @Test
  public void repairFirst() throws Exception {
    DriverDataSource dataSource =
        new DriverDataSource(null, "jdbc:h2:mem:flyway_db_repair;DB_CLOSE_DELAY=-1", "sa", "");

    Flyway flyway = new Flyway();
    flyway.setDataSource(dataSource);

    flyway.setLocations("migration/failed");
    assertEquals(1, flyway.info().all().length);

    try {
      flyway.migrate();
    } catch (FlywayException e) {
      // Should happen
    }
    assertEquals("1", flyway.info().current().getVersion().toString());
    assertEquals(MigrationState.FAILED, flyway.info().current().getState());

    flyway.repair();
    assertNull(flyway.info().current());
  }