/Lib/distutils/tests/test_install_scripts.py
http://unladen-swallow.googlecode.com/ · Python · 79 lines · 62 code · 16 blank · 1 comment · 2 complexity · e545553a25d255b6c88e9d2b94c26cf5 MD5 · raw file
- """Tests for distutils.command.install_scripts."""
- import os
- import unittest
- from distutils.command.install_scripts import install_scripts
- from distutils.core import Distribution
- from distutils.tests import support
- class InstallScriptsTestCase(support.TempdirManager,
- support.LoggingSilencer,
- unittest.TestCase):
- def test_default_settings(self):
- dist = Distribution()
- 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,
- )
- cmd = install_scripts(dist)
- self.assert_(not cmd.force)
- self.assert_(not cmd.skip_build)
- self.assert_(cmd.build_dir is None)
- self.assert_(cmd.install_dir is None)
- cmd.finalize_options()
- self.assert_(cmd.force)
- self.assert_(cmd.skip_build)
- self.assertEqual(cmd.build_dir, "/foo/bar")
- self.assertEqual(cmd.install_dir, "/splat/funk")
- def test_installation(self):
- source = self.mkdtemp()
- expected = []
- def write_script(name, text):
- expected.append(name)
- f = open(os.path.join(source, name), "w")
- f.write(text)
- f.close()
- write_script("script1.py", ("#! /usr/bin/env python2.3\n"
- "# bogus script w/ Python sh-bang\n"
- "pass\n"))
- write_script("script2.py", ("#!/usr/bin/python\n"
- "# bogus script w/ Python sh-bang\n"
- "pass\n"))
- write_script("shell.sh", ("#!/bin/sh\n"
- "# bogus shell script w/ sh-bang\n"
- "exit 0\n"))
- target = self.mkdtemp()
- dist = Distribution()
- dist.command_obj["build"] = support.DummyCommand(build_scripts=source)
- dist.command_obj["install"] = support.DummyCommand(
- install_scripts=target,
- force=1,
- skip_build=1,
- )
- cmd = install_scripts(dist)
- cmd.finalize_options()
- cmd.run()
- installed = os.listdir(target)
- for name in expected:
- self.assert_(name in installed)
- def test_suite():
- return unittest.makeSuite(InstallScriptsTestCase)
- if __name__ == "__main__":
- unittest.main(defaultTest="test_suite")