summaryrefslogtreecommitdiffstats
path: root/selftest/format-subunit-json
blob: d9d912cd29428c7a6766e2d2c0052ef5afa4e3d9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#!/usr/bin/env python3
# Copyright (C) 2008-2010 Jelmer Vernooij <jelmer@samba.org>
# Copyright (C) 2016 Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
# Published under the GNU GPL, v3 or later
import optparse
import os
import signal
import sys
import json

sys.path.insert(0, "bin/python")


def json_formatter(src_f, dest_f):
    """We're not even pretending to be a TestResult subclass; just read
    from stdin and look for elapsed-time tags."""
    results = {}

    for line in src_f:
        line = line.strip()
        if line[:14] == 'elapsed-time: ':
            name, time = line[14:].rsplit(':', 1)
            results[name] = float(time)

    json.dump(results, dest_f,
              sort_keys=True, indent=2, separators=(',', ': '))


def main():
    parser = optparse.OptionParser("format-subunit-json [options]")
    parser.add_option("--verbose", action="store_true",
                      help="ignored, for compatibility")
    parser.add_option("--immediate", action="store_true",
                      help="ignored, for compatibility")
    parser.add_option("--prefix", type="string", default=".",
                      help="Prefix to write summary.json to")
    opts, args = parser.parse_args()

    fn = os.path.join(opts.prefix, "summary.json")
    f = open(fn, 'w')
    json_formatter(sys.stdin, f)
    f.close()
    print()
    print("A JSON file summarising these tests performance found in:")
    print(" ", fn)


def handle_sigint(sig, stack):
    sys.exit(0)

signal.signal(signal.SIGINT, handle_sigint)
main()