1 import socket

     2 import threading

     3 import telnetlib

     4 import time

     5 

     6 from unittest import TestCase

     7 from test import test_support

     8 

     9 HOST = test_support.HOST

    10 

    11 def server(evt, serv):

    12     serv.listen(5)

    13     evt.set()

    14     try:

    15         conn, addr = serv.accept()

    16     except socket.timeout:

    17         pass

    18     finally:

    19         serv.close()

    20         evt.set()

    21 

    22 class GeneralTests(TestCase):

    23 

    24     def setUp(self):

    25         self.evt = threading.Event()

    26         self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    27         self.sock.settimeout(3)

    28         self.port = test_support.bind_port(self.sock)

    29         threading.Thread(target=server, args=(self.evt,self.sock)).start()

    30         self.evt.wait()

    31         self.evt.clear()

    32         time.sleep(.1)

    33 

    34     def tearDown(self):

    35         self.evt.wait()

    36 

    37     def testBasic(self):

    38         # connects

    39         telnet = telnetlib.Telnet(HOST, self.port)

    40         telnet.sock.close()

    41 

    42     def testTimeoutDefault(self):

    43         self.assertTrue(socket.getdefaulttimeout() is None)

    44         socket.setdefaulttimeout(30)

    45         try:

    46             telnet = telnetlib.Telnet("localhost", self.port)

    47         finally:

    48             socket.setdefaulttimeout(None)

    49         self.assertEqual(telnet.sock.gettimeout(), 30)

    50         telnet.sock.close()

    51 

    52     def testTimeoutNone(self):

    53         # None, having other default

    54         self.assertTrue(socket.getdefaulttimeout() is None)

    55         socket.setdefaulttimeout(30)

    56         try:

    57             telnet = telnetlib.Telnet(HOST, self.port, timeout=None)

    58         finally:

    59             socket.setdefaulttimeout(None)

    60         self.assertTrue(telnet.sock.gettimeout() is None)

    61         telnet.sock.close()

    62 

    63     def testTimeoutValue(self):

    64         telnet = telnetlib.Telnet("localhost", self.port, timeout=30)

    65         self.assertEqual(telnet.sock.gettimeout(), 30)

    66         telnet.sock.close()

    67 

    68     def testTimeoutOpen(self):

    69         telnet = telnetlib.Telnet()

    70         telnet.open("localhost", self.port, timeout=30)

    71         self.assertEqual(telnet.sock.gettimeout(), 30)

    72         telnet.sock.close()

    73 

    74 

    75 def test_main(verbose=None):

    76     test_support.run_unittest(GeneralTests)

    77 

    78 if __name__ == '__main__':

    79     test_main()