From 3cd8d57271f94666117605d656b8847157aacd2b Mon Sep 17 00:00:00 2001 From: liris Date: Fri, 7 Nov 2014 12:04:22 +0900 Subject: [PATCH] clear os.envrion. --- websocket/tests/test_websocket.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/websocket/tests/test_websocket.py b/websocket/tests/test_websocket.py index a6cae28..0503458 100644 --- a/websocket/tests/test_websocket.py +++ b/websocket/tests/test_websocket.py @@ -537,11 +537,25 @@ class UtilsTest(unittest.TestCase): class ProxyInfoTest(unittest.TestCase): def setUp(self): + self.http_proxy = os.environ.get("http_proxy", None) + self.https_proxy = os.environ.get("https_proxy", None) if "http_proxy" in os.environ: del os.environ["http_proxy"] if "https_proxy" in os.environ: del os.environ["https_proxy"] + def tearDown(self): + if self.http_proxy: + os.environ["http_proxy"] = self.http_proxy + elif "http_proxy" in os.environ: + del os.environ["http_proxy"] + + if self.https_proxy: + os.environ["https_proxy"] = self.https_proxy + elif "https_proxy" in os.environ: + del os.environ["https_proxy"] + + def testProxyFromArgs(self): self.assertEqual(_get_proxy_info(False, http_proxy_host="localhost"), ("localhost", 0, None)) self.assertEqual(_get_proxy_info(False, http_proxy_host="localhost", http_proxy_port=3128), ("localhost", 3128, None))