def patch_get_entry_info_for_gui_scripts():
import pkg_resources
def get_entry_info(self, group, name):
return self.get_entry_map("gui_scripts" if group == "console_scripts" else group).get(name)
with patch(pkg_resources.Distribution, "get_entry_info", get_entry_info):
yield
def patch_get_entry_map_for_gui_scripts():
import pkg_resources
_get_entry_map = pkg_resources.get_entry_map
def get_entry_map(dist, group=None):
return _get_entry_map(dist, "gui_scripts")
with patch(pkg_resources, "get_entry_map", get_entry_map):
yield
def test__patch_context(self):
with patch(os, "listdir", listdir_patch):
result = os.listdir(TEST_PATH)
self.assertEqual(result, [TEST_PATH])
self.assertNotEqual(os.listdir('.'), [TEST_PATH])
