X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=tests%2Fdcaegen2%2Ftestcases%2Fresources%2Frobot_library%2Fdmaap_simulator%2FDMaaPServer.py;h=c40a2ef3b8f4c37a3044c82e857237f9acd1fa78;hb=f862683a7e4363cde071458e399e28dd4afea33e;hp=37499be82ba0b75b186e05c243380ec3694c3fe7;hpb=a28f21ec8bd7c22aac03e59a933db7958b212fab;p=integration%2Fcsit.git

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 37499be8..c40a2ef3 100644
--- a/tests/dcaegen2/testcases/resources/robot_library/dmaap_simulator/DMaaPServer.py
+++ b/tests/dcaegen2/testcases/resources/robot_library/dmaap_simulator/DMaaPServer.py
@@ -1,18 +1,24 @@
-import BaseHTTPServer
-import DMaaPHandler
+import http.server
+from robot_library.dmaap_simulator import DMaaPHandler
 
 
-class DMaaPServer(BaseHTTPServer.HTTPServer):
+class DMaaPServer(http.server.HTTPServer):
 
     def __init__(self, server_address, protocol, dmaap_simulator):
 
         def handler_class_constructor(*args):
             DMaaPHandler.DMaaPHandler(dmaap_simulator, *args)
         DMaaPHandler.protocol_version = protocol
-        BaseHTTPServer.HTTPServer.__init__(self, server_address, handler_class_constructor)
+        http.server.HTTPServer.__init__(self, server_address, handler_class_constructor)
 
         serer_address = self.socket.getsockname()
-        print "Serving HTTP on", serer_address[0], "port", serer_address[1], "..."
+        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):