#!/usr/bin/env python
# TestDriverHelper.py
#
#   Copyright (C) 2006 CCLRC, Graeme Winter
#
#   This code is distributed under the BSD license, a copy of which is 
#   included in the root directory of this package.
#
# 25th May 2006
# 
# Tests for the Driver Helper functions.
# 

import os, sys

if not os.environ.has_key('XIA2CORE_ROOT'):
    raise RuntimeError, 'XIA2CORE_ROOT not defined'

sys.path.append(os.path.join(os.environ['XIA2CORE_ROOT'],
                             'Python',
                             'Driver'))

import DriverHelper

# this is used to execute test scripts
import subprocess

import unittest

class TestDriverHelper(unittest.TestCase):

    def setUp(self):
        pass

    def testscript(self):
        '''Test the DriverHelper script writing function.'''

        DriverHelper.script_writer(os.getcwd(),
                                   'testscript',
                                   'ExampleProgramCommandLineStandardInput',
                                   ['foobar'],
                                   os.environ,
                                   ['4\n'])

        if os.name == 'nt':
            subprocess.call(['testscript.bat'])
        else:
            subprocess.call(['bash', 'testscript.sh'])

        results = open('testscript.xout', 'r').readlines()

        self.assertEqual(len(results), 4)
        self.assertEqual(results[0].strip(), 'Hello, foobar!')

        return

    def testrandomnames(self):
        '''Test that the random name generation works.'''

        names = []
        for i in range(1000):
            n = DriverHelper.generate_random_name()
            if not n in names:
                names.append(n)

        self.assertEqual(len(names), 1000)

if __name__ == '__main__':
    unittest.main()