def test_unions(self):
file_name = 'a.mojom'
mojom_union = mojom_types_mojom.MojomUnion()
mojom_union.decl_data = mojom_types_mojom.DeclarationData(
short_name='AUnion',
source_file_info=mojom_types_mojom.SourceFileInfo(file_name=file_name))
field1 = mojom_types_mojom.UnionField(
decl_data=mojom_types_mojom.DeclarationData(short_name='field1'),
type=mojom_types_mojom.Type(
simple_type=mojom_types_mojom.SimpleType.BOOL))
field2 = mojom_types_mojom.UnionField(
decl_data=mojom_types_mojom.DeclarationData(
short_name='field2', declared_ordinal=5),
type=mojom_types_mojom.Type(
simple_type=mojom_types_mojom.SimpleType.DOUBLE))
mojom_union.fields = [field1, field2]
graph = mojom_files_mojom.MojomFileGraph()
union = module.Union()
translator = mojom_translator.FileTranslator(graph, file_name)
translator.UnionFromMojom(
union, mojom_types_mojom.UserDefinedType(union_type=mojom_union))
self.assertEquals(translator._module, union.module)
self.assertEquals('AUnion', union.name)
self.assertEquals(len(mojom_union.fields), len(union.fields))
for gold, f in zip(mojom_union.fields, union.fields):
self.assertEquals(gold.decl_data.short_name, f.name)
self.assertEquals(module.BOOL, union.fields[0].kind)
self.assertEquals(None, union.fields[0].ordinal)
self.assertEquals(module.DOUBLE, union.fields[1].kind)
self.assertEquals(5, union.fields[1].ordinal)
def UnionFromData(module, data):
union = mojom.Union(module=module)
union.name = data['name']
union.spec = 'x:' + module.namespace + '.' + union.name
module.kinds[union.spec] = union
# Stash fields data here temporarily.
union.fields_data = data['fields']
return union
def test_unions(self):
file_name = 'a.mojom'
mojom_union = mojom_types_mojom.MojomUnion()
mojom_union.decl_data = mojom_types_mojom.DeclarationData(
short_name='AUnion',
source_file_info=mojom_types_mojom.SourceFileInfo(
file_name=file_name))
field1 = mojom_types_mojom.UnionField(
decl_data=mojom_types_mojom.DeclarationData(short_name='field1',
declaration_order=0,
declared_ordinal=7),
type=mojom_types_mojom.Type(
simple_type=mojom_types_mojom.SimpleType.BOOL),
tag=7)
field2 = mojom_types_mojom.UnionField(
decl_data=mojom_types_mojom.DeclarationData(short_name='field2',
declaration_order=1),
type=mojom_types_mojom.Type(
simple_type=mojom_types_mojom.SimpleType.DOUBLE),
tag=8)
field3 = mojom_types_mojom.UnionField(
decl_data=mojom_types_mojom.DeclarationData(short_name='field3',
declaration_order=2,
declared_ordinal=0),
type=mojom_types_mojom.Type(
simple_type=mojom_types_mojom.SimpleType.INT32),
tag=0)
mojom_union.fields = [field3, field1, field2]
# mojom_fields_declaration_order lists, in declaration order, the indices
# of the fields in mojom_union.fields
mojom_fields_declaration_order = [1, 2, 0]
graph = mojom_files_mojom.MojomFileGraph()
union = module.Union()
translator = mojom_translator.FileTranslator(graph, file_name)
translator.UnionFromMojom(
union, mojom_types_mojom.UserDefinedType(union_type=mojom_union))
self.assertEquals(translator._module, union.module)
self.assertEquals('AUnion', union.name)
self.assertEquals(len(mojom_union.fields), len(union.fields))
for index, gold_index in enumerate(mojom_fields_declaration_order):
gold = mojom_union.fields[gold_index]
f = union.fields[index]
self.assertEquals(gold.decl_data.short_name, f.name)
if gold.decl_data.declared_ordinal >= 0:
self.assertEquals(gold.decl_data.declared_ordinal,
f.declared_tag)
else:
self.assertEquals(None, f.declared_tag)
self.assertEquals(gold.tag, f.ordinal)
self.assertEquals(module.BOOL, union.fields[0].kind)
self.assertEquals(module.DOUBLE, union.fields[1].kind)
self.assertEquals(module.INT32, union.fields[2].kind)
def _Union(module, parsed_union):
"""
Args:
module: {mojom.Module} Module currently being constructed.
parsed_union: {ast.Union} Parsed union.
Returns:
{mojom.Union} AST union.
"""
union = mojom.Union(module=module)
union.mojom_name = parsed_union.mojom_name
union.spec = 'x:' + module.mojom_namespace + '.' + union.mojom_name
module.kinds[union.spec] = union
# Stash fields parsed_union here temporarily.
union.fields_data = _ElemsOfType(parsed_union.body, ast.UnionField,
parsed_union.mojom_name)
union.attributes = _AttributeListToDict(parsed_union.attribute_list)
return union
