예제 #1
0
    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 = """
예제 #2
0
def db():
    """Provides a DiracBase object per test function"""
    return DiracBase()
예제 #3
0
def db():
    """Provides a DiracBase object per test function"""
    load_config_files()
    from GangaDirac.Lib.Backends.DiracBase import DiracBase
    yield DiracBase()
    clear_config()