move support code to a helper module to ease re-use
This commit is contained in:
parent
9e1ac2496f
commit
b8ab8b6da8
|
@ -0,0 +1,41 @@
|
|||
"""Support code for distutils test cases."""
|
||||
|
||||
import shutil
|
||||
import tempfile
|
||||
|
||||
|
||||
class TempdirManager(object):
|
||||
"""Mix-in class that handles temporary directories for test cases.
|
||||
|
||||
This is intended to be used with unittest.TestCase.
|
||||
"""
|
||||
|
||||
def setUp(self):
|
||||
super(TempdirManager, self).setUp()
|
||||
self.tempdirs = []
|
||||
|
||||
def tearDown(self):
|
||||
super(TempdirManager, self).tearDown()
|
||||
while self.tempdirs:
|
||||
d = self.tempdirs.pop()
|
||||
shutil.rmtree(d)
|
||||
|
||||
def mkdtemp(self):
|
||||
"""Create a temporary directory that will be cleaned up.
|
||||
|
||||
Returns the path of the directory.
|
||||
"""
|
||||
d = tempfile.mkdtemp()
|
||||
self.tempdirs.append(d)
|
||||
return d
|
||||
|
||||
|
||||
class DummyCommand:
|
||||
"""Class to store options for retrieval via set_undefined_options()."""
|
||||
|
||||
def __init__(self, **kwargs):
|
||||
for kw, val in kwargs.items():
|
||||
setattr(self, kw, val)
|
||||
|
||||
def ensure_finalized(self):
|
||||
pass
|
|
@ -1,37 +1,21 @@
|
|||
"""Tests for distutils.command.install_scripts."""
|
||||
|
||||
import os
|
||||
import shutil
|
||||
import tempfile
|
||||
import unittest
|
||||
|
||||
from distutils.command.install_scripts import install_scripts
|
||||
from distutils.core import Distribution
|
||||
|
||||
from distutils.tests import support
|
||||
|
||||
class InstallScriptsTestCase(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
self.tempdirs = []
|
||||
|
||||
def tearDown(self):
|
||||
while self.tempdirs:
|
||||
d = self.tempdirs.pop()
|
||||
shutil.rmtree(d)
|
||||
|
||||
def mkdtemp(self):
|
||||
"""Create a temporary directory that will be cleaned up.
|
||||
|
||||
Returns the path of the directory.
|
||||
"""
|
||||
d = tempfile.mkdtemp()
|
||||
self.tempdirs.append(d)
|
||||
return d
|
||||
class InstallScriptsTestCase(support.TempdirManager, unittest.TestCase):
|
||||
|
||||
def test_default_settings(self):
|
||||
dist = Distribution()
|
||||
dist.command_obj["build"] = DummyCommand(build_scripts="/foo/bar")
|
||||
dist.command_obj["install"] = DummyCommand(
|
||||
dist.command_obj["build"] = support.DummyCommand(
|
||||
build_scripts="/foo/bar")
|
||||
dist.command_obj["install"] = support.DummyCommand(
|
||||
install_scripts="/splat/funk",
|
||||
force=1,
|
||||
skip_build=1,
|
||||
|
@ -71,8 +55,8 @@ class InstallScriptsTestCase(unittest.TestCase):
|
|||
|
||||
target = self.mkdtemp()
|
||||
dist = Distribution()
|
||||
dist.command_obj["build"] = DummyCommand(build_scripts=source)
|
||||
dist.command_obj["install"] = DummyCommand(
|
||||
dist.command_obj["build"] = support.DummyCommand(build_scripts=source)
|
||||
dist.command_obj["install"] = support.DummyCommand(
|
||||
install_scripts=target,
|
||||
force=1,
|
||||
skip_build=1,
|
||||
|
@ -86,17 +70,6 @@ class InstallScriptsTestCase(unittest.TestCase):
|
|||
self.assert_(name in installed)
|
||||
|
||||
|
||||
class DummyCommand:
|
||||
"""Class to store options for retrieval via set_undefined_options()."""
|
||||
|
||||
def __init__(self, **kwargs):
|
||||
for kw, val in kwargs.items():
|
||||
setattr(self, kw, val)
|
||||
|
||||
def ensure_finalized(self):
|
||||
pass
|
||||
|
||||
|
||||
|
||||
def test_suite():
|
||||
return unittest.makeSuite(InstallScriptsTestCase)
|
||||
|
|
Loading…
Reference in New Issue