#!/usr/bin/env python3
# Copyright 2021 The Pigweed Authors
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may not
# use this file except in compliance with the License. You may obtain a copy of
# the License at
#
#     https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations under
# the License.
"""Finds files for a given product."""

import unittest

from pw_stm32cube_build import inject_init


class AddPreMainInitTest(unittest.TestCase):
    """add_pre_main_init tests."""

    def test_spaces(self):
        startup = '\n'.join(
            [
                '/* Call the clock system intitialization function.*/',
                '  bl  SystemInit   ',
                '/* Call static constructors */',
                '    bl __libc_init_array',
                '/* Call the application\'s entry point.*/',
                '  bl  main',
                '  bx  lr    ',
                '.size  Reset_Handler, .-Reset_Handler',
            ]
        )

        new_startup = inject_init.add_pre_main_init(startup)

        self.assertEqual(
            new_startup,
            '\n'.join(
                [
                    '/* Call the clock system intitialization function.*/',
                    '  bl  SystemInit   ',
                    '/* Call static constructors */',
                    '    bl __libc_init_array',
                    '/* Call the application\'s entry point.*/',
                    'bl pw_stm32cube_Init',
                    '  bl  main',
                    '  bx  lr    ',
                    '.size  Reset_Handler, .-Reset_Handler',
                ]
            ),
        )

    def test_tabs(self):
        startup = '\n'.join(
            [
                'LoopFillZerobss:',
                '	ldr	r3, = _ebss',
                '	cmp	r2, r3',
                '	bcc	FillZerobss',
                '/* Call static constructors */',
                '  bl __libc_init_array',
                '/* Call the application\'s entry point.*/',
                '	bl	main',
                '',
                'LoopForever:',
                '    b LoopForever',
            ]
        )

        new_startup = inject_init.add_pre_main_init(startup)

        self.assertEqual(
            new_startup,
            '\n'.join(
                [
                    'LoopFillZerobss:',
                    '	ldr	r3, = _ebss',
                    '	cmp	r2, r3',
                    '	bcc	FillZerobss',
                    '/* Call static constructors */',
                    '  bl __libc_init_array',
                    '/* Call the application\'s entry point.*/',
                    'bl pw_stm32cube_Init',
                    '	bl	main',
                    '',
                    'LoopForever:',
                    '    b LoopForever',
                ]
            ),
        )

    def test_main_not_found(self):
        startup = '\n'.join(
            [
                '/* Call the clock system intitialization function.*/',
                '  bl  SystemInit   ',
                '/* Call static constructors */',
                '    bl __libc_init_array',
                '/* Call the application\'s entry point.*/',
                '  bl  application_entry',
                '  bx  lr    ',
                '.size  Reset_Handler, .-Reset_Handler',
            ]
        )

        with self.assertRaises(ValueError):
            inject_init.add_pre_main_init(startup)


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