From ac27d1236fdbf69c826b5d2bb4c69ebbe8ab40a1 Mon Sep 17 00:00:00 2001 From: Maximilian Hils Date: Sun, 7 Sep 2014 18:01:30 +0200 Subject: improve change_server api, add example how to change the upstream server --- test/test_server.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) (limited to 'test') diff --git a/test/test_server.py b/test/test_server.py index d33bcc89..b128d0a2 100644 --- a/test/test_server.py +++ b/test/test_server.py @@ -387,14 +387,12 @@ class MasterRedirectRequest(tservers.TestMaster): def handle_request(self, f): request = f.request if request.path == "/p/201": - url = request.url - new = "http://127.0.0.1:%s/p/201" % self.redirect_port - - request.url = new - f.live.change_server(("127.0.0.1", self.redirect_port), False) - request.url = url - tutils.raises("SSL handshake error", f.live.change_server, ("127.0.0.1", self.redirect_port), True) - request.url = new + addr = f.live.c.server_conn.address + assert f.live.change_server(("127.0.0.1", self.redirect_port), ssl=False) + assert not f.live.change_server(("127.0.0.1", self.redirect_port), ssl=False) + tutils.raises("SSL handshake error", f.live.change_server, ("127.0.0.1", self.redirect_port), ssl=True) + assert f.live.change_server(addr, ssl=False) + request.url = "http://127.0.0.1:%s/p/201" % self.redirect_port tservers.TestMaster.handle_request(self, f) def handle_response(self, f): -- cgit v1.2.3