X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=tests%2Fdcaegen2%2Ftestcases%2Fresources%2Frobot_library%2FDmaapLibrary.py;h=f79ba04e938481618a634eaeaaa453db400ba7ff;hb=1a2435d72a46d8daec2a0259275fee4842da5a9a;hp=c9a0ff7bd60573400385bd077a551a54fe11185e;hpb=b950ff8ea39c666ba3e593d051cfd826be6fd080;p=integration%2Fcsit.git

diff --git a/tests/dcaegen2/testcases/resources/robot_library/DmaapLibrary.py b/tests/dcaegen2/testcases/resources/robot_library/DmaapLibrary.py
index c9a0ff7b..f79ba04e 100644
--- a/tests/dcaegen2/testcases/resources/robot_library/DmaapLibrary.py
+++ b/tests/dcaegen2/testcases/resources/robot_library/DmaapLibrary.py
@@ -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