def test_node(self):
f = bootsteps.StepFormatter()
f.draw_node = Mock()
step = Mock()
step.last = False
f.node(step, x=3)
f.draw_node.assert_called_with(step, f.node_scheme, {'x': 3})
step.last = True
f.node(step, x=3)
f.draw_node.assert_called_with(step, f.blueprint_scheme, {'x': 3})
def test_get_prefix(self):
f = bootsteps.StepFormatter()
s = Mock()
s.last = True
assert f._get_prefix(s) == f.blueprint_prefix
s2 = Mock()
s2.last = False
s2.conditional = True
assert f._get_prefix(s2) == f.conditional_prefix
s3 = Mock()
s3.last = s3.conditional = False
assert f._get_prefix(s3) == ''
def test_edge(self):
f = bootsteps.StepFormatter()
f.draw_edge = Mock()
a, b = Mock(), Mock()
a.last = True
f.edge(a, b, x=6)
f.draw_edge.assert_called_with(a, b, f.edge_scheme, {
'x': 6, 'arrowhead': 'none', 'color': 'darkseagreen3',
})
a.last = False
f.edge(a, b, x=6)
f.draw_edge.assert_called_with(a, b, f.edge_scheme, {
'x': 6,
})
