def testDefineEnum_Empty(self):
"""Test defining an empty enum."""
enum_descriptor = descriptor.EnumDescriptor()
enum_descriptor.name = 'Empty'
enum_class = definition.define_enum(enum_descriptor, 'whatever')
self.assertEquals('Empty', enum_class.__name__)
self.assertEquals('whatever', enum_class.__module__)
self.assertEquals(enum_descriptor, descriptor.describe_enum(enum_class))
def testDefineEnum_Empty(self):
"""Test defining an empty enum."""
enum_descriptor = descriptor.EnumDescriptor()
enum_descriptor.name = 'Empty'
enum_class = definition.define_enum(enum_descriptor, 'whatever')
self.assertEquals('Empty', enum_class.__name__)
self.assertEquals('whatever', enum_class.__module__)
self.assertEquals(enum_descriptor,
descriptor.describe_enum(enum_class))
def testDefineEnum(self):
"""Test defining an enum."""
red = descriptor.EnumValueDescriptor()
green = descriptor.EnumValueDescriptor()
blue = descriptor.EnumValueDescriptor()
red.name = 'RED'
red.number = 1
green.name = 'GREEN'
green.number = 2
blue.name = 'BLUE'
blue.number = 3
enum_descriptor = descriptor.EnumDescriptor()
enum_descriptor.name = 'Colors'
enum_descriptor.values = [red, green, blue]
enum_class = definition.define_enum(enum_descriptor, 'whatever')
self.assertEquals('Colors', enum_class.__name__)
self.assertEquals('whatever', enum_class.__module__)
self.assertEquals(enum_descriptor, descriptor.describe_enum(enum_class))
def testDefineEnum(self):
"""Test defining an enum."""
red = descriptor.EnumValueDescriptor()
green = descriptor.EnumValueDescriptor()
blue = descriptor.EnumValueDescriptor()
red.name = 'RED'
red.number = 1
green.name = 'GREEN'
green.number = 2
blue.name = 'BLUE'
blue.number = 3
enum_descriptor = descriptor.EnumDescriptor()
enum_descriptor.name = 'Colors'
enum_descriptor.values = [red, green, blue]
enum_class = definition.define_enum(enum_descriptor, 'whatever')
self.assertEquals('Colors', enum_class.__name__)
self.assertEquals('whatever', enum_class.__module__)
self.assertEquals(enum_descriptor,
descriptor.describe_enum(enum_class))
