Mirror
/
cpython
mirror of https://github.com/python/cpython
4
1
Fork 0
Code Issues Releases Wiki Activity

Rework test_record a bit to make the test more exact

This commit is contained in:
Éric Araujo 2011-08-20 07:00:41 +02:00
parent def15dafda
commit 9358bfdaff
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
Lib/distutils/tests/test_install.py
View File

@ -1,7 +1,6 @@
"""Tests for distutils.command.install.""" """Tests for distutils.command.install."""
import os import os
import os.path
import sys import sys
import unittest import unittest
import site import site
@ -167,33 +166,36 @@ class InstallTestCase(support.TempdirManager,
self.assertRaises(DistutilsOptionError, cmd.finalize_options) self.assertRaises(DistutilsOptionError, cmd.finalize_options)
def test_record(self): def test_record(self):
install_dir = self.mkdtemp() install_dir = self.mkdtemp()
pkgdir, dist = self.create_dist() project_dir, dist = self.create_dist(scripts=['hello'])
self.addCleanup(os.chdir, os.getcwd())
os.chdir(project_dir)
self.write_file('hello', "print('o hai')")
dist = Distribution()
cmd = install(dist) cmd = install(dist)
dist.command_obj['install'] = cmd dist.command_obj['install'] = cmd
cmd.root = install_dir cmd.root = install_dir
cmd.record = os.path.join(pkgdir, 'RECORD') cmd.record = os.path.join(project_dir, 'RECORD')
cmd.ensure_finalized() cmd.ensure_finalized()
cmd.run() cmd.run()
# let's check the RECORD file was created with one
# line (the egg info file)
f = open(cmd.record) f = open(cmd.record)
try: try:
self.assertEqual(len(f.readlines()), 1) content = f.read()
finally: finally:
f.close() f.close()
found = [os.path.basename(line) for line in content.splitlines()]
expected = ['hello',
'UNKNOWN-0.0.0-py%s.%s.egg-info' % sys.version_info[:2]]
self.assertEqual(found, expected)
def test_debug_mode(self): def test_debug_mode(self):
# this covers the code called when DEBUG is set # this covers the code called when DEBUG is set
old_logs_len = len(self.logs) old_logs_len = len(self.logs)
install_module.DEBUG = True install_module.DEBUG = True
try: try:
with captured_stdout() as stdout: with captured_stdout():
self.test_record() self.test_record()
finally: finally:
install_module.DEBUG = False install_module.DEBUG = False