Lib/test/test_ioctl.py

   1 import unittest
   2 from test.test_support import TestSkipped, run_unittest
   3 import os, struct
   4 try:
   5     import fcntl, termios
   6 except ImportError:
   7     raise TestSkipped("No fcntl or termios module")
   8 if not hasattr(termios,'TIOCGPGRP'):
   9     raise TestSkipped("termios module doesn't have TIOCGPGRP")
  10
  11 try:
  12     tty = open("/dev/tty", "r")
  13     tty.close()
  14 except IOError:
  15     raise TestSkipped("Unable to open /dev/tty")
  16
  17 try:
  18     import pty
  19 except ImportError:
  20     pty = None
  21
  22 class IoctlTests(unittest.TestCase):
  23     def test_ioctl(self):
  24         # If this process has been put into the background, TIOCGPGRP returns
  25         # the session ID instead of the process group id.
  26         ids = (os.getpgrp(), os.getsid(0))
  27         tty = open("/dev/tty", "r")
  28         r = fcntl.ioctl(tty, termios.TIOCGPGRP, "    ")
  29         rpgrp = struct.unpack("i", r)[0]
  30         self.assert_(rpgrp in ids, "%s not in %s" % (rpgrp, ids))
  31
  32     def test_ioctl_mutate(self):
  33         import array
  34         buf = array.array('i', [0])
  35         ids = (os.getpgrp(), os.getsid(0))
  36         tty = open("/dev/tty", "r")
  37         r = fcntl.ioctl(tty, termios.TIOCGPGRP, buf, 1)
  38         rpgrp = buf[0]
  39         self.assertEquals(r, 0)
  40         self.assert_(rpgrp in ids, "%s not in %s" % (rpgrp, ids))
  41
  42     def test_ioctl_signed_unsigned_code_param(self):
  43         if not pty:
  44             raise TestSkipped('pty module required')
  45         mfd, sfd = pty.openpty()
  46         try:
  47             if termios.TIOCSWINSZ < 0:
  48                 set_winsz_opcode_maybe_neg = termios.TIOCSWINSZ
  49                 set_winsz_opcode_pos = termios.TIOCSWINSZ & 0xffffffffL
  50             else:
  51                 set_winsz_opcode_pos = termios.TIOCSWINSZ
  52                 set_winsz_opcode_maybe_neg, = struct.unpack("i",
  53                         struct.pack("I", termios.TIOCSWINSZ))
  54
  55             our_winsz = struct.pack("HHHH",80,25,0,0)
  56             # test both with a positive and potentially negative ioctl code
  57             new_winsz = fcntl.ioctl(mfd, set_winsz_opcode_pos, our_winsz)
  58             new_winsz = fcntl.ioctl(mfd, set_winsz_opcode_maybe_neg, our_winsz)
  59         finally:
  60             os.close(mfd)
  61             os.close(sfd)
  62
  63 def test_main():
  64     run_unittest(IoctlTests)
  65
  66 if __name__ == "__main__":
  67     test_main()