def testEnums(self): """Tests that enum statements are correctly parsed.""" source = """\ module my_module { enum MyEnum1 { VALUE1, VALUE2 }; // No trailing comma. enum MyEnum2 { VALUE1 = -1, VALUE2 = 0, VALUE3 = + 987, // Check that space is allowed. VALUE4 = 0xAF12, VALUE5 = -0x09bcd, VALUE6 = VALUE5, VALUE7, // Leave trailing comma. }; } // my_module """ expected = ast.Mojom( ast.Module(('IDENTIFIER', 'my_module'), None), ast.ImportList(), [ast.Enum( 'MyEnum1', ast.EnumValueList([ast.EnumValue('VALUE1', None), ast.EnumValue('VALUE2', None)])), ast.Enum( 'MyEnum2', ast.EnumValueList([ast.EnumValue('VALUE1', '-1'), ast.EnumValue('VALUE2', '0'), ast.EnumValue('VALUE3', '+987'), ast.EnumValue('VALUE4', '0xAF12'), ast.EnumValue('VALUE5', '-0x09bcd'), ast.EnumValue('VALUE6', ('IDENTIFIER', 'VALUE5')), ast.EnumValue('VALUE7', None)]))]) self.assertEquals(parser.Parse(source, "my_file.mojom"), expected)
def p_enum_1(self, p): """enum : attribute_section ENUM NAME LBRACE enum_value_list \ RBRACE SEMI | attribute_section ENUM NAME LBRACE nonempty_enum_value_list \ COMMA RBRACE SEMI""" p[0] = ast.Enum( p[3], p[1], p[5], filename=self.filename, lineno=p.lineno(2))
def testValidInterfaceDefinitions(self): """Tests all types of definitions that can occur in an interface.""" source = """\ interface MyInterface { enum MyEnum { VALUE }; const int32 kMyConst = 123; MyMethod(int32 x) => (MyEnum y); }; """ expected = ast.Mojom( None, ast.ImportList(), [ast.Interface( 'MyInterface', None, ast.InterfaceBody( [ast.Enum('MyEnum', None, ast.EnumValueList( ast.EnumValue('VALUE', None, None))), ast.Const('kMyConst', 'int32', '123'), ast.Method( 'MyMethod', None, None, ast.ParameterList(ast.Parameter('x', None, None, 'int32')), ast.ParameterList(ast.Parameter('y', None, None, 'MyEnum')))]))]) self.assertEquals(parser.Parse(source, "my_file.mojom"), expected)
def testValidStructDefinitions(self): """Tests all types of definitions that can occur in a struct.""" source = """\ struct MyStruct { enum MyEnum { VALUE }; const double kMyConst = 1.23; int32 a; SomeOtherStruct b; // Invalidity detected at another stage. }; """ expected = ast.Mojom( None, ast.ImportList(), [ast.Struct( 'MyStruct', None, ast.StructBody( [ast.Enum('MyEnum', None, ast.EnumValueList( ast.EnumValue('VALUE', None, None))), ast.Const('kMyConst', 'double', '1.23'), ast.StructField('a', None, None, 'int32', None), ast.StructField('b', None, None, 'SomeOtherStruct', None)]))]) self.assertEquals(parser.Parse(source, "my_file.mojom"), expected)
def p_enum_2(self, p): """enum : attribute_section ENUM NAME SEMI""" p[0] = ast.Enum(p[3], p[1], None, filename=self.filename, lineno=p.lineno(2))
def testValidAttributes(self): """Tests parsing attributes (and attribute lists).""" # Note: We use structs because they have (optional) attribute lists. # Empty attribute list. source1 = "[] struct MyStruct {};" expected1 = ast.Mojom( None, ast.ImportList(), [ast.Struct('MyStruct', ast.AttributeList(), ast.StructBody())]) self.assertEquals(parser.Parse(source1, "my_file.mojom"), expected1) # One-element attribute list, with name value. source2 = "[MyAttribute=MyName] struct MyStruct {};" expected2 = ast.Mojom( None, ast.ImportList(), [ast.Struct( 'MyStruct', ast.AttributeList(ast.Attribute("MyAttribute", "MyName")), ast.StructBody())]) self.assertEquals(parser.Parse(source2, "my_file.mojom"), expected2) # Two-element attribute list, with one string value and one integer value. source3 = "[MyAttribute1 = \"hello\", MyAttribute2 = 5] struct MyStruct {};" expected3 = ast.Mojom( None, ast.ImportList(), [ast.Struct( 'MyStruct', ast.AttributeList([ast.Attribute("MyAttribute1", "hello"), ast.Attribute("MyAttribute2", 5)]), ast.StructBody())]) self.assertEquals(parser.Parse(source3, "my_file.mojom"), expected3) # Various places that attribute list is allowed. source4 = """\ [Attr0=0] module my_module; [Attr1=1] struct MyStruct { [Attr2=2] int32 a; }; [Attr3=3] union MyUnion { [Attr4=4] int32 a; }; [Attr5=5] enum MyEnum { [Attr6=6] a }; [Attr7=7] interface MyInterface { [Attr8=8] MyMethod([Attr9=9] int32 a) => ([Attr10=10] bool b); }; """ expected4 = ast.Mojom( ast.Module(('IDENTIFIER', 'my_module'), ast.AttributeList([ast.Attribute("Attr0", 0)])), ast.ImportList(), [ast.Struct( 'MyStruct', ast.AttributeList(ast.Attribute("Attr1", 1)), ast.StructBody( ast.StructField( 'a', ast.AttributeList([ast.Attribute("Attr2", 2)]), None, 'int32', None))), ast.Union( 'MyUnion', ast.AttributeList(ast.Attribute("Attr3", 3)), ast.UnionBody( ast.UnionField( 'a', ast.AttributeList([ast.Attribute("Attr4", 4)]), None, 'int32'))), ast.Enum( 'MyEnum', ast.AttributeList(ast.Attribute("Attr5", 5)), ast.EnumValueList( ast.EnumValue( 'VALUE', ast.AttributeList([ast.Attribute("Attr6", 6)]), None))), ast.Interface( 'MyInterface', ast.AttributeList(ast.Attribute("Attr7", 7)), ast.InterfaceBody( ast.Method( 'MyMethod', ast.AttributeList(ast.Attribute("Attr8", 8)), None, ast.ParameterList( ast.Parameter( 'a', ast.AttributeList([ast.Attribute("Attr9", 9)]), None, 'int32')), ast.ParameterList( ast.Parameter( 'b', ast.AttributeList([ast.Attribute("Attr10", 10)]), None, 'bool')))))]) self.assertEquals(parser.Parse(source4, "my_file.mojom"), expected4)