Tests: removed keepalive_interval for websocket tests.

Also increased read_timeout in frame_read() for slow hosts.
This commit is contained in:
Andrey Zelenkov
2019-08-22 21:28:03 +03:00
parent bc57d1d076
commit c47af243b0
2 changed files with 12 additions and 6 deletions

View File

@@ -12,6 +12,14 @@ class TestNodeWebsockets(TestApplicationNode):
def setUp(self): def setUp(self):
super().setUp() super().setUp()
self.assertIn(
'success',
self.conf(
{'http': {'websocket': {'keepalive_interval': 0}}}, 'settings'
),
'clear keepalive_interval',
)
self.skip_alerts.extend( self.skip_alerts.extend(
[ [
r'last message send failed', r'last message send failed',
@@ -1530,14 +1538,13 @@ class TestNodeWebsockets(TestApplicationNode):
self.ws.frame_write(sock, opcode, payload) # frame length is 101 self.ws.frame_write(sock, opcode, payload) # frame length is 101
self.check_close(sock, 1009) # 1009 - CLOSE_TOO_LARGE self.check_close(sock, 1009) # 1009 - CLOSE_TOO_LARGE
@unittest.skip('not yet')
def test_node_websockets_read_timeout(self): def test_node_websockets_read_timeout(self):
self.load('websockets/mirror') self.load('websockets/mirror')
self.assertIn( self.assertIn(
'success', 'success',
self.conf( self.conf(
{'http': {'websocket': {'read_timeout': 1}}}, 'settings' {'http': {'websocket': {'read_timeout': 5}}}, 'settings'
), ),
'configure read_timeout', 'configure read_timeout',
) )
@@ -1551,14 +1558,13 @@ class TestNodeWebsockets(TestApplicationNode):
self.check_close(sock, 1001) # 1001 - CLOSE_GOING_AWAY self.check_close(sock, 1001) # 1001 - CLOSE_GOING_AWAY
@unittest.skip('not yet')
def test_node_websockets_keepalive_interval(self): def test_node_websockets_keepalive_interval(self):
self.load('websockets/mirror') self.load('websockets/mirror')
self.assertIn( self.assertIn(
'success', 'success',
self.conf( self.conf(
{'http': {'websocket': {'keepalive_interval': 1}}}, 'settings' {'http': {'websocket': {'keepalive_interval': 5}}}, 'settings'
), ),
'configure keepalive_interval', 'configure keepalive_interval',
) )

View File

@@ -57,7 +57,7 @@ class TestApplicationWebsocket(TestApplicationProto):
def serialize_close(self, code = 1000, reason = ''): def serialize_close(self, code = 1000, reason = ''):
return struct.pack('!H', code) + reason.encode('utf-8') return struct.pack('!H', code) + reason.encode('utf-8')
def frame_read(self, sock, read_timeout=5): def frame_read(self, sock, read_timeout=10):
def recv_bytes(sock, bytes): def recv_bytes(sock, bytes):
data = b'' data = b''
while select.select([sock], [], [], read_timeout)[0]: while select.select([sock], [], [], read_timeout)[0]:
@@ -204,7 +204,7 @@ class TestApplicationWebsocket(TestApplicationProto):
op_code = self.OP_CONT op_code = self.OP_CONT
pos = end pos = end
def message_read(self, sock, read_timeout=5): def message_read(self, sock, read_timeout=10):
frame = self.frame_read(sock, read_timeout=read_timeout) frame = self.frame_read(sock, read_timeout=read_timeout)
while(not frame['fin']): while(not frame['fin']):