wehub-resource-sync
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
|
"""Print the slowest tests from Go's JSON test stream."""
|
|
|
|
import argparse
|
|
import json
|
|
import sys
|
|
|
|
|
|
def events_from(path):
|
|
stream = sys.stdin if path == "-" else open(path, "r", encoding="utf-8")
|
|
try:
|
|
for line in stream:
|
|
try:
|
|
yield json.loads(line)
|
|
except json.JSONDecodeError:
|
|
continue
|
|
finally:
|
|
if stream is not sys.stdin:
|
|
stream.close()
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(
|
|
description="Rank test cases from `go test -json` output by elapsed seconds."
|
|
)
|
|
parser.add_argument("json_file", nargs="?", default="-", help="JSON stream file, or stdin")
|
|
parser.add_argument("-n", "--limit", type=int, default=20, help="number of rows to print")
|
|
parser.add_argument("--min", type=float, default=0.0, help="minimum elapsed seconds")
|
|
args = parser.parse_args()
|
|
|
|
rows = []
|
|
for event in events_from(args.json_file):
|
|
if event.get("Action") not in {"pass", "fail", "skip"}:
|
|
continue
|
|
test = event.get("Test")
|
|
elapsed = event.get("Elapsed")
|
|
if not test or elapsed is None or elapsed < args.min:
|
|
continue
|
|
package = event.get("Package", "")
|
|
name = f"{package}.{test}" if package else test
|
|
rows.append((float(elapsed), event["Action"], name))
|
|
|
|
rows.sort(key=lambda row: row[0], reverse=True)
|
|
if not rows:
|
|
print("No test timing rows found.")
|
|
return
|
|
|
|
print(f"{'SECONDS':>8} {'STATE':<5} TEST")
|
|
for elapsed, action, name in rows[: args.limit]:
|
|
print(f"{elapsed:8.3f} {action:<5} {name}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|