diff options
author | Tarek Ziadé <ziade.tarek@gmail.com> | 2009-04-12 14:57:46 (GMT) |
---|---|---|
committer | Tarek Ziadé <ziade.tarek@gmail.com> | 2009-04-12 14:57:46 (GMT) |
commit | 0eb13041d27a605c9d9ea631dfa8d223e469fe8c (patch) | |
tree | 7f694e1b58ed353f9efaee67c214b077e6347f66 | |
parent | 222de0f713ef31bf5c877b9051bcdfc39c510dd4 (diff) | |
download | cpython-0eb13041d27a605c9d9ea631dfa8d223e469fe8c.zip cpython-0eb13041d27a605c9d9ea631dfa8d223e469fe8c.tar.gz cpython-0eb13041d27a605c9d9ea631dfa8d223e469fe8c.tar.bz2 |
Merged revisions 71509 via svnmerge from
svn+ssh://pythondev@svn.python.org/python/trunk
........
r71509 | tarek.ziade | 2009-04-12 16:53:51 +0200 (Sun, 12 Apr 2009) | 1 line
removed the print statements and added a test
........
-rw-r--r-- | Lib/distutils/command/config.py | 15 | ||||
-rw-r--r-- | Lib/distutils/tests/test_config_cmd.py | 42 |
2 files changed, 52 insertions, 5 deletions
diff --git a/Lib/distutils/command/config.py b/Lib/distutils/command/config.py index 34f9188..144c513 100644 --- a/Lib/distutils/command/config.py +++ b/Lib/distutils/command/config.py @@ -336,11 +336,16 @@ class config(Command): def dump_file(filename, head=None): + """Dumps a file content into log.info. + + If head is not None, will be dumped before the file content. + """ if head is None: - print(filename + ":") + log.info('%s' % filename) else: - print(head) - + log.info(head) file = open(filename) - sys.stdout.write(file.read()) - file.close() + try: + log.info(file.read()) + finally: + file.close() diff --git a/Lib/distutils/tests/test_config_cmd.py b/Lib/distutils/tests/test_config_cmd.py new file mode 100644 index 0000000..6fd1776 --- /dev/null +++ b/Lib/distutils/tests/test_config_cmd.py @@ -0,0 +1,42 @@ +"""Tests for distutils.command.config.""" +import unittest +import os + +from distutils.command.config import dump_file +from distutils.tests import support +from distutils import log + +class ConfigTestCase(support.LoggingSilencer, + support.TempdirManager, + unittest.TestCase): + + def _info(self, msg): + for line in msg.splitlines(): + self._logs.append(line) + + def setUp(self): + super(ConfigTestCase, self).setUp() + self._logs = [] + self.old_log = log.info + log.info = self._info + + def tearDown(self): + log.info = self.old_log + super(ConfigTestCase, self).tearDown() + + def test_dump_file(self): + this_file = os.path.splitext(__file__)[0] + '.py' + f = open(this_file) + try: + numlines = len(f.readlines()) + finally: + f.close() + + dump_file(this_file, 'I am the header') + self.assertEquals(len(self._logs), numlines+1) + +def test_suite(): + return unittest.makeSuite(ConfigTestCase) + +if __name__ == "__main__": + unittest.main(defaultTest="test_suite") |