- fixed #77
- I don't have IPv6 enviroment, now. So, I didn't test yet. - fixed some test warning.
This commit is contained in:
@@ -459,10 +459,9 @@ class SockOptTest(unittest.TestCase):
|
||||
@unittest.skipUnless(TEST_WITH_INTERNET, "Internet-requiring tests are disabled")
|
||||
def testSockOpt(self):
|
||||
sockopt = ((socket.IPPROTO_TCP, socket.TCP_NODELAY, 1),)
|
||||
s = ws.WebSocket(sockopt=sockopt)
|
||||
self.assertNotEqual(s.sock.getsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY), 0)
|
||||
s = ws.create_connection("ws://echo.websocket.org", sockopt=sockopt)
|
||||
self.assertNotEqual(s.sock.getsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY), 0)
|
||||
s.close()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
@@ -418,11 +418,9 @@ class WebSocket(object):
|
||||
if sslopt is None:
|
||||
sslopt = {}
|
||||
self.connected = False
|
||||
self.sock = socket.socket()
|
||||
for opts in DEFAULT_SOCKET_OPTION:
|
||||
self.sock.setsockopt(*opts)
|
||||
for opts in sockopt:
|
||||
self.sock.setsockopt(*opts)
|
||||
self.sock = None
|
||||
self._timeout = None
|
||||
self.sockopt = sockopt
|
||||
self.sslopt = sslopt
|
||||
self.get_mask_key = get_mask_key
|
||||
self.fire_cont_frame = fire_cont_frame
|
||||
@@ -454,7 +452,7 @@ class WebSocket(object):
|
||||
"""
|
||||
Get the websocket timeout(second).
|
||||
"""
|
||||
return self.sock.gettimeout()
|
||||
return self._timeout
|
||||
|
||||
def settimeout(self, timeout):
|
||||
"""
|
||||
@@ -462,7 +460,7 @@ class WebSocket(object):
|
||||
|
||||
timeout: timeout time(second).
|
||||
"""
|
||||
self.sock.settimeout(timeout)
|
||||
self._timeout = timeout
|
||||
|
||||
timeout = property(gettimeout, settimeout)
|
||||
|
||||
@@ -487,8 +485,21 @@ class WebSocket(object):
|
||||
|
||||
"""
|
||||
hostname, port, resource, is_secure = _parse_url(url)
|
||||
addrinfo_list = socket.getaddrinfo(hostname, 80, 0, 0, socket.SOL_TCP)
|
||||
if not addrinfo_list:
|
||||
raise WebSocketException("Host not found.: " + hostname + ":" + str(port))
|
||||
|
||||
family = addrinfo_list[0][0]
|
||||
self.sock = socket.socket(family)
|
||||
self.sock.settimeout(self.timeout)
|
||||
for opts in DEFAULT_SOCKET_OPTION:
|
||||
self.sock.setsockopt(*opts)
|
||||
for opts in self.sockopt:
|
||||
self.sock.setsockopt(*opts)
|
||||
# TODO: we need to support proxy
|
||||
self.sock.connect((hostname, port))
|
||||
address = addrinfo_list[0][4]
|
||||
self.sock.connect(address)
|
||||
|
||||
if is_secure:
|
||||
if HAVE_SSL:
|
||||
sslopt = dict(cert_reqs=ssl.CERT_REQUIRED,
|
||||
|
Reference in New Issue
Block a user