import unittest
from .header import Header

class HeaderUnittests(unittest.TestCase):
    def test_c_header_guard(self):
        hdr = Header('stdio.yaml')
        self.assertEqual('STDIO_H', hdr.header_guard())

        hdr = Header('LinkedListAllocator.yaml')
        self.assertEqual('LINKED_LIST_ALLOCATOR_H', hdr.header_guard())

    def test_mock_header_guard(self):
        hdr = Header('stdio.yaml')
        self.assertEqual('MOCK_STDIO_HH', hdr.header_guard(mock=True))

        hdr = Header('LinkedListAllocator.yaml')
        self.assertEqual('MOCK_LINKED_LIST_ALLOCATOR_HH', hdr.header_guard(mock=True))