def test_fixup_enums(self): enum = TestEnum() struct1 = TestStruct() self.assertTrue(hasattr(enum,"_VALUES_TO_NAMES")) struct1.myenum = 0 thrift_util.fixup_enums(struct1,{"myenum":TestEnum}) self.assertTrue(hasattr(struct1,"myenumAsString")) self.assertEquals(struct1.myenumAsString,'ENUM_ONE')
def test_fixup_enums(self): enum = TestEnum() struct1 = TestStruct() self.assertTrue(hasattr(enum, "_VALUES_TO_NAMES")) struct1.myenum = 0 thrift_util.fixup_enums(struct1, {"myenum": TestEnum}) self.assertTrue(hasattr(struct1, "myenumAsString")) self.assertEquals(struct1.myenumAsString, 'ENUM_ONE')