def register_classes(self): for cls in all_entity_classes(): self.register_class(cls)
def test_get_all_entity_classes(self): classes = all_entity_classes() ok_(Message in classes) ok_(PrivateMessage in classes)
def object(self): for cls in all_entity_classes(): if cls.__name__ == self.object_class: return cls.query.get(self.object_id) raise Exception("Unknown class: %s" % self.object_class)
def test_get_all_entity_classes(self): classes = all_entity_classes() ok_(Lead in classes) ok_(Account in classes) ok_(Contact in classes) ok_(Opportunity in classes)
def register_classes(self): for cls in all_entity_classes(): if not cls in self.indexed_classes: self.register_class(cls)