from func.overlord.client import Client
import func.jobthing as jobthing
import time
import sys

TEST_SLEEP = 5
EXTRA_SLEEP = 5

SLOW_COMMAND = 1
QUICK_COMMAND = 2
RAISES_EXCEPTION_COMMAND = 3
FAKE_COMMAND = 4
TESTS = [ SLOW_COMMAND, QUICK_COMMAND, RAISES_EXCEPTION_COMMAND, FAKE_COMMAND ]

def __tester(async,test):
   if async:
       client = Client("*",nforks=10,async=True)
       oldtime = time.time()

       job_id = -411
       print "======================================================"
       if test == SLOW_COMMAND:
           print "TESTING command that sleeps %s seconds" % TEST_SLEEP
           job_id = client.test.sleep(TEST_SLEEP)
       elif test == QUICK_COMMAND:
           print "TESTING a quick command"
           job_id = client.test.add(1,2)
       elif test == RAISES_EXCEPTION_COMMAND:
           print "TESTING a command that deliberately raises an exception"
           job_id = client.test.explode() # doesn't work yet
       elif test == FAKE_COMMAND:
           print "TESTING a command that does not exist"
           job_id = client.test.does_not_exist(1,2) # ditto
       print "======================================================"

       print "job_id = %s" % job_id
       while True:
           status = client.job_status(job_id)
           (code, results) = status
           nowtime = time.time()
           delta = int(nowtime - oldtime)
           if nowtime > oldtime + TEST_SLEEP + EXTRA_SLEEP:
               print "time expired, test failed"
               return
           if code == jobthing.JOB_ID_RUNNING:  
               print "task is still running, %s elapsed ..." % delta
           elif code == jobthing.JOB_ID_ASYNC_PARTIAL:
               print "task reports partial status, %s elapsed, results = %s" % (delta, results)
           elif code == jobthing.JOB_ID_FINISHED:
               print "(non-async) task complete, %s elapsed, results = %s" % (delta, results)
               return
           elif code == jobthing.JOB_ID_ASYNC_FINISHED:
               print "(async) task complete, %s elapsed, results = %s" % (delta, results)
               return
           else:
               print "job not found: %s, %s elapased" % (code, delta)
           time.sleep(1)
   else:
       print Client("*",nforks=10,async=False).test.sleep(5)
       print Client("*",nforks=10,async=False).test.bork(5)
       print Client("*",nforks=1,async=False).test.bork(5)

for t in TESTS:
    __tester(True,t)

print "======================================================="
print "Testing non-async call"
print __tester(False,-1)