test_select.py 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. from test import test_support
  2. import unittest
  3. import select
  4. import os
  5. import sys
  6. @unittest.skipIf(sys.platform[:3] in ('win', 'os2', 'riscos'),
  7. "can't easily test on this system")
  8. class SelectTestCase(unittest.TestCase):
  9. class Nope:
  10. pass
  11. class Almost:
  12. def fileno(self):
  13. return 'fileno'
  14. def test_error_conditions(self):
  15. self.assertRaises(TypeError, select.select, 1, 2, 3)
  16. self.assertRaises(TypeError, select.select, [self.Nope()], [], [])
  17. self.assertRaises(TypeError, select.select, [self.Almost()], [], [])
  18. self.assertRaises(TypeError, select.select, [], [], [], "not a number")
  19. def test_returned_list_identity(self):
  20. # See issue #8329
  21. r, w, x = select.select([], [], [], 1)
  22. self.assertIsNot(r, w)
  23. self.assertIsNot(r, x)
  24. self.assertIsNot(w, x)
  25. def test_select(self):
  26. cmd = 'for i in 0 1 2 3 4 5 6 7 8 9; do echo testing...; sleep 1; done'
  27. p = os.popen(cmd, 'r')
  28. for tout in (0, 1, 2, 4, 8, 16) + (None,)*10:
  29. if test_support.verbose:
  30. print 'timeout =', tout
  31. rfd, wfd, xfd = select.select([p], [], [], tout)
  32. if (rfd, wfd, xfd) == ([], [], []):
  33. continue
  34. if (rfd, wfd, xfd) == ([p], [], []):
  35. line = p.readline()
  36. if test_support.verbose:
  37. print repr(line)
  38. if not line:
  39. if test_support.verbose:
  40. print 'EOF'
  41. break
  42. continue
  43. self.fail('Unexpected return values from select():', rfd, wfd, xfd)
  44. p.close()
  45. # Issue 16230: Crash on select resized list
  46. def test_select_mutated(self):
  47. a = []
  48. class F:
  49. def fileno(self):
  50. del a[-1]
  51. return sys.__stdout__.fileno()
  52. a[:] = [F()] * 10
  53. self.assertEqual(select.select([], a, []), ([], a[:5], []))
  54. def test_main():
  55. test_support.run_unittest(SelectTestCase)
  56. test_support.reap_children()
  57. if __name__ == "__main__":
  58. test_main()