def do_test_load(self, config='', other_files={}, stdout_re=None, stderr_re=None): configFile = os.path.join('configdir', 'master.cfg') with open(configFile, "w") as f: f.write(config) for filename, contents in other_files.iteritems(): if isinstance(filename, type(())): fn = os.path.join('configdir', *filename) dn = os.path.dirname(fn) if not os.path.isdir(dn): os.makedirs(dn) else: fn = os.path.join('configdir', filename) with open(fn, "w") as f: f.write(contents) old_stdout, old_stderr = sys.stdout, sys.stderr stdout = sys.stdout = cStringIO.StringIO() stderr = sys.stderr = cStringIO.StringIO() try: checkconfig._loadConfig( basedir='configdir', configFile="master.cfg", quiet=False) finally: sys.stdout, sys.stderr = old_stdout, old_stderr if stdout_re: stdout = stdout.getvalue() self.failUnless(stdout_re.search(stdout), stdout) if stderr_re: stderr = stderr.getvalue() self.failUnless(stderr_re.search(stderr), stderr)
def do_test_load(self, config='', other_files={}, stdout_re=None, stderr_re=None): configFile = os.path.join('configdir', 'master.cfg') with open(configFile, "w") as f: f.write(config) for filename, contents in iteritems(other_files): if isinstance(filename, type(())): fn = os.path.join('configdir', *filename) dn = os.path.dirname(fn) if not os.path.isdir(dn): os.makedirs(dn) else: fn = os.path.join('configdir', filename) with open(fn, "w") as f: f.write(contents) old_stdout, old_stderr = sys.stdout, sys.stderr stdout = sys.stdout = NativeStringIO() stderr = sys.stderr = NativeStringIO() try: checkconfig._loadConfig(basedir='configdir', configFile="master.cfg", quiet=False) finally: sys.stdout, sys.stderr = old_stdout, old_stderr if stdout_re: stdout = stdout.getvalue() self.assertTrue(stdout_re.search(stdout), stdout) if stderr_re: stderr = stderr.getvalue() self.assertTrue(stderr_re.search(stderr), stderr)