class RESTOperationLiveServerTestCase(TestCase): def setUp(self): self.fake_server = HttpTestServer() self.fake_server.start() def tearDown(self): self.fake_server.stop() def test_OperationWillDoHTTPOperation(self): class TestHandler(SimpleGETHTTPRequestHandler): resp = 'HelloWorld!' self.fake_server.replace_request_handler_with(TestHandler) op = RESTOperation('GET', 'http://{0}:{1}/'.format( self.fake_server.hostname, self.fake_server.port)) op.execute() self.assertEqual(TestHandler.resp, op.response) def test_OperationWillSendHeaders(self): expected_headers = {'x-test-1' : 'hi', 'x-test-2': 'bye'} class TestHandler(SimpleGETHTTPRequestHandler): def do_GET(self): self.send_response(200) self.end_headers() special_keys = [k for k in self.headers if k.startswith('x-')] resp = '/n'.join(['{0} {1}'.format( x, self.headers[x]) for x in special_keys]) self.wfile.write(resp) return self.fake_server.replace_request_handler_with(TestHandler) op = RESTOperation('GET', 'http://{0}:{1}/'.format( self.fake_server.hostname, self.fake_server.port), '', expected_headers) op.execute() expected_response = '/n'.join( ['{0} {1}'.format(x, expected_headers[x]) for x in expected_headers]) self.assertEqual(expected_response, op.response)
def setUp(self): self.fake_server = HttpTestServer() self.fake_server.start()