def test_child_enum(self): file_name = 'a.mojom' mojom_enum = mojom_types_mojom.MojomEnum() mojom_enum.decl_data = mojom_types_mojom.DeclarationData( short_name='AnEnum', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name), container_type_key='struct_key') mojom_enum.values = [] graph = mojom_files_mojom.MojomFileGraph() mojom_struct = mojom_types_mojom.MojomStruct( fields=[], decl_data=mojom_types_mojom.DeclarationData( short_name='AStruct', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name))) add_version_info(mojom_struct, 0) graph.resolved_types = { 'struct_key': mojom_types_mojom.UserDefinedType(struct_type=mojom_struct) } enum = module.Enum() translator = mojom_translator.FileTranslator(graph, file_name) translator.EnumFromMojom( enum, mojom_types_mojom.UserDefinedType(enum_type=mojom_enum)) self.assertEquals(mojom_enum.decl_data.short_name, enum.name) self.assertEquals(len(mojom_enum.values), len(enum.fields))
def test_constant(self): file_name = 'a.mojom' graph = mojom_files_mojom.MojomFileGraph() mojom_const = mojom_types_mojom.DeclaredConstant() mojom_const.decl_data = mojom_types_mojom.DeclarationData( short_name='foo', container_type_key='struct_key') mojom_const.type = mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.INT64) mojom_const.value = mojom_types_mojom.Value() mojom_const.value.literal_value = mojom_types_mojom.LiteralValue( int64_value=20) mojom_struct = mojom_types_mojom.MojomStruct( fields=[], decl_data=mojom_types_mojom.DeclarationData( short_name='AStruct', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name))) add_version_info(mojom_struct, 0) graph.resolved_types = { 'struct_key': mojom_types_mojom.UserDefinedType(struct_type=mojom_struct) } const = module.Constant() translator = mojom_translator.FileTranslator(graph, file_name) translator.ConstantFromMojom(const, mojom_const) self.assertEquals(mojom_const.decl_data.short_name, const.name) self.assertEquals(module.INT64, const.kind) self.assertEquals('20', const.value) self.assertEquals(translator.UserDefinedFromTypeKey('struct_key'), const.parent_kind)
def test_enum(self): file_name = 'a.mojom' mojom_enum = mojom_types_mojom.MojomEnum() mojom_enum.decl_data = mojom_types_mojom.DeclarationData( short_name='AnEnum', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name)) value1 = mojom_types_mojom.EnumValue( decl_data=mojom_types_mojom.DeclarationData(short_name='val1'), initializer_value=mojom_types_mojom.Value( literal_value=mojom_types_mojom.LiteralValue(uint64_value=20)), int_value=20) value2 = mojom_types_mojom.EnumValue( decl_data=mojom_types_mojom.DeclarationData(short_name='val2'), int_value=70) mojom_enum.values = [value1, value2] graph = mojom_files_mojom.MojomFileGraph() enum = module.Enum() translator = mojom_translator.FileTranslator(graph, file_name) translator.EnumFromMojom( enum, mojom_types_mojom.UserDefinedType(enum_type=mojom_enum)) self.assertEquals(translator._module, enum.module) self.assertEquals(mojom_enum.decl_data.short_name, enum.name) self.assertEquals(len(mojom_enum.values), len(enum.fields)) self.assertEquals(value1.decl_data.short_name, enum.fields[0].name) self.assertEquals(value2.decl_data.short_name, enum.fields[1].name) self.assertEquals('20', enum.fields[0].value) self.assertIsNone(enum.fields[1].value) self.assertEquals(value1.int_value, enum.fields[0].numeric_value) self.assertEquals(value2.int_value, enum.fields[1].numeric_value)
def test_user_defined_type_type(self): graph = mojom_files_mojom.MojomFileGraph() mojom_struct = mojom_types_mojom.MojomStruct( decl_data=mojom_types_mojom.DeclarationData(short_name='FirstStruct')) type_key = 'some opaque string' mojom_struct.fields = [ # Make sure recursive structs are correctly handled. mojom_types_mojom.StructField( decl_data=mojom_types_mojom.DeclarationData(short_name='field00'), type=mojom_types_mojom.Type( type_reference=mojom_types_mojom.TypeReference(type_key=type_key))) ] add_version_info(mojom_struct, 1) graph.resolved_types = { type_key: mojom_types_mojom.UserDefinedType(struct_type=mojom_struct)} mojom_type = mojom_types_mojom.Type() mojom_type.type_reference = mojom_types_mojom.TypeReference( type_key=type_key) t = mojom_translator.FileTranslator(graph, None) result = t.KindFromMojom(mojom_type) self.assertTrue(module.IsStructKind(result)) self.assertEquals(mojom_struct.decl_data.short_name, result.name) self.assertEquals(result, result.fields[0].kind) self.assertEquals(type_key, result.type_key) # Make sure we create only one module object per type. result2 = t.KindFromMojom(mojom_type) self.assertIs(result, result2) # Nullable type reference mojom_type.type_reference.nullable = True nullable_result = t.KindFromMojom(mojom_type) self.assertTrue(module.IsNullableKind(nullable_result))
def test_interface(self): file_name = 'a.mojom' mojom_interface = mojom_types_mojom.MojomInterface( decl_data=mojom_types_mojom.DeclarationData( source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name)), interface_name='AnInterface') mojom_method = mojom_types_mojom.MojomMethod( ordinal=10, decl_data=mojom_types_mojom.DeclarationData( short_name='AMethod', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name)), parameters=mojom_types_mojom.MojomStruct(fields=[])) mojom_interface.methods = {10: mojom_method} interface = module.Interface() graph = mojom_files_mojom.MojomFileGraph() translator = mojom_translator.FileTranslator(graph, file_name) translator.InterfaceFromMojom( interface, mojom_types_mojom.UserDefinedType(interface_type=mojom_interface)) self.assertEquals(translator._module, interface.module) self.assertEquals(mojom_interface.interface_name, interface.name) self.assertEquals(mojom_method.ordinal, interface.methods[0].ordinal)
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 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 do_interface_test(self, specify_service_name): file_name = 'a.mojom' mojom_interface = mojom_types_mojom.MojomInterface( decl_data=mojom_types_mojom.DeclarationData( short_name='AnInterface', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name))) if specify_service_name: mojom_interface.service_name = 'test::TheInterface' mojom_interface.decl_data.attributes = [mojom_types_mojom.Attribute( key='ServiceName', value=mojom_types_mojom.LiteralValue( string_value='test::TheInterface'))] else: mojom_interface.service_name = None mojom_method10 = mojom_types_mojom.MojomMethod( ordinal=10, decl_data=mojom_types_mojom.DeclarationData( short_name='AMethod10', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name)), parameters=mojom_types_mojom.MojomStruct(fields=[])) mojom_method0 = mojom_types_mojom.MojomMethod( ordinal=0, decl_data=mojom_types_mojom.DeclarationData( short_name='AMethod0', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name)), parameters=mojom_types_mojom.MojomStruct(fields=[])) mojom_method7 = mojom_types_mojom.MojomMethod( ordinal=7, decl_data=mojom_types_mojom.DeclarationData( short_name='AMethod10', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name)), parameters=mojom_types_mojom.MojomStruct(fields=[])) mojom_interface.methods = {10: mojom_method10, 0: mojom_method0, 7: mojom_method7} interface = module.Interface() graph = mojom_files_mojom.MojomFileGraph() translator = mojom_translator.FileTranslator(graph, file_name) translator.InterfaceFromMojom(interface, mojom_types_mojom.UserDefinedType( interface_type=mojom_interface)) self.assertEquals(translator._module, interface.module) self.assertEquals('AnInterface', interface.name) self.assertEquals(0, interface.methods[0].ordinal) self.assertEquals(7, interface.methods[1].ordinal) self.assertEquals(10, interface.methods[2].ordinal) if specify_service_name: self.assertEquals('test::TheInterface', interface.service_name) else: self.assertEquals(None, interface.service_name)
def test_structs(self): file_name = 'a.mojom' graph = mojom_files_mojom.MojomFileGraph() mojom_file = mojom_files_mojom.MojomFile(file_name='a.mojom', module_namespace='foo.bar') graph.files = {mojom_file.file_name: mojom_file} mojom_struct = mojom_types_mojom.MojomStruct( decl_data=mojom_types_mojom.DeclarationData( short_name='FirstStruct')) mojom_struct.fields = [ mojom_types_mojom.StructField( decl_data=mojom_types_mojom.DeclarationData( short_name='field01', declared_ordinal=5), type=mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.BOOL)), mojom_types_mojom.StructField( decl_data=mojom_types_mojom.DeclarationData( short_name='field02'), type=mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.DOUBLE), default_value=mojom_types_mojom.DefaultFieldValue( value=mojom_types_mojom.Value( literal_value=mojom_types_mojom.LiteralValue( double_value=15)))), ] mojom_struct.decl_data.source_file_info = mojom_types_mojom.SourceFileInfo( file_name=mojom_file.file_name) struct = module.Struct() translator = mojom_translator.FileTranslator(graph, file_name) translator.StructFromMojom( struct, mojom_types_mojom.UserDefinedType(struct_type=mojom_struct)) self.assertEquals('FirstStruct', struct.name) self.assertEquals(translator._module, struct.module) self.assertEquals(len(mojom_struct.fields), len(struct.fields)) for gold, f in zip(mojom_struct.fields, struct.fields): self.assertEquals(f.name, gold.decl_data.short_name) self.assertEquals(module.BOOL, struct.fields[0].kind) self.assertEquals(5, struct.fields[0].ordinal) self.assertEquals(module.DOUBLE, struct.fields[1].kind) self.assertEquals(None, struct.fields[1].ordinal) self.assertEquals('15.0', struct.fields[1].default)
def test_attributes_none(self): mojom_enum = mojom_types_mojom.MojomEnum() mojom_enum.decl_data = mojom_types_mojom.DeclarationData() graph = mojom_files_mojom.MojomFileGraph() attributes = mojom_translator.FileTranslator( graph, None).AttributesFromMojom(mojom_enum) self.assertFalse(attributes)
def test_constant_value(self): file_name = 'a.mojom' mojom_const = mojom_types_mojom.DeclaredConstant( decl_data=mojom_types_mojom.DeclarationData( short_name='AConst', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name)), type=mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.INT64), value=mojom_types_mojom.Value( literal_value=mojom_types_mojom.LiteralValue(int64_value=30))) user_defined_value = mojom_types_mojom.UserDefinedValue() user_defined_value.declared_constant = mojom_const graph = mojom_files_mojom.MojomFileGraph() graph.resolved_values = {'value_key': user_defined_value} mojom = mojom_types_mojom.Value( user_value_reference=mojom_types_mojom.UserValueReference( identifier='SOMEID', value_key='value_key')) translator = mojom_translator.FileTranslator(graph, file_name) const_value = translator.ValueFromMojom(mojom) self.assertIs(translator.ConstantFromValueKey('value_key'), const_value.constant) self.assertIs(mojom_const.decl_data.short_name, const_value.name)
def test_method(self): file_name = 'a.mojom' mojom_method = mojom_types_mojom.MojomMethod( ordinal=10, decl_data=mojom_types_mojom.DeclarationData( short_name='AMethod', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name))) param1 = mojom_types_mojom.StructField( decl_data=mojom_types_mojom.DeclarationData(short_name='a_param'), type=mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.UINT32)) param2 = mojom_types_mojom.StructField( decl_data=mojom_types_mojom.DeclarationData(short_name='b_param'), type=mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.UINT64)) mojom_method.parameters = mojom_types_mojom.MojomStruct( fields=[param1, param2]) interface = module.Interface() graph = mojom_files_mojom.MojomFileGraph() translator = mojom_translator.FileTranslator(graph, file_name) method = translator.MethodFromMojom(mojom_method, interface) self.assertEquals(mojom_method.decl_data.short_name, method.name) self.assertEquals(interface, method.interface) self.assertEquals(mojom_method.ordinal, method.ordinal) self.assertIsNone(method.response_parameters) self.assertEquals(len(mojom_method.parameters.fields), len(method.parameters)) self.assertEquals(param1.decl_data.short_name, method.parameters[0].name) self.assertEquals(param2.decl_data.short_name, method.parameters[1].name) # Add empty return params. mojom_method.response_params = mojom_types_mojom.MojomStruct(fields=[]) method = translator.MethodFromMojom(mojom_method, interface) self.assertEquals([], method.response_parameters) # Add non-empty return params. mojom_method.response_params.fields = [param1] method = translator.MethodFromMojom(mojom_method, interface) self.assertEquals(param1.decl_data.short_name, method.response_parameters[0].name)
def build_decl_data(short_name): """Builds and returns a DeclarationData with the given short_name. Args: short_name: {str} short_name to use Returns: {mojom_types_mojom.DeclarationData} With the given short_name """ return mojom_types_mojom.DeclarationData(short_name=short_name)
def test_enum_value(self): file_name = 'a.mojom' mojom_enum = mojom_types_mojom.MojomEnum() mojom_enum.decl_data = mojom_types_mojom.DeclarationData( short_name='AnEnum', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name)) value1 = mojom_types_mojom.EnumValue( decl_data=mojom_types_mojom.DeclarationData( short_name='val1', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name)), enum_type_key='enum_key', initializer_value=mojom_types_mojom.Value( literal_value=mojom_types_mojom.LiteralValue(uint64_value=20)), int_value=20) value2 = mojom_types_mojom.EnumValue( decl_data=mojom_types_mojom.DeclarationData(short_name='val2'), enum_type_key='enum_key', int_value=70) mojom_enum.values = [value1, value2] graph = mojom_files_mojom.MojomFileGraph() graph.resolved_types = { 'enum_key': mojom_types_mojom.UserDefinedType(enum_type=mojom_enum) } graph.resolved_values = { 'enum_value1': mojom_types_mojom.UserDefinedValue(enum_value=value1), 'enum_value2': mojom_types_mojom.UserDefinedValue(enum_value=value2), } mojom = mojom_types_mojom.Value( user_value_reference=mojom_types_mojom.UserValueReference( identifier='SOMEID', value_key='enum_value1')) translator = mojom_translator.FileTranslator(graph, file_name) enum_value = translator.ValueFromMojom(mojom) enum = translator.UserDefinedFromTypeKey('enum_key') self.assertIs(enum, enum_value.enum) self.assertIs(value1.decl_data.short_name, enum_value.name)
def test_contained_declarations(self): graph = mojom_files_mojom.MojomFileGraph() file_name = 'root/f.mojom' mojom_enum = mojom_types_mojom.MojomEnum( values=[], decl_data=mojom_types_mojom.DeclarationData( short_name='AnEnum', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name), container_type_key='parent_key')) graph.resolved_types = { 'enum_key': mojom_types_mojom.UserDefinedType(enum_type=mojom_enum)} mojom_const = mojom_types_mojom.DeclaredConstant( decl_data=mojom_types_mojom.DeclarationData( short_name='AConst', container_type_key='parent_key'), type=mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.INT64), value=mojom_types_mojom.Value( literal_value=mojom_types_mojom.LiteralValue( int64_value=30))) user_defined_value = mojom_types_mojom.UserDefinedValue() user_defined_value.declared_constant = mojom_const graph.resolved_values = {'value_key': user_defined_value} contained_declarations = mojom_types_mojom.ContainedDeclarations( enums=['enum_key'], constants=['value_key']) translator = mojom_translator.FileTranslator(graph, file_name) struct = module.Struct(name='parent') translator._type_cache['parent_key'] = struct translator.PopulateContainedDeclarationsFromMojom( struct, contained_declarations) self.assertEquals( mojom_enum.decl_data.short_name, struct.enums[0].name) self.assertEquals(struct, struct.enums[0].parent_kind) self.assertEquals( mojom_const.decl_data.short_name, struct.constants[0].name) self.assertEquals(struct, struct.constants[0].parent_kind)
def test_imported_struct(self): graph = mojom_files_mojom.MojomFileGraph() graph.files = { 'a.mojom': mojom_files_mojom.MojomFile(file_name='a.mojom', specified_file_name='', module_namespace='namespace', imports=['root/c.mojom']), 'root/c.mojom': mojom_files_mojom.MojomFile(file_name='root/c.mojom', specified_file_name='', module_namespace='otherns', imports=[]), } mojom_struct = mojom_types_mojom.MojomStruct() mojom_struct.decl_data = mojom_types_mojom.DeclarationData( short_name='AStruct', source_file_info=mojom_types_mojom.SourceFileInfo( file_name='root/c.mojom')) mojom_struct.fields = [] add_version_info(mojom_struct, 0) type_key = 'some_type_key' graph.resolved_types = { type_key: mojom_types_mojom.UserDefinedType(struct_type=mojom_struct) } struct = module.Struct() # Translate should create the imports. translator = mojom_translator.FileTranslator(graph, 'a.mojom') translator.Translate() struct = translator.UserDefinedFromTypeRef( mojom_types_mojom.Type( type_reference=mojom_types_mojom.TypeReference( type_key=type_key))) self.assertEquals( translator._transitive_imports['root/c.mojom']['module'], struct.module) self.assertEquals(translator._transitive_imports['root/c.mojom'], struct.imported_from)
def test_parameter(self): # Parameters are encoded as fields in a struct. mojom_param = mojom_types_mojom.StructField( decl_data=mojom_types_mojom.DeclarationData(short_name='param0', declared_ordinal=5), type=mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.UINT64), default_value=mojom_types_mojom.Value( literal_value=mojom_types_mojom.LiteralValue(uint64_value=20))) graph = mojom_files_mojom.MojomFileGraph() translator = mojom_translator.FileTranslator(graph, '') param = translator.ParamFromMojom(mojom_param) self.assertEquals(mojom_param.decl_data.short_name, param.name) self.assertEquals(module.UINT64, param.kind) self.assertEquals(mojom_param.decl_data.declared_ordinal, param.ordinal)
def do_interface_test(self, nullable, interface_request): # Build a MojomInterface file_name = 'a.mojom' mojom_interface = mojom_types_mojom.MojomInterface( decl_data=mojom_types_mojom.DeclarationData( short_name='AnInterface', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name))) mojom_interface.methods = {} # Register the MojomInterface in a MojomFileGraph graph = mojom_files_mojom.MojomFileGraph() type_key = 'some_type_key' graph.resolved_types = { type_key: mojom_types_mojom.UserDefinedType(interface_type=mojom_interface) } # Build a reference to the interface. type_ref = mojom_types_mojom.Type( type_reference=mojom_types_mojom.TypeReference( type_key=type_key, nullable=nullable, is_interface_request=interface_request)) # Construct a translator translator = mojom_translator.FileTranslator(graph, file_name) # Translate the MojomInterface referenced by type_ref. interface = translator.UserDefinedFromTypeRef(type_ref) # Check the translation if interface_request: self.assertEquals('AnInterface', interface.kind.name) self.assertEquals('some_type_key', interface.kind.type_key) else: self.assertEquals('AnInterface', interface.name) self.assertEquals('some_type_key', interface.type_key) self.assertEquals(nullable, interface.is_nullable) self.assertEquals(interface_request, isinstance(interface, module.InterfaceRequest))
def test_attributes(self): mojom_enum = mojom_types_mojom.MojomEnum() mojom_enum.decl_data = mojom_types_mojom.DeclarationData() gold = { 'foo': 'bar', 'other': 'thing', 'hello': 'world', 'min_version': 2, 'pi': 3.14159, 'is_happy': True } mojom_enum.decl_data.attributes = [] for key, value in gold.iteritems(): mojom_enum.decl_data.attributes.append( mojom_types_mojom.Attribute(key=key, value=self.literal_value(value))) graph = mojom_files_mojom.MojomFileGraph() attributes = mojom_translator.FileTranslator( graph, None).AttributesFromMojom(mojom_enum) self.assertEquals(gold, attributes)
def test_method(self): file_name = 'a.mojom' mojom_method = mojom_types_mojom.MojomMethod( ordinal=10, min_version=6, decl_data=mojom_types_mojom.DeclarationData( short_name='AMethod', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name))) param1 = mojom_types_mojom.StructField( decl_data=mojom_types_mojom.DeclarationData(short_name='a_param'), type=mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.UINT32), offset=21, bit=6, min_version=11) param2 = mojom_types_mojom.StructField( decl_data=mojom_types_mojom.DeclarationData(short_name='b_param'), type=mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.UINT64), offset=22, bit=7, min_version=12) mojom_method.parameters = mojom_types_mojom.MojomStruct( fields=[param1, param2], version_info=build_version_info(2), decl_data=build_decl_data('Not used')) interface = module.Interface('MyInterface') graph = mojom_files_mojom.MojomFileGraph() translator = mojom_translator.FileTranslator(graph, file_name) method = translator.MethodFromMojom(mojom_method, interface) self.assertEquals(mojom_method.decl_data.short_name, method.name) self.assertEquals(interface, method.interface) self.assertEquals(mojom_method.ordinal, method.ordinal) self.assertEquals(mojom_method.min_version, method.min_version) self.assertIsNone(method.response_parameters) self.assertEquals(len(mojom_method.parameters.fields), len(method.parameters)) self.assertEquals(param1.decl_data.short_name, method.parameters[0].name) self.assertEquals(param2.decl_data.short_name, method.parameters[1].name) self.assertEquals('MyInterface_AMethod_Params', method.param_struct.name) self.assertEquals(len(mojom_method.parameters.fields), len(method.param_struct.fields)) for i in xrange(0, len(mojom_method.parameters.fields)): gold = mojom_method.parameters.fields[i] f = method.param_struct.fields_in_ordinal_order[i] self.assertEquals(gold.decl_data.short_name, f.name) self.assertEquals(gold.offset, f.computed_offset) self.assertEquals(gold.bit, f.computed_bit) self.assertEquals(gold.min_version, f.computed_min_version) # Add empty return params. mojom_method.response_params = mojom_types_mojom.MojomStruct(fields=[]) add_version_info(mojom_method.response_params, 0) add_decl_data(mojom_method.response_params, 'AMethod_Response') method = translator.MethodFromMojom(mojom_method, interface) self.assertEquals([], method.response_parameters) # Add non-empty return params. mojom_method.response_params.fields = [param1] method = translator.MethodFromMojom(mojom_method, interface) self.assertEquals(param1.decl_data.short_name, method.response_parameters[0].name)
def test_structs(self): file_name = 'a.mojom' graph = mojom_files_mojom.MojomFileGraph() mojom_file = mojom_files_mojom.MojomFile(file_name='a.mojom', module_namespace='foo.bar') graph.files = {mojom_file.file_name: mojom_file} mojom_struct = mojom_types_mojom.MojomStruct( decl_data=mojom_types_mojom.DeclarationData( short_name='FirstStruct')) mojom_struct.fields = [ mojom_types_mojom.StructField( decl_data=mojom_types_mojom.DeclarationData( short_name='field03', declaration_order=2), type=mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.BOOL), offset=21, bit=6, min_version=11), mojom_types_mojom.StructField( decl_data=mojom_types_mojom.DeclarationData( short_name='field01', declared_ordinal=1, declaration_order=0), type=mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.BOOL), offset=17, bit=1, min_version=4), mojom_types_mojom.StructField( decl_data=mojom_types_mojom.DeclarationData( short_name='field02', declaration_order=1), type=mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.DOUBLE), offset=0, bit=0, min_version=0, default_value=mojom_types_mojom.DefaultFieldValue( value=mojom_types_mojom.Value( literal_value=mojom_types_mojom.LiteralValue( double_value=15)))), ] mojom_struct.version_info = [ mojom_types_mojom.StructVersion(version_number=0, num_bytes=67, num_fields=1), mojom_types_mojom.StructVersion(version_number=1, num_bytes=76, num_fields=3), ] # mojom_fields_declaration_order lists, in declaration order, the indices # of the fields in mojom_types_mojom.StructField. mojom_fields_declaration_order = [1, 2, 0] mojom_struct.decl_data.source_file_info = mojom_types_mojom.SourceFileInfo( file_name=mojom_file.file_name) struct = module.Struct() translator = mojom_translator.FileTranslator(graph, file_name) translator.StructFromMojom( struct, mojom_types_mojom.UserDefinedType(struct_type=mojom_struct)) self.assertEquals('FirstStruct', struct.name) self.assertEquals(translator._module, struct.module) self.assertEquals(len(mojom_struct.fields), len(struct.fields)) for index, gold_index in enumerate(mojom_fields_declaration_order): gold = mojom_struct.fields[gold_index] f = struct.fields[index] self.assertEquals(f.name, gold.decl_data.short_name) if gold.decl_data.declared_ordinal >= 0: self.assertEquals(gold.decl_data.declared_ordinal, f.ordinal) else: self.assertEquals(None, f.ordinal) self.assertEquals(gold_index, f.computed_ordinal) self.assertEquals(gold.offset, f.computed_offset) self.assertEquals(gold.bit, f.computed_bit) self.assertEquals(gold.min_version, f.computed_min_version) self.assertEquals(struct.fields_in_ordinal_order[index].name, mojom_struct.fields[index].decl_data.short_name) self.assertEquals(2, len(struct.versions)) for i in xrange(0, 2): self.assertEquals(mojom_struct.version_info[i].version_number, struct.versions[i].version) self.assertEquals(mojom_struct.version_info[i].num_bytes, struct.versions[i].num_bytes) self.assertEquals(mojom_struct.version_info[i].num_fields, struct.versions[i].num_fields) self.assertEquals(module.BOOL, struct.fields[0].kind) self.assertEquals(module.DOUBLE, struct.fields[1].kind) self.assertEquals('15.0', struct.fields[1].default)
def test_basics(self): graph = mojom_files_mojom.MojomFileGraph(resolved_types={}) file_name = 'root/f.mojom' imported_file_name = 'other/a.mojom' second_level_imported_file_name = 'something/other.mojom' mojom_file = mojom_files_mojom.MojomFile( file_name=file_name, specified_file_name='specified_file_name', module_namespace='somens', imports=[imported_file_name]) imported_file = mojom_files_mojom.MojomFile( file_name=imported_file_name, specified_file_name='', module_namespace='somens', imports=[second_level_imported_file_name]) second_level_imported_file = mojom_files_mojom.MojomFile( file_name=second_level_imported_file_name, specified_file_name='', module_namespace='somens') graph.files = { file_name: mojom_file, imported_file_name: imported_file, second_level_imported_file_name: second_level_imported_file } mojom_interface = mojom_types_mojom.MojomInterface( methods={}, decl_data=mojom_types_mojom.DeclarationData( short_name='AnInterface', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name))) graph.resolved_types[ 'interface_key'] = mojom_types_mojom.UserDefinedType( interface_type=mojom_interface) mojom_struct = mojom_types_mojom.MojomStruct( fields=[], decl_data=mojom_types_mojom.DeclarationData( short_name='AStruct', full_identifier='foo.AStruct', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name))) add_version_info(mojom_struct, 0) graph.resolved_types['struct_key'] = mojom_types_mojom.UserDefinedType( struct_type=mojom_struct) mojom_union = mojom_types_mojom.MojomUnion( fields=[], decl_data=mojom_types_mojom.DeclarationData( short_name='AUnion', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name))) graph.resolved_types['union_key'] = mojom_types_mojom.UserDefinedType( union_type=mojom_union) mojom_enum = mojom_types_mojom.MojomEnum( values=[], decl_data=mojom_types_mojom.DeclarationData( short_name='AnEnum', source_file_info=mojom_types_mojom.SourceFileInfo( file_name=file_name))) graph.resolved_types['enum_key'] = mojom_types_mojom.UserDefinedType( enum_type=mojom_enum) mojom_const = mojom_types_mojom.DeclaredConstant( decl_data=mojom_types_mojom.DeclarationData(short_name='AConst'), type=mojom_types_mojom.Type( simple_type=mojom_types_mojom.SimpleType.INT64), value=mojom_types_mojom.Value( literal_value=mojom_types_mojom.LiteralValue(int64_value=30))) graph.resolved_constants = {'constant_key': mojom_const} mojom_file.declared_mojom_objects = mojom_files_mojom.KeysByType( interfaces=['interface_key'], structs=['struct_key'], unions=['union_key'], top_level_enums=['enum_key'], top_level_constants=['constant_key']) mod = mojom_translator.FileTranslator(graph, file_name).Translate() self.assertEquals('f.mojom', mod.name) self.assertEquals(mojom_file.specified_file_name, mod.specified_name) self.assertEquals(mojom_file.file_name, mod.path) self.assertEquals(mojom_file.module_namespace, mod.namespace) self.assertEquals(1, len(mod.imports)) self.assertEquals('a.mojom', mod.imports[0]['module_name']) self.assertEquals(imported_file.module_namespace, mod.imports[0]['namespace']) self.assertEquals(imported_file.file_name, mod.imports[0]['module'].path) self.assertEquals(2, len(mod.transitive_imports)) transitive_imports_paths = [ imp['module'].path for imp in mod.transitive_imports ] self.assertIn(imported_file_name, transitive_imports_paths) self.assertIn(second_level_imported_file_name, transitive_imports_paths) self.assertEquals('AnInterface', mod.interfaces[0].name) # Interfaces should be assigned their name as their spec. self.assertEquals('AnInterface', mod.interfaces[0].spec) self.assertEquals(mojom_struct.decl_data.short_name, mod.structs[0].name) # The struct was given a full_identifier so its spec should be that. self.assertEquals(mojom_struct.decl_data.full_identifier, mod.structs[0].spec) self.assertEquals(mojom_union.decl_data.short_name, mod.unions[0].name) # The union was given a short name but not a full_identifier so its spec # should be the short name. self.assertEquals(mojom_union.decl_data.short_name, mod.unions[0].spec) self.assertEquals(mojom_enum.decl_data.short_name, mod.enums[0].name) self.assertEquals(mojom_const.decl_data.short_name, mod.constants[0].name) imported_mod = mojom_translator.FileTranslator( graph, imported_file_name).Translate() self.assertFalse(imported_mod.specified_name)