Viewing file: test_build_scripts.py (2.47 KB) -rw-r--r-- Select action/file-type: (+) | (+) | (+) | Code (+) | Session (+) | (+) | SDB (+) | (+) | (+) | (+) | (+) | (+) |
"""Tests for distutils.command.build_scripts."""
import os import unittest
from distutils.command.build_scripts import build_scripts from distutils.core import Distribution
from distutils.tests import support
class BuildScriptsTestCase(support.TempdirManager, support.LoggingSilencer, unittest.TestCase):
def test_default_settings(self): cmd = self.get_build_scripts_cmd("/foo/bar", []) self.assert_(not cmd.force) self.assert_(cmd.build_dir is None)
cmd.finalize_options()
self.assert_(cmd.force) self.assertEqual(cmd.build_dir, "/foo/bar")
def test_build(self): source = self.mkdtemp() target = self.mkdtemp() expected = self.write_sample_scripts(source)
cmd = self.get_build_scripts_cmd(target, [os.path.join(source, fn) for fn in expected]) cmd.finalize_options() cmd.run()
built = os.listdir(target) for name in expected: self.assert_(name in built)
def get_build_scripts_cmd(self, target, scripts): import sys dist = Distribution() dist.scripts = scripts dist.command_obj["build"] = support.DummyCommand( build_scripts=target, force=1, executable=sys.executable ) return build_scripts(dist)
def write_sample_scripts(self, dir): expected = [] expected.append("script1.py") self.write_script(dir, "script1.py", ("#! /usr/bin/env python2.3\n" "# bogus script w/ Python sh-bang\n" "pass\n")) expected.append("script2.py") self.write_script(dir, "script2.py", ("#!/usr/bin/python\n" "# bogus script w/ Python sh-bang\n" "pass\n")) expected.append("shell.sh") self.write_script(dir, "shell.sh", ("#!/bin/sh\n" "# bogus shell script w/ sh-bang\n" "exit 0\n")) return expected
def write_script(self, dir, name, text): f = open(os.path.join(dir, name), "w") f.write(text) f.close()
def test_suite(): return unittest.makeSuite(BuildScriptsTestCase)
if __name__ == "__main__": unittest.main(defaultTest="test_suite")
|