# Testing select module
from test.test_support import verbose, reap_children, TestSkipped, run_unittest
import select
import os
import unittest

class Nope:
    pass

class Almost:
    def fileno(self):
        return 'fileno'

class TestSelect(unittest.TestCase):
    
    def test_errors(self):
        # test some known error conditions
        self.assertRaises(TypeError, select.select, 1, 2, 3)
        self.assertRaises(TypeError, select.select, [Nope()], [], [])
        self.assertRaises(TypeError, select.select, [Almost()], [], [])
        self.assertRaises(TypeError, select.select, [], [], [], 'not a number')
        
    def test_behavior(self):
        import sys
        if sys.platform[:3] in ('win', 'mac', 'os2', 'riscos'):
            raise TestSkipped("Can't test select easily on " + sys.platform)

        cmd = 'for i in 0 1 2 3 4 5 6 7 8 9; do echo testing...; sleep 1; done'
        p = os.popen(cmd, 'r')
        for tout in (0, 1, 2, 4, 8, 16) + (None,)*10:
            if verbose:
                print 'timeout =', tout
            rfd, wfd, xfd = select.select([p], [], [], tout)
            if (rfd, wfd, xfd) == ([], [], []):
                continue
            if (rfd, wfd, xfd) == ([p], [], []):
                line = p.readline()
                if verbose:
                    print repr(line)
                if not line:
                    if verbose:
                        print 'EOF'
                    break
                continue
            self.fail('Unexpected return values from select(): %s %s %s'
                      %(rfd, wfd, xfd))
        p.close()
        reap_children()


def test_main():
    run_unittest(TestSelect)

if __name__ == '__main__':
    test_main()