tests: Register test cases automatically from test_p2p_*.py

This removes need for editing run-p2p-tests.py whenever new test files
are added.

Signed-hostap: Jouni Malinen <j@w1.fi>
This commit is contained in:
Jouni Malinen 2013-03-09 23:18:20 +02:00
parent 823d572f99
commit 2a4291ffdc

View file

@ -7,6 +7,7 @@
# See README for more details. # See README for more details.
import os import os
import re
import sys import sys
import time import time
@ -14,9 +15,6 @@ import logging
from wpasupplicant import WpaSupplicant from wpasupplicant import WpaSupplicant
import test_p2p_grpform
import test_p2p_autogo
def main(): def main():
idx = 1 idx = 1
if len(sys.argv) > 1 and sys.argv[1] == '-d': if len(sys.argv) > 1 and sys.argv[1] == '-d':
@ -46,8 +44,12 @@ def main():
print "DEV: " + d.ifname + ": " + d.p2p_dev_addr() print "DEV: " + d.ifname + ": " + d.p2p_dev_addr()
tests = [] tests = []
test_p2p_grpform.add_tests(tests) for t in os.listdir("."):
test_p2p_autogo.add_tests(tests) m = re.match(r'(test_p2p_.*)\.py$', t)
if m:
print "Import test cases from " + t
mod = __import__(m.group(1))
mod.add_tests(tests)
passed = [] passed = []
failed = [] failed = []