import unittest
from codegen.types import Variable, Function, Struct
from .cgen import as_str

test_variable_list = [
    (Variable('i', typedef='int'), 'int i;'),
]

test_function_list = [
    (Function('f', result='void'), 'void f();'),
    (Function('sum', result='uint32_t', args={'a': 'uint32_t', 'b': 'uint32_t'}),
     'uint32_t sum(uint32_t a, uint32_t b);'),
    (Function('printf', result='void', args={'format': 'const char *restrict', '...': None}),
     'void printf(const char *restrict format, ...);'),
]

test_struct_list = [
    (Struct('xyz', [ Variable('i', typedef='int'), Function('f', result='void') ]),
"""typedef struct {
  int i;
  void (*f)();
} xyz;"""),
]

class AsStrUnittests(unittest.TestCase):
    def test_as_str_variable(self):
        for item, expect in test_variable_list:
            with self.subTest(expect):
                self.assertEqual(as_str(item), expect)

    def test_as_str_function(self):
        for item, expect in test_function_list:
            with self.subTest(expect):
                self.assertEqual(as_str(item), expect)

    def test_as_str_struct(self):
        for item, expect in test_struct_list:
            with self.subTest(expect):
                self.assertEqual(as_str(item), expect)