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 == '''+-----+------+------------+