package liquibase.integration.ant;
   
   import liquibase.Liquibase;
   import liquibase.util.ui.UIFactory;
   import org.apache.tools.ant.BuildException;
   
   /**
    * Ant task for migrating a database forward testing rollback.
    */
  public class DatabaseUpdateTestingRollbackTask extends BaseLiquibaseTask {
      private boolean dropFirst = false;
  
      public boolean isDropFirst() {
          return dropFirst;
      }
  
      public void setDropFirst(boolean dropFirst) {
          this.dropFirst = dropFirst;
      }
  
      @Override
      public void execute() throws BuildException {
          if (!shouldRun()) {
              return;
          }
  
          super.execute();
  
          Liquibase liquibase = null;
          try {
              liquibase = createLiquibase();
  
              if (isPromptOnNonLocalDatabase() && !liquibase.isSafeToRunMigration()
                      && UIFactory.getInstance().getFacade().promptForNonLocalDatabase(liquibase.getDatabase())) {
                  throw new BuildException("Chose not to run against non-production database");
              }
  
              if (isDropFirst()) {
                  liquibase.dropAll();
              }
  
              liquibase.updateTestingRollback(getContexts());
  
          } catch (Exception e) {
              throw new BuildException(e);
          } finally {
              closeDatabase(liquibase);
          }
      }
  }