diff options
Diffstat (limited to 'testenv/test/http_test.py')
-rw-r--r-- | testenv/test/http_test.py | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/testenv/test/http_test.py b/testenv/test/http_test.py new file mode 100644 index 0000000..462ac6e --- /dev/null +++ b/testenv/test/http_test.py @@ -0,0 +1,61 @@ +from misc.colour_terminal import print_green +from server.http.http_server import HTTPd, HTTPSd +from test.base_test import BaseTest, HTTP, HTTPS + + +class HTTPTest(BaseTest): + + """ Class for HTTP Tests. """ + + # Temp Notes: It is expected that when pre-hook functions are executed, + # only an empty test-dir exists. pre-hook functions are executed just prior + # to the call to Wget is made. post-hook functions will be executed + # immediately after the call to Wget returns. + + def __init__(self, + pre_hook=None, + test_params=None, + post_hook=None, + protocols=(HTTP,), + req_protocols=None): + super(HTTPTest, self).__init__(pre_hook, + test_params, + post_hook, + protocols, + req_protocols) + + def setup(self): + self.server_setup() + self.ready = True + + def begin(self): + if not self.ready: + # this is to maintain compatibility with scripts that + # don't call setup() + self.setup() + with self: + # If any exception occurs, self.__exit__ will be immediately called. + # We must call the parent method in the end in order to verify + # whether the tests succeeded or not. + if self.ready: + self.do_test() + print_green("Test Passed.") + else: + self.tests_passed = False + return super(HTTPTest, self).begin() + + def instantiate_server_by(self, protocol): + server = {HTTP: HTTPd, + HTTPS: HTTPSd}[protocol]() + server.start() + + return server + + def request_remaining(self): + return [s.server_inst.get_req_headers() + for s in self.servers] + + def stop_server(self): + for server in self.servers: + server.server_inst.shutdown() +# vim: set ts=4 sts=4 sw=4 tw=80 et : |