/*
    * Copyright 2007-2008 The Kuali Foundation
    *
    * Licensed under the Educational Community 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.opensource.org/licenses/ecl2.php
    *
   * 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 org.kuali.rice.test.data;
  
  import org.apache.commons.lang.StringUtils;
  import org.kuali.rice.test.SQLDataLoader;
  
  /**
   * Utilities for unit test data annotations. 
   * 
   * @author Kuali Rice Team (rice.collab@kuali.org)
   *
   */
  public class UnitTestDataUtils {
  
      public static void executeDataLoader(UnitTestData[] data) throws Exception {
          for (UnitTestData d: data) {
              executeDataLoader(d);
          }
      }
  
      public static void executeDataLoader(UnitTestData data) throws Exception {
          SQLDataLoader sqlDataLoader;
          for (UnitTestData.Type type : data.order()) {
              switch (type) {
                  case SQL_FILES : 
                      for (UnitTestFile file : data.sqlFiles()) {
                          sqlDataLoader = new SQLDataLoader(file.filename(), file.delimiter());
                          sqlDataLoader.runSql();
                      }
                      break;
                  case SQL_STATEMENTS : 
                      for (UnitTestSql statement : data.sqlStatements()) {
                          sqlDataLoader = new SQLDataLoader(statement.value());
                          sqlDataLoader.runSql();
                      }
                      break;
                  default: break;
              }
          }
          
          if (!StringUtils.isEmpty(data.filename())) {
              if (!StringUtils.isEmpty(data.value()))
                  throw new RuntimeException("UnitTestDataArtifact may not specify both SQL file and content");
              sqlDataLoader = new SQLDataLoader(data.filename(), data.delimiter());
              sqlDataLoader.runSql();
          } else if (!StringUtils.isEmpty(data.value())) {
              sqlDataLoader = new SQLDataLoader(data.value());
              sqlDataLoader.runSql();
          }
      }
  }