Index: tools/json_schema_compiler/cpp_type_generator_test.py |
diff --git a/tools/json_schema_compiler/cpp_type_generator_test.py b/tools/json_schema_compiler/cpp_type_generator_test.py |
index 81ee04de136b0658b1cb618a7d4b130a1a0c96fe..ac162620a90c82a2e59b003ba5830dccf10d8dbd 100644 |
--- a/tools/json_schema_compiler/cpp_type_generator_test.py |
+++ b/tools/json_schema_compiler/cpp_type_generator_test.py |
@@ -62,11 +62,11 @@ class CppTypeGeneratorTest(unittest.TestCase): |
manager = CppTypeGenerator('', self.tabs, 'tabs_api') |
prop = self.tabs.functions['move'].params[0] |
self.assertEquals('TAB_IDS_ARRAY', |
- manager.GetEnumValue(prop, model.PropertyType.ARRAY)) |
+ manager.GetEnumValue(prop, model.PropertyType.ARRAY.name)) |
self.assertEquals('TAB_IDS_INTEGER', |
- manager.GetEnumValue(prop, model.PropertyType.INTEGER)) |
+ manager.GetEnumValue(prop, model.PropertyType.INTEGER.name)) |
self.assertEquals('TabIdsType', |
- manager.GetEnumType(prop)) |
+ manager.GetChoicesEnumType(prop)) |
def testGetTypeSimple(self): |
manager = CppTypeGenerator('', self.tabs, 'tabs_api') |