import logging
import os
import unittest

import testchill.chill
import testchill.gcov
import testchill.test
import testchill.util


_runbuild=True

def runtest(tclist):
    if _runbuild:
        for tc in tclist:
            tc.setUp()
            tc.setresult(tc.run())
            tc.tearDown()

def runchilltest(tclist, runvalidate=False, runstdout=False):
    for tc in tclist:
        tc.setUp()
        tc.compile_src(tc)
        tc.run_script(tc)
        tc.compile_gensrc(tc)
        if runvalidate:
            tc.check_run_script_validate(tc)
        if runstdout:
            tc.check_run_script_stdout(tc)
        tc.tearDown()

class TestChillTestCases(unittest.TestCase):
    def config(self, **kwargs):
        cargs = {
                'omega_dir': self.omega_dev_dir,
                'chill_dir': self.chill_dev_dir,
                'bin_dir': self.bin_dir,
                'build_cuda': False,
                'script_lang': None,
                'version': 'dev'
            }
        cargs.update(kwargs)
        return testchill.chill.ChillConfig(**cargs)
    
    def config_rel(self, **kwargs):
        kwargs['version'] = 'release'
        kwargs['omega_dir'] = self.omega_rel_dir
        kwargs['chill_dir'] = self.chill_rel_dir
        return self.config(**kwargs)
    
    def setUp(self):
        self.chill_dev_dir = os.getenv('CHILL_DEV_SRC')
        self.chill_rel_dir = os.getenv('CHILL_RELEASE_SRC')
        self.omega_dev_dir = os.getenv('OMEGA_DEV_SRC')
        self.omega_rel_dir = os.getenv('OMEGA_RELEASE_SRC')
        self.bin_dir = os.getenv('STAGING_DIR_BIN')
        self.wd = os.getenv('STAGING_DIR_WD')
        self.build_options = {'coverage':False}
        
        testchill.util.shell('cp', [os.path.join(self.chill_dev_dir, 'examples/cuda-chill/cudaize.lua'), self.wd])
        testchill.util.shell('cp', [os.path.join(self.chill_dev_dir, 'examples/cuda-chill/cudaize.py'), self.wd])
        
        self.config_test_func = {
                0: lambda conf: conf.default_script_lang(),
                1: lambda conf: conf.name(),
                2: lambda conf: conf.make_depend_target(),
                3: lambda conf: conf.make_target(),
                4: lambda conf: conf.make_args()
            }
        self.config_test_data = [
                ((self.omega_dev_dir, self.chill_dev_dir, self.bin_dir, False, None,     'dev'),     ('script', 'chill',              'depend-chill',      'chill')),
                ((self.omega_dev_dir, self.chill_dev_dir, self.bin_dir, False, 'lua',    'dev'),     ('script', 'chill-lua',          'depend-chill',      'chill')),
                ((self.omega_dev_dir, self.chill_dev_dir, self.bin_dir, False, 'python', 'dev'),     ('script', 'chill-python',       'depend-chill',      'chill')),
                ((self.omega_dev_dir, self.chill_dev_dir, self.bin_dir, True,  None,     'dev'),     ('lua',    'cuda-chill',         'depend-cuda-chill', 'cuda-chill')),
                ((self.omega_dev_dir, self.chill_dev_dir, self.bin_dir, True,  'python', 'dev'),     ('lua',    'cuda-chill-python',  'depend-cuda-chill', 'cuda-chill')),
                ((self.omega_rel_dir, self.chill_rel_dir, self.bin_dir, False, None,     'release'), ('script', 'chill-release',      'depend',            'chill')),
                ((self.omega_rel_dir, self.chill_rel_dir, self.bin_dir, True,  None,     'release'), ('lua',    'cuda-chill-release', 'depend-cuda-chill', 'cuda-chill'))
            ]
    
    def tearDown(self):
        pass
    
    def _run_ChillConfig_test(self, n):
        for args, expected in self.config_test_data:
            val = self.config_test_func[n](testchill.chill.ChillConfig(*args))
            exp = expected[n]
            self.assertEqual(val, exp)
    
    def test_ChillConfig_default_script_lang(self):
        self._run_ChillConfig_test(0)
    
    def test_ChillConfig_name(self):
        self._run_ChillConfig_test(1)
    
    def test_ChillConfig_make_depend_target(self):
        self._run_ChillConfig_test(2)
    
    def test_ChillConfig_make_target(self):
        self._run_ChillConfig_test(3)
    
    #def test_ChillConfig_make_args(self):
    #    self._run_ChillConfig_test(4)
    
    def test_chill_dev(self):
        tc = testchill.chill.BuildChillTestCase(self.config(), self.build_options)
        self.assertEqual(tc.config.name(), 'chill')
        self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_dev_dir)
        self.assertEqual(tc.config.make_depend_target(), 'depend-chill')
        self.assertEqual(tc.config.make_target(), 'chill')
        self.assertEqual(tc.name, 'chill')
        logging.info('Building ' + tc.name)
        runtest([tc])
    
    def test_chill_dev_lua(self):
        tc = testchill.chill.BuildChillTestCase(self.config(script_lang='lua'), self.build_options)
        self.assertEqual(tc.config.name(), 'chill-lua')
        self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_dev_dir)
        self.assertEqual(tc.config.make_depend_target(), 'depend-chill')
        self.assertEqual(tc.config.make_target(), 'chill')
        self.assertEqual(tc.config.make_args(), 'SCRIPT_LANG=lua')
        self.assertEqual(tc.name, 'chill-lua')
        logging.info('Building ' + tc.name)
        runtest([tc])
    
    def test_chill_dev_python(self):
        tc = testchill.chill.BuildChillTestCase(self.config(script_lang='python'), self.build_options)
        self.assertEqual(tc.config.name(), 'chill-python')
        self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_dev_dir)
        self.assertEqual(tc.config.make_depend_target(), 'depend-chill')
        self.assertEqual(tc.config.make_target(), 'chill')
        self.assertEqual(tc.config.make_args(), 'SCRIPT_LANG=python')
        self.assertEqual(tc.name, 'chill-python')
        logging.info('Building ' + tc.name)
        runtest([tc])
    
    def test_cudachill_dev(self):
        tc = testchill.chill.BuildChillTestCase(self.config(build_cuda=True), self.build_options)
        self.assertEqual(tc.config.name(), 'cuda-chill')
        self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_dev_dir)
        self.assertEqual(tc.config.make_depend_target(), 'depend-cuda-chill')
        self.assertEqual(tc.config.make_target(), 'cuda-chill')
        self.assertEqual(tc.name, 'cuda-chill')
        logging.info('Building ' + tc.name)
        runtest([tc])
    
    def test_cudachill_dev(self):
        tc = testchill.chill.BuildChillTestCase(self.config(build_cuda=True, script_lang='python'), self.build_options)
        self.assertEqual(tc.config.name(), 'cuda-chill-python')
        self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_dev_dir)
        self.assertEqual(tc.config.make_depend_target(), 'depend-cuda-chill')
        self.assertEqual(tc.config.make_target(), 'cuda-chill')
        self.assertEqual(tc.name, 'cuda-chill-python')
        logging.info('Building ' + tc.name)
        runtest([tc])

    def test_chill_release(self):
        tc = testchill.chill.BuildChillTestCase(self.config_rel(), self.build_options)
        self.assertEqual(tc.config.name(), 'chill-release')
        self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_rel_dir)
        self.assertEqual(tc.config.make_depend_target(), 'depend')
        self.assertEqual(tc.config.make_target(), 'chill')
        self.assertEqual(tc.name, 'chill-release')
        logging.info('Building ' + tc.name)
        runtest([tc])
    
    def test_cudachill_release(self):
        tc = testchill.chill.BuildChillTestCase(self.config_rel(build_cuda=True), self.build_options)
        self.assertEqual(tc.config.name(), 'cuda-chill-release')
        self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_rel_dir)
        self.assertEqual(tc.config.env()['CUDACHILL'], 'true')
        self.assertEqual(tc.config.make_depend_target(), 'depend-cuda-chill')
        self.assertEqual(tc.config.make_target(), 'cuda-chill')
        self.assertEqual(tc.name, 'cuda-chill-release')
        logging.info('Building ' + tc.name)
        runtest([tc])
    
    def test_run_chill(self):
        config = self.config()
        btc = testchill.chill.BuildChillTestCase(config, self.build_options)
        runtest([btc])
        tc = testchill.chill.RunChillTestCase(config, 'test-cases/chill/test_scale.script', 'test-cases/chill/mm.c', wd=self.wd)
        self.assertEqual(tc.chill_src, 'mm.c')
        self.assertEqual(tc.chill_script, 'test_scale.script')
        self.assertEqual(tc.chill_src_path, os.path.join(os.getcwd(), 'test-cases/chill/mm.c'))
        self.assertEqual(tc.chill_script_path, os.path.join(os.getcwd(), 'test-cases/chill/test_scale.script'))
        self.assertEqual(tc.chill_gensrc, 'rose_mm.c')
        self.assertEqual(tc.name, 'chill:test_scale.script')
        runchilltest([tc])
    
    def test_run_cudachill(self):
        config = self.config(build_cuda=True)
        btc = testchill.chill.BuildChillTestCase(config, self.build_options)
        runtest([btc])
        tc = testchill.chill.RunChillTestCase(config, 'test-cases/examples/cuda-chill/mm.lua', 'test-cases/examples/cuda-chill/mm.c', wd=self.wd)
        self.assertEqual(tc.chill_src, 'mm.c')
        self.assertEqual(tc.chill_script, 'mm.lua')
        self.assertEqual(tc.chill_src_path, os.path.join(os.getcwd(), 'test-cases/examples/cuda-chill/mm.c'))
        self.assertEqual(tc.chill_script_path, os.path.join(os.getcwd(), 'test-cases/examples/cuda-chill/mm.lua'))
        self.assertEqual(tc.chill_gensrc, 'rose_mm.cu')
        self.assertEqual(tc.name, 'cuda-chill:mm.lua')
        runchilltest([tc])
    
    def test_chill_coverage(self):
        tc = testchill.chill.BuildChillTestCase(self.config(), options={'coverage':True}, coverage_set=testchill.gcov.GcovSet())
        self.assertEqual(tc.config.name(), 'chill')
        self.assertEqual(tc.config.env()['OMEGAHOME'], self.omega_dev_dir)
        self.assertEqual(tc.config.make_depend_target(), 'depend-chill')
        self.assertEqual(tc.config.make_target(), 'chill')
        self.assertEqual(tc.name, 'chill')
        self.assertTrue(tc.options['coverage'])
        logging.info('Building ' + tc.name)
        if _runbuild:
            runtest([tc])
            self.assertTrue(os.path.exists(os.path.join(self.chill_dev_dir, 'ir_rose.gcno')))