test_build_scripts.py 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596
  1. """Tests for distutils.command.build_scripts."""
  2. import os
  3. import textwrap
  4. from distutils import sysconfig
  5. from distutils.command.build_scripts import build_scripts
  6. from distutils.core import Distribution
  7. from distutils.tests import support
  8. import jaraco.path
  9. class TestBuildScripts(support.TempdirManager):
  10. def test_default_settings(self):
  11. cmd = self.get_build_scripts_cmd("/foo/bar", [])
  12. assert not cmd.force
  13. assert cmd.build_dir is None
  14. cmd.finalize_options()
  15. assert cmd.force
  16. assert cmd.build_dir == "/foo/bar"
  17. def test_build(self):
  18. source = self.mkdtemp()
  19. target = self.mkdtemp()
  20. expected = self.write_sample_scripts(source)
  21. cmd = self.get_build_scripts_cmd(
  22. target, [os.path.join(source, fn) for fn in expected]
  23. )
  24. cmd.finalize_options()
  25. cmd.run()
  26. built = os.listdir(target)
  27. for name in expected:
  28. assert name in built
  29. def get_build_scripts_cmd(self, target, scripts):
  30. import sys
  31. dist = Distribution()
  32. dist.scripts = scripts
  33. dist.command_obj["build"] = support.DummyCommand(
  34. build_scripts=target, force=True, executable=sys.executable
  35. )
  36. return build_scripts(dist)
  37. @staticmethod
  38. def write_sample_scripts(dir):
  39. spec = {
  40. 'script1.py': textwrap.dedent("""
  41. #! /usr/bin/env python2.3
  42. # bogus script w/ Python sh-bang
  43. pass
  44. """).lstrip(),
  45. 'script2.py': textwrap.dedent("""
  46. #!/usr/bin/python
  47. # bogus script w/ Python sh-bang
  48. pass
  49. """).lstrip(),
  50. 'shell.sh': textwrap.dedent("""
  51. #!/bin/sh
  52. # bogus shell script w/ sh-bang
  53. exit 0
  54. """).lstrip(),
  55. }
  56. jaraco.path.build(spec, dir)
  57. return list(spec)
  58. def test_version_int(self):
  59. source = self.mkdtemp()
  60. target = self.mkdtemp()
  61. expected = self.write_sample_scripts(source)
  62. cmd = self.get_build_scripts_cmd(
  63. target, [os.path.join(source, fn) for fn in expected]
  64. )
  65. cmd.finalize_options()
  66. # https://bugs.python.org/issue4524
  67. #
  68. # On linux-g++-32 with command line `./configure --enable-ipv6
  69. # --with-suffix=3`, python is compiled okay but the build scripts
  70. # failed when writing the name of the executable
  71. old = sysconfig.get_config_vars().get('VERSION')
  72. sysconfig._config_vars['VERSION'] = 4
  73. try:
  74. cmd.run()
  75. finally:
  76. if old is not None:
  77. sysconfig._config_vars['VERSION'] = old
  78. built = os.listdir(target)
  79. for name in expected:
  80. assert name in built