openstack/deb-python-websocket-client
Code Issues Proposed changes

clear os.envrion.

Browse Source
This commit is contained in:
liris
2014-11-07 12:04:22 +09:00
parent 19f2ef97d9
commit 3cd8d57271
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
websocket/tests/test_websocket.py
View File

@@ -537,11 +537,25 @@ class UtilsTest(unittest.TestCase):
class ProxyInfoTest(unittest.TestCase): class ProxyInfoTest(unittest.TestCase):
def setUp(self): 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: if "http_proxy" in os.environ:
del os.environ["http_proxy"] del os.environ["http_proxy"]
if "https_proxy" in os.environ: if "https_proxy" in os.environ:
del os.environ["https_proxy"] 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): 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"), ("localhost", 0, None))
self.assertEqual(_get_proxy_info(False, http_proxy_host="localhost", http_proxy_port=3128), ("localhost", 3128, None)) self.assertEqual(_get_proxy_info(False, http_proxy_host="localhost", http_proxy_port=3128), ("localhost", 3128, None))