summaryrefslogtreecommitdiffstats
path: root/Lib/test/libregrtest/save_env.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/libregrtest/save_env.py')
-rw-r--r--Lib/test/libregrtest/save_env.py13
1 files changed, 13 insertions, 0 deletions
diff --git a/Lib/test/libregrtest/save_env.py b/Lib/test/libregrtest/save_env.py
index 31931f2..e7c27a6 100644
--- a/Lib/test/libregrtest/save_env.py
+++ b/Lib/test/libregrtest/save_env.py
@@ -7,6 +7,7 @@ import shutil
import sys
import sysconfig
import threading
+import urllib.request
import warnings
from test import support
from test.libregrtest.utils import print_warning
@@ -68,8 +69,20 @@ class saved_test_environment:
'files', 'locale', 'warnings.showwarning',
'shutil_archive_formats', 'shutil_unpack_formats',
'asyncio.events._event_loop_policy',
+ 'urllib.requests._url_tempfiles', 'urllib.requests._opener',
)
+ def get_urllib_requests__url_tempfiles(self):
+ return list(urllib.request._url_tempfiles)
+ def restore_urllib_requests__url_tempfiles(self, tempfiles):
+ for filename in tempfiles:
+ support.unlink(filename)
+
+ def get_urllib_requests__opener(self):
+ return urllib.request._opener
+ def restore_urllib_requests__opener(self, opener):
+ urllib.request._opener = opener
+
def get_asyncio_events__event_loop_policy(self):
return support.maybe_get_event_loop_policy()
def restore_asyncio_events__event_loop_policy(self, policy):