| #!/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() |