Example #1
0
    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)
Example #3
0
  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))
Example #4
0
    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))
Example #5
0
    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)
Example #6
0
    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)
Example #7
0
 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))
Example #8
0
    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)
Example #9
0
    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)
Example #10
0
    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)
Example #11
0
    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))
Example #12
0
    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)
Example #13
0
    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)
Example #14
0
  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)
Example #15
0
  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)))
Example #16
0
    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))
Example #17
0
    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)
Example #18
0
    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)
Example #19
0
    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)