From 6c345b0ea1a3b2e2ea9a8bd671f11a02efa2aad3 Mon Sep 17 00:00:00 2001
From: Auke Klazema <klazema@astron.nl>
Date: Wed, 19 Jun 2019 12:05:29 +0000
Subject: [PATCH] SW-720: Add missing handler to ResourceAssigner (merge from
 trunk)

---
 SAS/ResourceAssignment/ResourceAssigner/lib/raservice.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/SAS/ResourceAssignment/ResourceAssigner/lib/raservice.py b/SAS/ResourceAssignment/ResourceAssigner/lib/raservice.py
index d802a9ca254..66709f74dff 100755
--- a/SAS/ResourceAssignment/ResourceAssigner/lib/raservice.py
+++ b/SAS/ResourceAssignment/ResourceAssigner/lib/raservice.py
@@ -57,7 +57,7 @@ class SpecifiedTaskEventMessageHandler(RATaskSpecifiedEventMessageHandler):
         except Exception as e:
             logger.error(str(e))
 
-__all__ = ["SpecifiedTaskListener"]
+__all__ = ["SpecifiedTaskEventMessageHandler"]
 
 
 def main():
@@ -91,8 +91,10 @@ def main():
 
     with ResourceAssigner(exchange=options.exchange,
                           broker=options.broker,
-                          radb_dbcreds=radb_dbcreds):
-        with RATaskSpecifiedBusListener(exchange=options.exchange,
+                          radb_dbcreds=radb_dbcreds) as assigner:
+        with RATaskSpecifiedBusListener(handler_type=SpecifiedTaskEventMessageHandler,
+                                        handler_kwargs={"assigner": assigner},
+                                        exchange=options.exchange,
                                         broker=options.broker):
             with ScheduleChecker(exchange=options.exchange,
                                  broker=options.broker):
-- 
GitLab