def test_array_type(self): array_types = [ (False, False, -1), (False, False, 10), (True, False, -1), (True, True, -1), (False, True, -1), (False, True, 10), ] g = mojom_files_mojom.MojomFileGraph() t = mojom_translator.FileTranslator(g, None) for array_nullable, element_nullable, size in array_types: a = mojom_types_mojom.Type() a.array_type = mojom_types_mojom.ArrayType(nullable=array_nullable, fixed_length=size) a.array_type.element_type = mojom_types_mojom.Type( string_type=mojom_types_mojom.StringType( nullable=element_nullable)) result = t.KindFromMojom(a) self.assertTrue(module.IsArrayKind(result)) self.assertTrue(module.IsStringKind(result.kind)) self.assertEquals(array_nullable, module.IsNullableKind(result)) self.assertEquals(element_nullable, module.IsNullableKind(result.kind)) if size < 0: self.assertIsNone(result.length) else: self.assertEquals(size, result.length)
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_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_map_type(self): map_types = [ (False, False), (True, False), (False, True), (True, True), ] g = mojom_files_mojom.MojomFileGraph() t = mojom_translator.FileTranslator(g, None) for map_nullable, value_nullable in map_types: m = mojom_types_mojom.Type() m.map_type = mojom_types_mojom.MapType(nullable=map_nullable) m.map_type.key_type = mojom_types_mojom.Type( string_type=mojom_types_mojom.StringType()) m.map_type.value_type = mojom_types_mojom.Type( handle_type=mojom_types_mojom.HandleType( kind=mojom_types_mojom.HandleType.Kind.SHARED_BUFFER, nullable=value_nullable)) result = t.KindFromMojom(m) self.assertTrue(module.IsMapKind(result)) self.assertTrue(module.IsStringKind(result.key_kind)) self.assertTrue(module.IsSharedBufferKind(result.value_kind)) self.assertEquals(map_nullable, module.IsNullableKind(result)) self.assertEquals(value_nullable, module.IsNullableKind(result.value_kind))
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 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_handle_type(self): handle_types = [ (mojom_types_mojom.HandleType.Kind.UNSPECIFIED, False, module.HANDLE), (mojom_types_mojom.HandleType.Kind.MESSAGE_PIPE, False, module.MSGPIPE), (mojom_types_mojom.HandleType.Kind.DATA_PIPE_CONSUMER, False, module.DCPIPE), (mojom_types_mojom.HandleType.Kind.DATA_PIPE_PRODUCER, False, module.DPPIPE), (mojom_types_mojom.HandleType.Kind.SHARED_BUFFER, False, module.SHAREDBUFFER), (mojom_types_mojom.HandleType.Kind.UNSPECIFIED, True, module.NULLABLE_HANDLE), (mojom_types_mojom.HandleType.Kind.MESSAGE_PIPE, True, module.NULLABLE_MSGPIPE), (mojom_types_mojom.HandleType.Kind.DATA_PIPE_CONSUMER, True, module.NULLABLE_DCPIPE), (mojom_types_mojom.HandleType.Kind.DATA_PIPE_PRODUCER, True, module.NULLABLE_DPPIPE), (mojom_types_mojom.HandleType.Kind.SHARED_BUFFER, True, module.NULLABLE_SHAREDBUFFER), ] g = mojom_files_mojom.MojomFileGraph() t = mojom_translator.FileTranslator(g, None) for mojom, nullable, golden in handle_types: h = mojom_types_mojom.Type() h.handle_type = mojom_types_mojom.HandleType(kind=mojom, nullable=nullable) self.assertEquals(golden, t.KindFromMojom(h))
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_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_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 test_string_type(self): g = mojom_files_mojom.MojomFileGraph() t = mojom_translator.FileTranslator(g, None) s = mojom_types_mojom.Type(string_type=mojom_types_mojom.StringType()) self.assertEquals(module.STRING, t.KindFromMojom(s)) s.string_type.nullable = True self.assertEquals(module.NULLABLE_STRING, t.KindFromMojom(s))
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 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_simple_type(self): simple_types = [ (mojom_types_mojom.SimpleType.BOOL, module.BOOL), (mojom_types_mojom.SimpleType.INT8, module.INT8), (mojom_types_mojom.SimpleType.INT16, module.INT16), (mojom_types_mojom.SimpleType.INT32, module.INT32), (mojom_types_mojom.SimpleType.INT64, module.INT64), (mojom_types_mojom.SimpleType.UINT8, module.UINT8), (mojom_types_mojom.SimpleType.UINT16, module.UINT16), (mojom_types_mojom.SimpleType.UINT32, module.UINT32), (mojom_types_mojom.SimpleType.UINT64, module.UINT64), (mojom_types_mojom.SimpleType.FLOAT, module.FLOAT), (mojom_types_mojom.SimpleType.DOUBLE, module.DOUBLE), ] g = mojom_files_mojom.MojomFileGraph() t = mojom_translator.FileTranslator(g, None) for mojom, golden in simple_types: self.assertEquals( golden, t.KindFromMojom(mojom_types_mojom.Type(simple_type=mojom)))
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_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)
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)