Dcaegen2 Jenkins jobs are failing

[integration/csit.git] / tests / dcaegen2 / testcases / resources / robot_library / DmaapLibrary.py

diff --git a/tests/dcaegen2/testcases/resources/robot_library/DmaapLibrary.py b/tests/dcaegen2/testcases/resources/robot_library/DmaapLibrary.py
index c9a0ff7..07985ea 100644 (file)
--- a/tests/dcaegen2/testcases/resources/robot_library/DmaapLibrary.py
+++ b/tests/dcaegen2/testcases/resources/robot_library/DmaapLibrary.py
@@ -1,10 +1,10 @@
-from Queue import Queue
+from queue import Queue
 
 import robot.api.logger as logger
 import threading
 import time
 
-import DcaeVariables
+from robot_library import DcaeVariables
 from robot_library.dmaap_simulator import DMaaPServer
 from robot_library.dmaap_simulator.DMaaPQueue import DMaaPQueue
 
@@ -49,6 +49,7 @@ class DmaapLibrary(object):
 
     @staticmethod
     def cleanup_ves_events():
+        DmaapLibrary.dmaap_server.reset_dmaap_succesfull_code()
         if DmaapLibrary.server_thread is not None:
             DmaapLibrary.dmaap_queue.clean_up_event()
             logger.console("DMaaP event queue is cleaned up")
@@ -77,3 +78,7 @@ class DmaapLibrary(object):
                 return 'true'
             evt_str = DmaapLibrary.dmaap_queue.deque_event()
         return 'false'
+
+    @staticmethod
+    def set_successfull_dmaap_code(code):
+        DmaapLibrary.dmaap_server.set_dmaap_successfull_code(int(code))
\ No newline at end of file