def test_to_vtype_with_layout_with_one_field(self):
layout_to_convert = layout.Layout(
bit_size=32,
bit_alignment=16,
fields=[
layout.Field(
bit_offset=0,
name='x',
layout=layout.Layout(
bit_size=32,
bit_alignment=16,
)
)
],
)
type_definition = c_ast.CStruct([
c_ast.CField(
name='x',
type_definition=c_ast.CTypeReference('some_type'),
),
])
self.type_description_visitor.get_description.return_value = (
'some_description'
)
actual = self.layout_to_vtype_converter.to_vtype(
layout_to_convert,
type_definition,
'some_types',
)
self.type_description_visitor.get_description.assert_called_with(
c_ast.CTypeReference('some_type'),
'some_types',
)
expected = [4, {'x': [0, 'some_description']}]
self.assertEqual(actual, expected)
def test_to_vtype_with_layout_with_bit_field(self):
layout_to_convert = layout.Layout(
bit_size=48,
bit_alignment=16,
fields=[
layout.Field(
bit_offset=33,
name='x',
layout=layout.Layout(
bit_size=9,
bit_alignment=16,
bit_field=True,
)
)
],
)
type_definition = c_ast.CStruct([
c_ast.CField(
name='x',
type_definition=c_ast.CTypeReference('some_type'),
bit_size=9,
),
])
self.type_description_visitor.get_description.return_value = [
'some_type']
actual = self.layout_to_vtype_converter.to_vtype(
layout_to_convert,
type_definition,
'some_types',
)
self.type_description_visitor.get_description.assert_called_with(
c_ast.CTypeReference('some_type'),
'some_types',
)
expected = [
6, {
'x': [
4, [
'BitField', {
'start_bit': 1,
'end_bit': 10,
'target': 'some_type',
}
],
],
},
]
self.assertEqual(actual, expected)
def _create_type_name_with_fields(self, tok):
"""Creates CField() list from parsed token."""
result = []
for field in tok["field"]:
field.type_definition = tok.type_definition
bit_size = None
if field.bitfield:
bit_size = self.expression_parser.parse(field.bitfield)
type_definition = self._create_type_definition(
tok.type_definition, field)
field_ast = c_ast.CField(name=field.type_instance_name.first,
bit_size=bit_size,
attributes=tok.attributes,
type_definition=type_definition)
field_ast.add_attributes(field.attributes)
result.append(field_ast)
return result
def test_to_vtype_with_layout_with_three_fields(self):
layout_to_convert = layout.Layout(
bit_size=48,
bit_alignment=32,
fields=[
layout.Field(
bit_offset=0,
name='x',
layout=layout.Layout(
bit_size=16,
bit_alignment=8,
)
),
layout.Field(
bit_offset=16,
name='y',
layout=layout.Layout(
bit_size=8,
bit_alignment=32,
)
),
layout.Field(
bit_offset=24,
name='z',
layout=layout.Layout(
bit_size=24,
bit_alignment=16,
)
),
],
)
some_type = c_ast.CTypeReference('some_type')
some_other_type = c_ast.CTypeReference('some_other_type')
type_definition = c_ast.CStruct([
c_ast.CField(
name='x',
type_definition=some_type,
),
c_ast.CField(
name='y',
type_definition=some_other_type,
),
c_ast.CField(
name='z',
type_definition=some_type,
),
])
self.type_description_visitor.get_description.side_effect = (
'some_description',
'some_other_description',
'some_description',
)
actual = self.layout_to_vtype_converter.to_vtype(
layout_to_convert,
type_definition,
'some_types',
)
expected_get_description_calls = [
mock.call(some_type, 'some_types'),
mock.call(some_other_type, 'some_types'),
mock.call(some_type, 'some_types'),
]
self.assertEqual(
self.type_description_visitor.get_description.call_args_list,
expected_get_description_calls,
)
expected = [
6, {
'x': [0, 'some_description'],
'y': [2, 'some_other_description'],
'z': [3, 'some_description'],
}
]
self.assertEqual(actual, expected)