Home | History | Annotate | Download | only in tests 
      1 """Tests for distutils.spawn."""
      2 import unittest
      3 import os
      4 import time
      5 from test.test_support import captured_stdout, run_unittest
      6 
      7 from distutils.spawn import _nt_quote_args
      8 from distutils.spawn import spawn, find_executable
      9 from distutils.errors import DistutilsExecError
     10 from distutils.tests import support
     11 
     12 class SpawnTestCase(support.TempdirManager,
     13                     support.LoggingSilencer,
     14                     unittest.TestCase):
     15 
     16     def test_nt_quote_args(self):
     17 
     18         for (args, wanted) in ((['with space', 'nospace'],
     19                                 ['"with space"', 'nospace']),
     20                                (['nochange', 'nospace'],
     21                                 ['nochange', 'nospace'])):
     22             res = _nt_quote_args(args)
     23             self.assertEqual(res, wanted)
     24 
     25 
     26     @unittest.skipUnless(os.name in ('nt', 'posix'),
     27                          'Runs only under posix or nt')
     28     def test_spawn(self):
     29         tmpdir = self.mkdtemp()
     30 
     31         # creating something executable
     32         # through the shell that returns 1
     33         if os.name == 'posix':
     34             exe = os.path.join(tmpdir, 'foo.sh')
     35             self.write_file(exe, '#!/bin/sh\nexit 1')
     36             os.chmod(exe, 0777)
     37         else:
     38             exe = os.path.join(tmpdir, 'foo.bat')
     39             self.write_file(exe, 'exit 1')
     40 
     41         os.chmod(exe, 0777)
     42         self.assertRaises(DistutilsExecError, spawn, [exe])
     43 
     44         # now something that works
     45         if os.name == 'posix':
     46             exe = os.path.join(tmpdir, 'foo.sh')
     47             self.write_file(exe, '#!/bin/sh\nexit 0')
     48             os.chmod(exe, 0777)
     49         else:
     50             exe = os.path.join(tmpdir, 'foo.bat')
     51             self.write_file(exe, 'exit 0')
     52 
     53         os.chmod(exe, 0777)
     54         spawn([exe])  # should work without any error
     55 
     56 def test_suite():
     57     return unittest.makeSuite(SpawnTestCase)
     58 
     59 if __name__ == "__main__":
     60     run_unittest(test_suite())
     61