CSIT tests update for Synchronous VES collector

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

diff --git a/tests/dcaegen2/testcases/resources/robot_library/dmaap_simulator/DMaaPServer.py b/tests/dcaegen2/testcases/resources/robot_library/dmaap_simulator/DMaaPServer.py
index 37499be..8b66fd4 100644 (file)
--- a/tests/dcaegen2/testcases/resources/robot_library/dmaap_simulator/DMaaPServer.py
+++ b/tests/dcaegen2/testcases/resources/robot_library/dmaap_simulator/DMaaPServer.py
@@ -14,6 +14,12 @@ class DMaaPServer(BaseHTTPServer.HTTPServer):
         serer_address = self.socket.getsockname()
         print "Serving HTTP on", serer_address[0], "port", serer_address[1], "..."
 
+    def set_dmaap_successfull_code(self,code_number):
+        DMaaPHandler.DMaaPHandler.succes_response_code=code_number
+
+    def reset_dmaap_succesfull_code(self):
+        DMaaPHandler.DMaaPHandler.succes_response_code=DMaaPHandler.DMaaPHandler.DEFAULT_SUCCES_RESPONSE_CODE
+
 
 def create_dmaap_server(dmaap_simulator, protocol="HTTP/1.0", port=3904):
     server_address = ('', port)