def test_pprint_routes(capfd):
result = []
r = jina_pb2.RouteProto()
r.status.code = jina_pb2.StatusProto.ERROR
r.status.exception.stacks.extend(['r1\nline1', 'r2\nline2'])
result.append(r)
r = jina_pb2.RouteProto()
r.status.code = jina_pb2.StatusProto.ERROR_CHAINED
r.status.exception.stacks.extend(['line1', 'line2'])
result.append(r)
r = jina_pb2.RouteProto()
r.status.code = jina_pb2.StatusProto.SUCCESS
result.append(r)
rr = Request()
rr.routes.extend(result)
pprint_routes(rr)
out, err = capfd.readouterr()
assert '⚪' in out
assert '🟢' in out
assert 'Pod' in out
assert 'Time' in out
assert 'Exception' in out
assert 'r1' in out
assert 'line1r2' in out
assert 'line2' in out
assert 'line1line2' in out
def test_pprint_routes(capfd):
result = []
r = jina_pb2.RouteProto()
r.status.code = jina_pb2.StatusProto.ERROR
r.status.exception.stacks.extend(['r1\nline1', 'r2\nline2'])
result.append(r)
r = jina_pb2.RouteProto()
r.status.code = jina_pb2.StatusProto.ERROR_CHAINED
r.status.exception.stacks.extend(['line1', 'line2'])
result.append(r)
r = jina_pb2.RouteProto()
r.status.code = jina_pb2.StatusProto.SUCCESS
result.append(r)
rr = Request()
rr.routes.extend(result)
pprint_routes(rr)
out, err = capfd.readouterr()
assert out == '''+-----+------+------------+