selftest/testlist.py

   1 # testlist.py -- Test list
   2 # Copyright (C) 2012 Jelmer Vernooij <jelmer@samba.org>
   3 #
   4 # This program is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU General Public License
   6 # as published by the Free Software Foundation; version 3
   7 # of the License or (at your option) any later version of
   8 # the License.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program; if not, write to the Free Software
  17 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
  18 # MA  02110-1301, USA.
  19
  20 """Selftest test list management."""
  21
  22 __all__ = ['find_in_list', 'read_test_regexes']
  23
  24 import re
  25
  26 def find_in_list(list, fullname):
  27     """Find test in list.
  28
  29     :param list: List with 2-tuples with regex and reason
  30     """
  31     for (regex, reason) in list:
  32         if re.match(regex, fullname):
  33             if reason is not None:
  34                 return reason
  35             else:
  36                 return ""
  37     return None
  38
  39
  40 def read_test_regexes(f):
  41     """Read tuples with regular expression and optional string from a file.
  42
  43     :param f: File-like object to read from
  44     :return: Iterator over tuples with regular expression and test name
  45     """
  46     for l in f.readlines():
  47         l = l.strip()
  48         if l[0] == "#":
  49             continue
  50         try:
  51             (test, reason) = l.split("#", 1)
  52         except ValueError:
  53             yield l, None
  54         else:
  55             yield test.strip(), reason.strip()