]>
Commit | Line | Data |
---|---|---|
76b903ee | 1 | """ |
b2441318 | 2 | # SPDX-License-Identifier: GPL-2.0 |
76b903ee LB |
3 | tdc_helper.py - tdc helper functions |
4 | ||
5 | Copyright (C) 2017 Lucas Bates <lucasb@mojatatu.com> | |
6 | """ | |
7 | ||
8 | def get_categorized_testlist(alltests, ucat): | |
9 | """ Sort the master test list into categories. """ | |
10 | testcases = dict() | |
11 | ||
12 | for category in ucat: | |
13 | testcases[category] = list(filter(lambda x: category in x['category'], alltests)) | |
14 | ||
15 | return(testcases) | |
16 | ||
17 | ||
18 | def get_unique_item(lst): | |
19 | """ For a list, return a set of the unique items in the list. """ | |
20 | return list(set(lst)) | |
21 | ||
22 | ||
23 | def get_test_categories(alltests): | |
24 | """ Discover all unique test categories present in the test case file. """ | |
25 | ucat = [] | |
26 | for t in alltests: | |
27 | ucat.extend(get_unique_item(t['category'])) | |
28 | ucat = get_unique_item(ucat) | |
29 | return ucat | |
30 | ||
31 | def list_test_cases(testlist): | |
32 | """ Print IDs and names of all test cases. """ | |
33 | for curcase in testlist: | |
34 | print(curcase['id'] + ': (' + ', '.join(curcase['category']) + ") " + curcase['name']) | |
35 | ||
36 | ||
37 | def list_categories(testlist): | |
38 | """ Show all categories that are present in a test case file. """ | |
39 | categories = set(map(lambda x: x['category'], testlist)) | |
40 | print("Available categories:") | |
41 | print(", ".join(str(s) for s in categories)) | |
42 | print("") | |
43 | ||
44 | ||
45 | def print_list(cmdlist): | |
46 | """ Print a list of strings prepended with a tab. """ | |
47 | for l in cmdlist: | |
48 | if (type(l) == list): | |
49 | print("\t" + str(l[0])) | |
50 | else: | |
51 | print("\t" + str(l)) | |
52 | ||
53 | ||
54 | def print_sll(items): | |
55 | print("\n".join(str(s) for s in items)) | |
56 | ||
57 | ||
58 | def print_test_case(tcase): | |
59 | """ Pretty-printing of a given test case. """ | |
60 | for k in tcase.keys(): | |
61 | if (type(tcase[k]) == list): | |
62 | print(k + ":") | |
63 | print_list(tcase[k]) | |
64 | else: | |
65 | print(k + ": " + tcase[k]) | |
66 | ||
67 | ||
68 | def show_test_case_by_id(testlist, caseID): | |
69 | """ Find the specified test case to pretty-print. """ | |
70 | if not any(d.get('id', None) == caseID for d in testlist): | |
71 | print("That ID does not exist.") | |
72 | exit(1) | |
73 | else: | |
74 | print_test_case(next((d for d in testlist if d['id'] == caseID))) | |
75 | ||
76 |