Avoid global side effect in test_ensurepip

This commit is contained in:
Nick Coghlan 2013-11-13 22:24:58 +10:00
parent ca351e6bc5
commit d3e83e2a3e
1 changed files with 8 additions and 3 deletions

View File

@ -2,6 +2,8 @@
import unittest.mock import unittest.mock
import ensurepip import ensurepip
import test.support import test.support
import os
import os.path
class TestEnsurePipVersion(unittest.TestCase): class TestEnsurePipVersion(unittest.TestCase):
@ -17,9 +19,12 @@ def setUp(self):
self.run_pip = run_pip_patch.start() self.run_pip = run_pip_patch.start()
self.addCleanup(run_pip_patch.stop) self.addCleanup(run_pip_patch.stop)
os_environ_patch = unittest.mock.patch("ensurepip.os.environ", {}) # Avoid side effects on the actual os module
self.os_environ = os_environ_patch.start() os_patch = unittest.mock.patch("ensurepip.os")
self.addCleanup(os_environ_patch.stop) patched_os = os_patch.start()
self.addCleanup(os_patch.stop)
patched_os.path = os.path
self.os_environ = patched_os.environ = os.environ.copy()
def test_basic_bootstrapping(self): def test_basic_bootstrapping(self):
ensurepip.bootstrap() ensurepip.bootstrap()