package org.ogf.saga.job.run;
   
   import org.junit.Ignore;
   import org.junit.Test;
   import org.ogf.saga.error.NoSuccessException;
   import org.ogf.saga.job.Job;
   import org.ogf.saga.job.JobDescription;
   import org.ogf.saga.job.abstracts.Attribute;
   import org.ogf.saga.job.base.JobBaseTest;
  import org.ogf.saga.task.State;
  
  /* **************************************************
  * *** Centre de Calcul de l'IN2P3 - Lyon (France) ***
  * ***             http://cc.in2p3.fr/             ***
  * ***************************************************
  * File:   JobRunDescriptionTest
  * Author: Nicolas Demesy (nicolas.demesy@bt.com)
  * Date:   30 janv. 2008
  * ***************************************************
  * Description: 
  * This test suite is made to be sure that the plug-in 
  * transmits well the requested attributes
  */
  public abstract class RequirementsTest extends JobBaseTest {
  	
  	protected RequirementsTest(String jobprotocol) throws Exception {
          super(jobprotocol);
      }
  
      /**
       * Runs job on a requested working directory
       */
      @Test
      public void test_run_inWorkingDirectory() throws Exception {
          
      	// prepare
      	Attribute[] attributes = new Attribute[1];
      	attributes[0] = new Attribute(JobDescription.WORKINGDIRECTORY, "/dummy");
      	JobDescription desc = createJob("/bin/pwd", attributes, null);
      	
      	Job job = null;
      	try {
  	    	// submit
  	        job = runJob(desc);
  	        
  	        // wait for end
  	        job.waitFor();
  	        
  	        assertEquals(
  	                State.FAILED,
  	                job.getState());
          }
          catch (NoSuccessException noSuccess) {
          	// test is successful is exception instance of BadResource
              if (!noSuccess.getClass().getName().endsWith("BadResource")) {
                  throw noSuccess;
              }
          }
          finally {
          	if(job != null) {
          		job.waitFor(Float.valueOf(FINALY_TIMEOUT));
          	}
          }        
      }
      
     
      /**
       * Runs job, requests an impossible queue and expects FAILED status
       */
      @Test
      public void test_run_queueRequirement() throws Exception {
          
      	// prepare a simple job
      	JobDescription desc = createSimpleJob();
      	// ask for a impossible queue 
      	desc.setAttribute(JobDescription.QUEUE, "impossible_azerty");
      	
      	Job job = null;
      	try {
  	    	// submit
  	        job = runJob(desc);
  	        
  	        // wait for end
  	        job.waitFor();
  	        
  	        assertEquals(
  	                State.FAILED,
  	                job.getState()); 	        
          }
          catch (NoSuccessException noSuccess) {
          	// test may be successful
          }
          finally {
          	if(job != null) {
          		job.waitFor(Float.valueOf(FINALY_TIMEOUT));
          	}
          }
      }
      
     /**
      * Runs job, estimated an impossible job duration and expects FAILED status
      */
     @Test
     public void test_run_cpuTimeRequirement() throws Exception {
         
     	// prepare a simple job
     	JobDescription desc = createSimpleJob();
     	// and inform the scheduler to the estimate time is 14 days
     	desc.setAttribute(JobDescription.TOTALCPUTIME, String.valueOf(60*60*24*14));
     	
     	Job job = null;
     	try {
     		// submit
 	        job = runJob(desc);
 	        
 	        // wait the end
 	        job.waitFor();  
 	        
 	        // check job status
 	        assertEquals(
 	                State.FAILED,
 	                job.getState());
     	}
     	catch (NoSuccessException noSuccess) {
         	// test is successful is exception instance of BadResource
             if (!noSuccess.getClass().getName().endsWith("BadResource")) {
                 throw noSuccess;
             }
         }
     	finally {
         	if(job != null) {
         		job.waitFor(Float.valueOf(FINALY_TIMEOUT));
         	}
         }
 	        
     }
     
     /**
      * Runs simple job, requests 256 Mo of memory and expects DONE status (it is just an information)
      */
     @Test
     public void test_run_memoryRequirement() throws Exception {
         
     	// prepare a job witch requires 256 Mo of RAM
     	Attribute[] attributes = new Attribute[1];
     	attributes[0] = new Attribute(JobDescription.TOTALPHYSICALMEMORY, "256");
     	JobDescription desc =  createJob(SIMPLE_JOB_BINARY, attributes, null);
     	
     	// submit
         Job job = runJob(desc);
         
         // wait for the end
         job.waitFor();  
         
         // check job status
         assertEquals(
                 State.DONE,
                 job.getState());       
     }
         
     /**
      * Runs simple job and expects FAILED status
      */
     @Test
     @Ignore
     public void test_run_processRequirement() throws Exception {
         
     	// prepare a job witch requires 1 million of nodes
     	Attribute[] attributes = new Attribute[2];
     	attributes[0] = new Attribute(JobDescription.NUMBEROFPROCESSES, "1000000");
     	attributes[1] = new Attribute(JobDescription.PROCESSESPERHOST, "1");
     	JobDescription desc =  createJob(SIMPLE_JOB_BINARY, attributes, null);
     	
     	Job job = null;
     	try {
     		// submit
 	        job = runJob(desc);
 	        
 	        // wait the end
 	        job.waitFor();  
 	        
 	        // check job status
 	        assertEquals(
 	                State.FAILED,
 	                job.getState());
     	}
     	catch (NoSuccessException noSuccess) {
         	// test is successful is exception instance of BadResource
             if (!noSuccess.getClass().getName().endsWith("BadResource")) {
                 throw noSuccess;
             }
         }
     	finally {
         	if(job != null) {
         		job.waitFor(Float.valueOf(FINALY_TIMEOUT));
         	}
         }
     }
 
 }