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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
#!/usr/bin/env python3
#
# Wireshark tests
# By Gerald Combs <gerald@wireshark.org>
#
# Ported from a set of Bash scripts which were copyright 2005 Ulf Lamping
#
# SPDX-License-Identifier: GPL-2.0-or-later
#
'''Main test script'''
# To do:
# - Avoid printing Python tracebacks when we assert? It looks like we'd need
# to override unittest.TextTestResult.addFailure().
import argparse
import codecs
import os.path
import suite_external
import sys
import unittest
import fixtures
# Required to make fixtures available to tests!
import fixtures_ws
_all_test_groups = None
@fixtures.fixture(scope='session')
def all_test_groups():
return _all_test_groups
def find_test_ids(suite, all_ids):
if hasattr(suite, '__iter__'):
for s in suite:
find_test_ids(s, all_ids)
else:
all_ids.append(suite.id())
def main():
parser = argparse.ArgumentParser(description='Wireshark unit tests')
cap_group = parser.add_mutually_exclusive_group()
cap_group.add_argument('-E', '--disable-capture', action='store_true', help='Disable capture tests')
release_group = parser.add_mutually_exclusive_group()
release_group.add_argument('--enable-release', action='store_true', help='Enable release tests')
parser.add_argument('-p', '--program-path', default=os.path.curdir, help='Path to Wireshark executables.')
parser.add_argument('-x', '--add-external-tests', action='append', help='Path to an external test definition (.json) file.')
parser.add_argument('--skip-missing-programs',
help='Skip tests that lack programs from this list instead of failing'
' them. Use "all" to ignore all missing programs.')
list_group = parser.add_mutually_exclusive_group()
list_group.add_argument('-l', '--list', action='store_true', help='List tests. One of "all" or a full or partial test name.')
list_group.add_argument('--list-suites', action='store_true', help='List all suites.')
list_group.add_argument('--list-groups', action='store_true', help='List all suites and groups.')
list_group.add_argument('--list-cases', action='store_true', help='List all suites, groups, and cases.')
parser.add_argument('-v', '--verbose', action='store_const', const=2, default=1, help='Verbose tests.')
parser.add_argument('tests_to_run', nargs='*', metavar='test', default=['all'], help='Tests to run. One of "all" or a full or partial test name. Default is "all".')
args = parser.parse_args()
# XXX This should be a fixture.
suite_external.add_external_configs(args.add_external_tests)
all_tests = unittest.defaultTestLoader.discover(os.path.dirname(__file__), pattern='suite_*')
all_ids = []
find_test_ids(all_tests, all_ids)
run_ids = []
for tid in all_ids:
for ttr in args.tests_to_run:
ttrl = ttr.lower()
if ttrl == 'all':
run_ids = all_ids
break
if ttrl in tid.lower():
run_ids.append(tid)
if not run_ids:
print('No tests found. You asked for:\n ' + '\n '.join(args.tests_to_run))
parser.print_usage()
sys.exit(1)
if args.list:
print('\n'.join(run_ids))
sys.exit(0)
all_suites = set()
for aid in all_ids:
aparts = aid.split('.')
all_suites |= {aparts[0]}
all_suites = sorted(all_suites)
all_groups = set()
for aid in all_ids:
aparts = aid.split('.')
if aparts[1].startswith('group_'):
all_groups |= {'.'.join(aparts[:2])}
else:
all_groups |= {aparts[0]}
all_groups = sorted(all_groups)
global _all_test_groups
_all_test_groups = all_groups
if args.list_suites:
print('\n'.join(all_suites))
sys.exit(0)
if args.list_groups:
print('\n'.join(all_groups))
sys.exit(0)
if args.list_cases:
cases = set()
for rid in run_ids:
rparts = rid.split('.')
cases |= {'.'.join(rparts[:2])}
print('\n'.join(list(cases)))
sys.exit(0)
if codecs.lookup(sys.stdout.encoding).name != 'utf-8':
import locale
sys.stderr.write('Warning: Output encoding is {0} and not utf-8.\n'.format(sys.stdout.encoding))
sys.stdout = codecs.getwriter(locale.getpreferredencoding())(sys.stdout.buffer, 'backslashreplace')
sys.stderr = codecs.getwriter(locale.getpreferredencoding())(sys.stderr.buffer, 'backslashreplace')
run_suite = unittest.defaultTestLoader.loadTestsFromNames(run_ids)
runner = unittest.TextTestRunner(verbosity=args.verbose)
# for unittest compatibility (not needed with pytest)
fixtures_ws.fixtures.create_session(args)
try:
test_result = runner.run(run_suite)
finally:
# for unittest compatibility (not needed with pytest)
fixtures_ws.fixtures.destroy_session()
if test_result.errors:
sys.exit(2)
if test_result.failures:
sys.exit(1)
if __name__ == '__main__':
main()
|
