def setUp(self): self.returnObject = None self.toCheck = {} def execute(command, timeout=60, env=None, cwd=None, shell=False): import inspect frame = inspect.currentframe() fedInVars = inspect.getargvalues(frame).locals del frame for key, value in self.toCheck.iteritems(): if key in fedInVars: self.assertEqual(fedInVars[key], value) return self.returnObject def add_process(this, command, command_args=(), command_kwargs={}, timeout=60, env=None, cwd=None, shell=False, priority=5, callback_func=None, callback_args=(), callback_kwargs={}): import inspect frame = inspect.currentframe() fedInVars = inspect.getargvalues(frame).locals del frame for key, value in self.toCheck.iteritems(): if key in fedInVars: self.assertEqual(fedInVars[key], value) return self.returnObject self.db = DiracBase() from Ganga.Core.GangaThread.WorkerThreads.WorkerThreadPool import WorkerThreadPool setattr(sys.modules[self.db.__module__], 'execute', execute) setattr(WorkerThreadPool, "add_process", add_process) self.script = """
def db(): """Provides a DiracBase object per test function""" return DiracBase()
def db(): """Provides a DiracBase object per test function""" load_config_files() from GangaDirac.Lib.Backends.DiracBase import DiracBase yield DiracBase() clear_config()