tests/python/testsu.py

   1 #!/usr/bin/env python2.2
   2 from __future__ import generators
   3 import unittest
   4 import sys, os
   5 from os.path import dirname, abspath, join
   6
   7 rox_lib = dirname(dirname(dirname(abspath(sys.argv[0]))))
   8 sys.path.insert(0, join(rox_lib, 'python'))
   9
  10 from rox import su, tasks, g
  11
  12 assert os.getuid() != 0, "Can't run tests as root"
  13
  14 class TestSU(unittest.TestCase):
  15         def testSu(self):
  16                 def run():
  17                         maker = su.SuProxyMaker(
  18                                 'Need to become root to test this module.')
  19                         yield maker.blocker
  20                         root = maker.get_root()
  21
  22                         response = root.spawnvpe(os.P_NOWAIT, 'false', ['false'])
  23                         yield response
  24                         pid = response.result
  25                         assert pid
  26                         response = root.waitpid(pid, 0)
  27                         yield response
  28                         (pid, status) = response.result
  29                         assert status == 0x100
  30
  31                         response = root.spawnvpe(os.P_WAIT, 'true', ['true'])
  32                         yield response
  33                         assert response.result == 0
  34
  35                         response = root.getuid()
  36                         yield response
  37                         assert response.result == 0
  38
  39                         response = root.setuid(os.getuid())
  40                         yield response
  41                         assert response.result is None
  42
  43                         response = root.getuid()
  44                         yield response
  45                         assert response.result == os.getuid()
  46
  47                         root.finish()
  48                         g.mainquit()
  49
  50                 tasks.Task(run())
  51                 g.mainloop()
  52
  53 sys.argv.append('-v')
  54 unittest.main()