def test_member_name_override(self): self.output.write('class Approval(Object):\n') writer = ClassWriter('Approval', 'TEST', self.output) writer.write_simple_members('Field', {'fields': { 'URL': {}, 'url': {} }}, 'fields') self.check_output("""\ class Approval(Object): url = Field('URL') url_ = Field('url') """) # no error log: self.log.check()
def test_method_override(self): writer = ClassWriter('Update', '???', self.output) writer.write_header() writer.write_members(self.writer.format_action, {'actions': { "update_obj": {} }}, 'actions') writer.write_footer() self.check_output(''' class Update(Object): code = '???' @property def update_obj(self): """ The object referenced by this update. """ return self.session.api.from_data( self.session, dict( ID=self.update_obj_id, objCode=self.update_obj_code )) api.register(Update) ''')
def setUp(self): self.output = StringIO() self.writer = ClassWriter('FooBar', 'FOOB', self.output) self.log = LogCapture() self.addCleanup(self.log.uninstall)