#
# Simple reply test
#
import os
import unittest
import srutil
import srio
import credentials
import time
import urllib
import subprocess
import json

class Test(unittest.TestCase):
    def runTest(self):

        user = credentials.SRUSER
        token = credentials.SRTOKEN
        projName = 'TEST(Suite2)-Project12'
        jobName = 'Job'
        locName = 'Location'
        fieldName = 'a'
        passphrase = '12345'

        # help12.py is the actual test - run it in the background
        # so script below can send reply asynchronously
        helperPath = os.getcwd() + '/Suite2/help12.py'
        cwdPath = os.getcwd()
        p = subprocess.Popen(['python', helperPath], cwd=cwdPath, preexec_fn=os.setpgrp)

        # Make sure the initial message has time to reach server
        time.sleep(5)

        f = open('srio.project_encoded', 'r')
        srio.sr_project_encoded=f.read()
        f.close()

        f = open('srio.jobid', 'r')
        srio.sr_jobid=f.read()
        f.close()

        f = open('srio.loc_encoded', 'r')
        location_encoded = f.read()
        f.close()

        srio.sr_location_map[locName] = location_encoded

        f = open('srio.msgid', 'r')
        msgid=f.read()
        f.close()

        f = open('srio.aeskey', 'r')
        srio.sr_aeskey = f.read()
        f.close()

        srio.sr_userid = credentials.SRUSER
        srio.sr_token = credentials.SRTOKEN
        srio.sr_passphrase = passphrase

        # Send reply
        reply_content = json.dumps([{'name':fieldName, 'value':'Goodbye World'}])
        result = srutil.SR_put_reply(locName, msgid, reply_content)
        if (result[0] != srio.SR_OK):
            self.fail()

        # Wait for subprocess to complete - return code indicates result
        retcode = p.wait()
        if retcode != 0:
            self.fail()

        srutil.SR_delete_project()