def test_colorize(self):
"""
Passing ``colorize=True`` will colorize the output.
"""
fd = BytesIO()
tree = Tree()
tree.merge_tasks([action_task, action_task_end])
render_task_nodes(
write=fd.write,
nodes=tree.nodes(),
field_limit=0,
colorize=True)
C = COLORS(colored)
self.assertThat(
fd.getvalue(),
ExactlyEquals(
u'\n'.join([
C.root(u'f3a32bb3-ea6b-457c-aa99-08a3d0491ab4'),
u'\u2514\u2500\u2500 {}'.format(
C.success(u'app:action@1/started')),
u' \u251c\u2500\u2500 {}: {}'.format(
C.prop(u'timestamp'), u'1425356800'),
u' \u2514\u2500\u2500 {}'.format(
C.success(u'app:action@2/succeeded')),
u' \u2514\u2500\u2500 {}: {}'.format(
C.prop('timestamp'), u'1425356802'),
u'\n',
]).encode('utf-8')))
def test_node_success(self):
"""
Task nodes indicating success are colored.
"""
tree = Tree()
tree.merge_tasks([action_task])
node = tree.nodes()[0][1].copy()
node.success = True
C = COLORS(colored)
get_name = get_name_factory(C)
self.assertThat(
get_name(node),
ExactlyEquals(C.success(node.name)))