import unittest
import zcc
import outstream
import ppci
import io
import os
import target


class ZccTestCase(unittest.TestCase):
    """ Tests the compiler driver """

    def do(self, filenames, imps=[]):
        basedir = os.path.join('..', 'examples', 'c3')
        arg_list = [os.path.join(basedir, fn) for fn in filenames]
        for fn in imps:
            arg_list.append('-i')
            arg_list.append(os.path.join(basedir, fn))
        arg_list.append('--target')
        arg_list.append('arm')
        args = zcc.parser.parse_args(arg_list)
        self.assertEqual(0, zcc.main(args))

    def testBurn2(self):
        self.do(['burn2.c3'], ['stm32f4xx.c3'])

    def testComments(self):
        self.do(['comments.c3'])

    def testCast(self):
        self.do(['cast.c3'])

    def testFunctions(self):
        self.do(['functions.c3'])

    def testSectionAddress(self):
        src = """module tst;
            function void t2() {var int t3; t3 = 2;}
        """
        f = io.StringIO(src)
        diag = ppci.DiagnosticsManager()
        outs = outstream.TextOutputStream()
        tg = target.armtarget
        self.assertTrue(zcc.zcc([f], [], tg, outs, diag))
        code = outs.getSection('code')
        self.assertEqual(0x08000000, code.address)
        data = outs.getSection('data')
        self.assertEqual(0x20000000, data.address)


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