diff options
author | Miss Islington (bot) <31488909+miss-islington@users.noreply.github.com> | 2022-05-08 15:12:19 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-05-08 15:12:19 (GMT) |
commit | c63c8ac2389d715c761f56bfcf17a685b62a0bd3 (patch) | |
tree | 5b4e82a1f54099f9a75217ac0045aa071dcf4188 /Lib/test/test_urllib.py | |
parent | a85bdd7e025f8b87b88d914f4df8f0b620398ea9 (diff) | |
download | cpython-c63c8ac2389d715c761f56bfcf17a685b62a0bd3.zip cpython-c63c8ac2389d715c761f56bfcf17a685b62a0bd3.tar.gz cpython-c63c8ac2389d715c761f56bfcf17a685b62a0bd3.tar.bz2 |
bpo-45046: Support context managers in unittest (GH-28045)
Add methods enterContext() and enterClassContext() in TestCase.
Add method enterAsyncContext() in IsolatedAsyncioTestCase.
Add function enterModuleContext().
(cherry picked from commit 086c6b1b0fe8d47ebd15512d7bdcb64c60a360f0)
Co-authored-by: Serhiy Storchaka <storchaka@gmail.com>
Diffstat (limited to 'Lib/test/test_urllib.py')
-rw-r--r-- | Lib/test/test_urllib.py | 7 |
1 files changed, 1 insertions, 6 deletions
diff --git a/Lib/test/test_urllib.py b/Lib/test/test_urllib.py index 82f1d9d..bc6e74c 100644 --- a/Lib/test/test_urllib.py +++ b/Lib/test/test_urllib.py @@ -232,17 +232,12 @@ class ProxyTests(unittest.TestCase): def setUp(self): # Records changes to env vars - self.env = os_helper.EnvironmentVarGuard() + self.env = self.enterContext(os_helper.EnvironmentVarGuard()) # Delete all proxy related env vars for k in list(os.environ): if 'proxy' in k.lower(): self.env.unset(k) - def tearDown(self): - # Restore all proxy related env vars - self.env.__exit__() - del self.env - def test_getproxies_environment_keep_no_proxies(self): self.env.set('NO_PROXY', 'localhost') proxies = urllib.request.getproxies_environment() |