import unittest
import glob
import zcc
import outstream
import ppci
import io
import os


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

    def do(self, filenames):
        basedir = 'c3examples'
        filenames = [os.path.join(basedir, fn) for fn in filenames]
        args = zcc.parser.parse_args(filenames + [])
        self.assertEqual(0, zcc.main(args))

    def t2estExamples(self):
        """ Test all examples in the c3/examples directory """
        example_filenames = glob.glob('./c3examples/*.c3')
        for filename in example_filenames:
            self.do(filename)

    def testBurn(self):
        self.do(['stm32f4xx.c3', 'burn.c3'])

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

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

    def testCast(self):
        self.do(['cast.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()
        self.assertTrue(zcc.zcc([f], 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()