115 lines
3.7 KiB
Python
115 lines
3.7 KiB
Python
#!/usr/bin/env python3
|
|
# Copyright 2022 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.
|
|
"""Tests for bloaty configuration tooling."""
|
|
|
|
import unittest
|
|
|
|
from pw_bloat import bloaty_config
|
|
|
|
|
|
class BloatyConfigTest(unittest.TestCase):
|
|
"""Tests that the bloaty config tool produces the expected config."""
|
|
def test_map_segments_to_memory_regions(self) -> None:
|
|
"""Ensures the mapping works correctly based on a real example."""
|
|
segments = {
|
|
3: (int(0x800f268), int(0x8100200)),
|
|
5: (int(0x20004650), int(0x20020650)),
|
|
6: (int(0x20020650), int(0x20030000)),
|
|
1: (int(0x8000200), int(0x800f060)),
|
|
4: (int(0x20000208), int(0x20004650)),
|
|
2: (int(0x20000000), int(0x20000208)),
|
|
0: (int(0x8000000), int(0x8000200)),
|
|
}
|
|
memory_regions = {
|
|
'FLASH': {
|
|
0: (int(0x8000200), int(0x8100200))
|
|
},
|
|
'RAM': {
|
|
0: (int(0x20000000), int(0x20030000))
|
|
},
|
|
'VECTOR_TABLE': {
|
|
0: (int(0x8000000), int(0x8000200))
|
|
},
|
|
}
|
|
expected = {
|
|
3: 'FLASH',
|
|
5: 'RAM',
|
|
6: 'RAM',
|
|
1: 'FLASH',
|
|
4: 'RAM',
|
|
2: 'RAM',
|
|
0: 'VECTOR_TABLE',
|
|
}
|
|
actual = bloaty_config.map_segments_to_memory_regions(
|
|
segments=segments, memory_regions=memory_regions)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_generate_memoryregions_data_source(self) -> None:
|
|
"""Ensures the formatted generation works correctly."""
|
|
segments_to_memory_regions = {
|
|
0: 'RAM',
|
|
1: 'RAM',
|
|
13: 'FLASH',
|
|
}
|
|
config = bloaty_config.generate_memoryregions_data_source(
|
|
segments_to_memory_regions)
|
|
expected = '\n'.join((
|
|
r'custom_data_source: {',
|
|
r' name: "memoryregions"',
|
|
r' base_data_source: "segments"',
|
|
r' rewrite: {',
|
|
r' pattern:"^LOAD #0 \\[.*\\]$"',
|
|
r' replacement:"RAM"',
|
|
r' }',
|
|
r' rewrite: {',
|
|
r' pattern:"^LOAD #1 \\[.*\\]$"',
|
|
r' replacement:"RAM"',
|
|
r' }',
|
|
r' rewrite: {',
|
|
r' pattern:"^LOAD #13 \\[.*\\]$"',
|
|
r' replacement:"FLASH"',
|
|
r' }',
|
|
r' rewrite: {',
|
|
r' pattern:".*"',
|
|
r' replacement:"Not resident in memory"',
|
|
r' }',
|
|
r'}',
|
|
r'',
|
|
))
|
|
self.assertEqual(expected, config)
|
|
|
|
def test_generate_utilization_data_source(self) -> None:
|
|
config = bloaty_config.generate_utilization_data_source()
|
|
expected = '\n'.join((
|
|
'custom_data_source: {',
|
|
' name:"utilization"',
|
|
' base_data_source:"sections"',
|
|
' rewrite: {',
|
|
' pattern:"unused_space"',
|
|
' replacement:"Free space"',
|
|
' }',
|
|
' rewrite: {',
|
|
' pattern:".*"',
|
|
' replacement:"Used space"',
|
|
' }',
|
|
'}',
|
|
'',
|
|
))
|
|
self.assertEqual(expected, config)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|