aboutsummaryrefslogtreecommitdiffstats
path: root/tests/dcaegen2/testcases/resources/robot_library/dmaap_simulator/DMaaPServer.py
blob: 8b66fd4dea0623e6b7d4d630df6136b5734abff0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import BaseHTTPServer
import DMaaPHandler


class DMaaPServer(BaseHTTPServer.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)

        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)
    httpd = DMaaPServer(server_address, protocol, dmaap_simulator)

    return httpd