102 lines · python
1#!/usr/bin/python2# SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause)3# Basic sanity check of perf JSON output as specified in the man page.4 5import argparse6import sys7import json8 9ap = argparse.ArgumentParser()10ap.add_argument('--no-args', action='store_true')11ap.add_argument('--interval', action='store_true')12ap.add_argument('--system-wide-no-aggr', action='store_true')13ap.add_argument('--system-wide', action='store_true')14ap.add_argument('--event', action='store_true')15ap.add_argument('--per-core', action='store_true')16ap.add_argument('--per-thread', action='store_true')17ap.add_argument('--per-cache', action='store_true')18ap.add_argument('--per-cluster', action='store_true')19ap.add_argument('--per-die', action='store_true')20ap.add_argument('--per-node', action='store_true')21ap.add_argument('--per-socket', action='store_true')22ap.add_argument('--file', type=argparse.FileType('r'), default=sys.stdin)23args = ap.parse_args()24 25Lines = args.file.readlines()26 27def isfloat(num):28 try:29 float(num)30 return True31 except ValueError:32 return False33 34 35def isint(num):36 try:37 int(num)38 return True39 except ValueError:40 return False41 42def is_counter_value(num):43 return isfloat(num) or num == '<not counted>' or num == '<not supported>'44 45def check_json_output(expected_items):46 checks = {47 'aggregate-number': lambda x: isfloat(x),48 'core': lambda x: True,49 'counter-value': lambda x: is_counter_value(x),50 'cgroup': lambda x: True,51 'cpu': lambda x: isint(x),52 'cache': lambda x: True,53 'cluster': lambda x: True,54 'die': lambda x: True,55 'event': lambda x: True,56 'event-runtime': lambda x: isfloat(x),57 'interval': lambda x: isfloat(x),58 'metric-unit': lambda x: True,59 'metric-value': lambda x: isfloat(x),60 'metricgroup': lambda x: True,61 'node': lambda x: True,62 'pcnt-running': lambda x: isfloat(x),63 'socket': lambda x: True,64 'thread': lambda x: True,65 'unit': lambda x: True,66 }67 input = '[\n' + ','.join(Lines) + '\n]'68 for item in json.loads(input):69 if expected_items != -1:70 count = len(item)71 if count != expected_items and count >= 1 and count <= 6 and 'metric-value' in item:72 # Events that generate >1 metric may have isolated metric73 # values and possibly other prefixes like interval, core,74 # aggregate-number, or event-runtime/pcnt-running from multiplexing.75 pass76 elif count != expected_items and count >= 1 and count <= 5 and 'metricgroup' in item:77 pass78 elif count != expected_items:79 raise RuntimeError(f'wrong number of fields. counted {count} expected {expected_items}'80 f' in \'{item}\'')81 for key, value in item.items():82 if key not in checks:83 raise RuntimeError(f'Unexpected key: key={key} value={value}')84 if not checks[key](value):85 raise RuntimeError(f'Check failed for: key={key} value={value}')86 87 88try:89 if args.no_args or args.system_wide or args.event:90 expected_items = 791 elif args.interval or args.per_thread or args.system_wide_no_aggr:92 expected_items = 893 elif args.per_core or args.per_socket or args.per_node or args.per_die or args.per_cluster or args.per_cache:94 expected_items = 995 else:96 # If no option is specified, don't check the number of items.97 expected_items = -198 check_json_output(expected_items)99except:100 print('Test failed for input:\n' + '\n'.join(Lines))101 raise102