Home | History | Annotate | Download | only in test 
      1 '''
      2    Tests for commands module
      3    Nick Mathewson
      4 '''
      5 import unittest
      6 import os, tempfile, re
      7 
      8 from test.test_support import run_unittest, reap_children, import_module, \
      9                               check_warnings
     10 
     11 # Silence Py3k warning
     12 commands = import_module('commands', deprecated=True)
     13 
     14 # The module says:
     15 #   "NB This only works (and is only relevant) for UNIX."
     16 #
     17 # Actually, getoutput should work on any platform with an os.popen, but
     18 # I'll take the comment as given, and skip this suite.
     19 
     20 if os.name != 'posix':
     21     raise unittest.SkipTest('Not posix; skipping test_commands')
     22 
     23 
     24 class CommandTests(unittest.TestCase):
     25 
     26     def test_getoutput(self):
     27         self.assertEqual(commands.getoutput('echo xyzzy'), 'xyzzy')
     28         self.assertEqual(commands.getstatusoutput('echo xyzzy'), (0, 'xyzzy'))
     29 
     30         # we use mkdtemp in the next line to create an empty directory
     31         # under our exclusive control; from that, we can invent a pathname
     32         # that we _know_ won't exist.  This is guaranteed to fail.
     33         dir = None
     34         try:
     35             dir = tempfile.mkdtemp()
     36             name = os.path.join(dir, "foo")
     37 
     38             status, output = commands.getstatusoutput('cat ' + name)
     39             self.assertNotEqual(status, 0)
     40         finally:
     41             if dir is not None:
     42                 os.rmdir(dir)
     43 
     44     def test_getstatus(self):
     45         # This pattern should match 'ls -ld /.' on any posix
     46         # system, however perversely configured.  Even on systems
     47         # (e.g., Cygwin) where user and group names can have spaces:
     48         #     drwxr-xr-x   15 Administ Domain U     4096 Aug 12 12:50 /
     49         #     drwxr-xr-x   15 Joe User My Group     4096 Aug 12 12:50 /
     50         # Note that the first case above has a space in the group name
     51         # while the second one has a space in both names.
     52         # Special attributes supported:
     53         #   + = has ACLs
     54         #   @ = has Mac OS X extended attributes
     55         #   . = has a SELinux security context
     56         pat = r'''d.........   # It is a directory.
     57                   [.+@]?       # It may have special attributes.
     58                   \s+\d+       # It has some number of links.
     59                   [^/]*        # Skip user, group, size, and date.
     60                   /\.          # and end with the name of the file.
     61                '''
     62 
     63         with check_warnings((".*commands.getstatus.. is deprecated",
     64                              DeprecationWarning)):
     65             self.assertTrue(re.match(pat, commands.getstatus("/."), re.VERBOSE))
     66 
     67 
     68 def test_main():
     69     run_unittest(CommandTests)
     70     reap_children()
     71 
     72 
     73 if __name__ == "__main__":
     74     test_main()
     75