package net.kwfgrid.gwes;
   
   import java.io.*;
   import java.rmi.RemoteException;
   
   import junit.framework.Assert;
   import net.kwfgrid.gwes.exception.*;
   import net.kwfgrid.gworkflowdl.structure.*;
   import org.apache.log4j.Logger;
  
  /**
   * Abstract base class for test cases.
   */
  public abstract class LocalGWESAbstractTestCase
      extends AbstractTestCase  {
  
      protected GWES gwes;
  
      String workflowID;
  
      static Logger logger = Logger.getLogger(LocalGWESAbstractTestCase.class);
  
      /**
       * Constructor.
       */
      public LocalGWESAbstractTestCase(String testName) throws LoggingException {
          super(testName);
          gwes = new GWESEngine();
      }
  
      public GWES getGwes() {
          return gwes;
      }
  
      public String testGWES(String gworkflowdlPath, int expectedStatus) throws StateTransitionException, WorkflowFormatException, WorkflowSecurityException, NoSuchWorkflowException, InterruptedException, DatabaseException, LoggingException {
          String retGWorkflowDL = null;
          try {
              if (gwes == null) {
                  gwes = getGwes();
              }
              workflowID = gwes.initiate(readfile(gworkflowdlPath), userID);
              logger.debug("workflowID: " + workflowID);
              gwes.start(workflowID, userID);
  
              int status = waitWorkflowStatus(gwes, workflowID);
  
              retGWorkflowDL = gwes.getWorkflowDescription(workflowID, userID);
              logger.debug(retGWorkflowDL);
  
              Assert.assertEquals("Exit status of workflow "+workflowID, expectedStatus, status);
          } catch (GWESException e) {
              logger.error("GWES exception: "+e,e);
              Assert.fail("Got unpredicted GWES exception");
          } catch (RemoteException e) {
              logger.error("GWES Remote exception: "+e,e);
              Assert.fail("GWES Remote exception");
          } catch (IOException e) {
              logger.error("GWES IO exception: "+e,e);
              Assert.fail("GWES IO exception");
          }
  
          return retGWorkflowDL;
      }
  
      /**
        * waits until workflow completes, terminates or is supended
        * @throws net.kwfgrid.gwes.exception.NoSuchWorkflowException
        * @throws InterruptedException
        */
       public int waitWorkflowStatus(GWES gwes, String workflowID) throws NoSuchWorkflowException, InterruptedException, RemoteException, DatabaseException, LoggingException, GWESException, WorkflowSecurityException {
           Logger logger = Logger.getLogger(gwes.getClass());
           int status = gwes.getStatus(workflowID, userID);
           logger.debug("Status: "+ WorkflowStatus.getStatusAsString(status));
           while (status != WorkflowStatus.STATUS_COMPLETED
                   && status != WorkflowStatus.STATUS_TERMINATED
                   && status != WorkflowStatus.STATUS_SUSPENDED) {
               int oldStatus = status;
               status = gwes.waitForStatusChangeFrom(workflowID, oldStatus, userID);
               logger.debug("Status: "+ WorkflowStatus.getStatusAsString(status));
           }
          return status;
       }
  
  }