def check_for_comments(expected):
args_template = self.args['template']
comments.cb_story(self.args)
self.assertEquals(expected, self.args['template'])
comments.cb_story_end(self.args)
self.assertEquals(expected, self.args['template'])
self.args['template'] = args_template
def check_for_comments(expected):
args_template = self.args['template']
comments.cb_story(self.args)
self.assertEquals(expected, self.args['template'])
comments.cb_story_end(self.args)
self.assertEquals(expected, self.args['template'])
self.args['template'] = args_template
def check_comment_output(self, expected, delete_datadir=True, **kwargs):
"""Posts a comment and checks its rendered output.
Note that this deletes the datadir before it posts the
comment!
"""
self.data['display_comment_default'] = True
self.comment(**kwargs)
comments.cb_story(self.args)
self.args['template'] = ''
comments.cb_story_end(self.args)
self.assertEquals(expected, self.args['template'])
if delete_datadir:
self.delete_datadir()
def check_comment_output(self, expected, delete_datadir=True, **kwargs):
"""Posts a comment and checks its rendered output.
Note that this deletes the datadir before it posts the
comment!
"""
self.data['display_comment_default'] = True
self.comment(**kwargs)
comments.cb_story(self.args)
self.args['template'] = ''
comments.cb_story_end(self.args)
self.assertEquals(expected, self.args['template'])
if delete_datadir:
self.delete_datadir()
def test_cb_story_end_renders_comments(self):
self.comment()
# check that cb_story_end() renders comments.
self.data['display_comment_default'] = True
# no comments. check both cb_story_end's return value and
# args['template'].
self.args['template'] = 'foo'
self.assertEquals('foo', comments.cb_story_end(self.args))
self.assertEquals('foo', self.args['template'])
# one comment
self.renderer.flavour['comment'] = '$cmt_time '
expected = '%s ' % self.timestamp
self.check_comment_output(expected, delete_datadir=False)
# two comments
self.frozen_time.timestamp += 1
expected += '%s ' % self.frozen_time.timestamp
self.check_comment_output(expected)
def test_cb_story_end_renders_comments(self):
self.comment()
# check that cb_story_end() renders comments.
self.data['display_comment_default'] = True
# no comments. check both cb_story_end's return value and
# args['template'].
self.args['template'] = 'foo'
self.assertEquals('foo', comments.cb_story_end(self.args))
self.assertEquals('foo', self.args['template'])
# one comment
self.renderer.flavour['comment'] = '$cmt_time '
expected = '%s ' % self.timestamp
self.check_comment_output(expected, delete_datadir=False)
# two comments
self.frozen_time.timestamp += 1
expected += '%s ' % self.frozen_time.timestamp
self.check_comment_output(expected)