Handle contained declarations for interfaces and structs.

mojo/public/tools/bindings/pylib/mojom/generate/mojom_translator_unittest.py

 # Copyright 2015 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import unittest
 import module
 
 try:
   import mojom_translator
   import mojom_files_mojom
@@ skipping 220 matching lines @@
         decl_data=mojom_types_mojom.DeclarationData(short_name='val2'),
         enum_type_key='AnEnum',
         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), None)
+        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(value1.int_value, enum.fields[0].value)
     self.assertEquals(value2.int_value, enum.fields[1].value)
@@ skipping 189 matching lines @@
           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)))
+    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)))
+    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.PopulateContainedDeclarationsFromMojom(
+        struct, contained_declarations)
+
+    self.assertEquals(
+        'parent.' + 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)
+
 
 @unittest.skipUnless(bindings_imported, 'Could not import python bindings.')
 class TestEvalValue(unittest.TestCase):
 
   def test_literal_value(self):
     mojom = mojom_types_mojom.Value()
     mojom.literal_value = mojom_types_mojom.LiteralValue(int64_value=20)
 
     graph = mojom_files_mojom.MojomFileGraph()
     const = mojom_translator.FileTranslator(graph, None).EvalValue(mojom)
@@ skipping 184 matching lines @@
     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))
 
 if __name__ == '__main__':
   unittest.main()