selftest/run.py

   1 #!/usr/bin/python -u
   2 # Bootstrap Samba and run a number of tests against it.
   3 # Copyright (C) 2012 Jelmer Vernooij <jelmer@samba.org>
   4
   5 # This program is free software; you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation; either version 3 of the License, or
   8 # (at your option) any later version.
   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, see <http://www.gnu.org/licenses/>.
  17
  18 import os
  19 import tempfile
  20 import warnings
  21
  22 # expand strings from %ENV
  23 def expand_environment_strings(s, vars):
  24     # we use a reverse sort so we do the longer ones first
  25     for k in sorted(vars.keys(), reverse=True):
  26         v = vars[k]
  27         s = s.replace("$%s" % k, v)
  28     return s
  29
  30
  31 def expand_command_list(cmd):
  32     if not "$LISTOPT" in cmd:
  33         return None
  34     return cmd.replace("$LISTOPT", "--list")
  35
  36
  37 def expand_command_run(cmd, supports_loadfile, supports_idlist, subtests=None):
  38     """Expand a test command.
  39
  40     :param cmd: Command to expand
  41     :param supports_loadfile: Whether command supports loadfile
  42     :param supports_idlist: Whether the command supports running specific
  43         subtests
  44     :param subtests: List of subtests to run - None for all subtests
  45     :return: Tuple with command to run and temporary file to remove after
  46         running (or None)
  47     """
  48     # Generate a file with the individual tests to run, if the
  49     # test runner for this test suite supports it.
  50     if subtests is None:
  51         return (cmd.replace("$LOADLIST", ""), None)
  52     if supports_loadfile:
  53         (fd, listid_file) = tempfile.mkstemp()
  54         f = os.fdopen(fd, 'w')
  55         try:
  56             for test in subtests:
  57                 f.write(test+"\n")
  58         finally:
  59             f.close()
  60         return (
  61             cmd.replace("$LOADLIST", "--load-list=%s" % listid_file),
  62             listid_file)
  63     elif supports_idlist:
  64         cmd += " " + " ".join(subtests)
  65         return (cmd, None)
  66     else:
  67         warnings.warn(
  68             "Running subtests requested, but command does not support "
  69             "this.")
  70         return (cmd, None)
  71
  72
  73 def exported_envvars_str(vars, names):
  74     out = ""
  75     for n in names:
  76         if not n in vars:
  77             continue
  78         out += "%s=%s\n" % (n, vars[n])
  79     return out