content: 
  /*
   * Copyright Contributors to the OpenCue Project
   *
   * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
   * in compliance with the License. You may obtain a copy of the License at
   *
   * http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software distributed under the License
   * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
   * or implied. See the License for the specific language governing permissions and limitations under
   * the License.
   */
  
  package com.imageworks.spcue.test;
  
  import org.junit.AssumptionViolatedException;
  import org.junit.rules.TestRule;
  import org.junit.runner.Description;
  import org.junit.runners.model.Statement;
  
  import com.imageworks.spcue.config.DatabaseEngine;
  
  public class AssumingPostgresEngine implements TestRule {
  
      private DatabaseEngine dbEngine;
  
      public AssumingPostgresEngine() {}
  
      @Override
      public Statement apply(Statement base, Description description) {
          return new Statement() {
              @Override
              public void evaluate() throws Throwable {
                  if (dbEngine == DatabaseEngine.POSTGRES) {
                      base.evaluate();
                  } else {
                      throw new AssumptionViolatedException("Current database engine is "
                              + dbEngine.toString() + ", test requires POSTGRES. Skipping");
                  }
              }
          };
      }
  
      public DatabaseEngine getDbEngine() {
          return dbEngine;
      }
  
      public void setDbEngine(DatabaseEngine dbEngine) {
          this.dbEngine = dbEngine;
      }
  }
  