From 168b7958219b4be34a1922fa27c92a50a1f9d8d2 Mon Sep 17 00:00:00 2001
From: Cristiano Urban <cristiano.urban@inaf.it>
Date: Mon, 20 Sep 2021 13:44:31 +0200
Subject: [PATCH] Allow to change the time to wait directly.

Signed-off-by: Cristiano Urban <cristiano.urban@inaf.it>
---
 transfer_service/task_executor.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/transfer_service/task_executor.py b/transfer_service/task_executor.py
index 50afd05..f5cdb18 100644
--- a/transfer_service/task_executor.py
+++ b/transfer_service/task_executor.py
@@ -29,8 +29,11 @@ class TaskExecutor(Process):
         self.destQueue = None
         super(TaskExecutor, self).__init__()
 
-    def wait(self):
-        time.sleep(self.execWaitTime)
+    def wait(self, timeout = None):
+        if timeout is None:
+            time.sleep(self.execWaitTime)
+        else:
+            time.sleep(timeout)
 
     def setSourceQueueName(self, srcQueueName):
         self.srcQueue = JobQueue(srcQueueName)
-- 
GitLab