unittest.mock: set file_spec on first use

a74561a5 · Michael Foord · 50a8c0ef · a74561a5
Kaydet (Commit) a74561a5 authored Mar 25, 2012 tarafından Michael Foord
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 2 deletions

mock.py Lib/unittest/mock.py +6 -2

No files found.
--- a/Lib/unittest/mock.py
+++ b/Lib/unittest/mock.py
@@ -2138,11 +2138,15 @@ FunctionAttributes = set([
    'func_name',
 ])

-import _io
-file_spec = list(set(dir(_io.TextIOWrapper)).union(set(dir(_io.BytesIO))))

+file_spec = None

 def mock_open(mock=None, read_data=None):
+    global file_spec
+    if file_spec is None:
+        import _io
+        file_spec = list(set(dir(_io.TextIOWrapper)).union(set(dir(_io.BytesIO))))
+
    if mock is None:
        mock = MagicMock(spec=file_spec)