RAHHH
This commit is contained in:
@@ -0,0 +1,25 @@
|
||||
###############################################################################
|
||||
#
|
||||
# The MIT License (MIT)
|
||||
#
|
||||
# Copyright (c) typedef int GmbH
|
||||
#
|
||||
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
# of this software and associated documentation files (the "Software"), to deal
|
||||
# in the Software without restriction, including without limitation the rights
|
||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
# copies of the Software, and to permit persons to whom the Software is
|
||||
# furnished to do so, subject to the following conditions:
|
||||
#
|
||||
# The above copyright notice and this permission notice shall be included in
|
||||
# all copies or substantial portions of the Software.
|
||||
#
|
||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
# THE SOFTWARE.
|
||||
#
|
||||
###############################################################################
|
||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -0,0 +1,319 @@
|
||||
###############################################################################
|
||||
#
|
||||
# The MIT License (MIT)
|
||||
#
|
||||
# Copyright (c) typedef int GmbH
|
||||
#
|
||||
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
# of this software and associated documentation files (the "Software"), to deal
|
||||
# in the Software without restriction, including without limitation the rights
|
||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
# copies of the Software, and to permit persons to whom the Software is
|
||||
# furnished to do so, subject to the following conditions:
|
||||
#
|
||||
# The above copyright notice and this permission notice shall be included in
|
||||
# all copies or substantial portions of the Software.
|
||||
#
|
||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
# THE SOFTWARE.
|
||||
#
|
||||
###############################################################################
|
||||
|
||||
import os
|
||||
import struct
|
||||
|
||||
|
||||
if os.environ.get('USE_TWISTED', False):
|
||||
from twisted.trial import unittest
|
||||
from twisted.internet.address import IPv4Address
|
||||
from twisted.internet.task import Clock
|
||||
|
||||
from autobahn.twisted.websocket import WebSocketServerProtocol
|
||||
from autobahn.twisted.websocket import WebSocketServerFactory
|
||||
from autobahn.twisted.websocket import WebSocketClientProtocol
|
||||
from autobahn.twisted.websocket import WebSocketClientFactory
|
||||
from autobahn.websocket.compress_deflate import PerMessageDeflate
|
||||
from autobahn.testutil import FakeTransport
|
||||
|
||||
from unittest.mock import MagicMock, patch
|
||||
from txaio.testutil import replace_loop
|
||||
|
||||
from base64 import b64decode
|
||||
|
||||
@patch('base64.b64encode')
|
||||
def create_client_frame(b64patch, **kwargs):
|
||||
"""
|
||||
Kind-of hack-y; maybe better to re-factor the Protocol to have a
|
||||
frame-encoder method-call? Anyway, makes a throwaway protocol
|
||||
encode a frame for us, collects the .sendData call and returns
|
||||
the data that would have gone out. Accepts all the kwargs that
|
||||
WebSocketClientProtocol.sendFrame() accepts.
|
||||
"""
|
||||
|
||||
# only real way to inject a "known" secret-key for the headers
|
||||
# to line up... :/
|
||||
b64patch.return_value = b'QIatSt9QkZPyS4QQfdufO8TgkL0='
|
||||
|
||||
factory = WebSocketClientFactory(protocols=['wamp.2.json'])
|
||||
factory.protocol = WebSocketClientProtocol
|
||||
factory.doStart()
|
||||
proto = factory.buildProtocol(IPv4Address('TCP', '127.0.0.9', 65534))
|
||||
proto.transport = FakeTransport()
|
||||
proto.connectionMade()
|
||||
proto.data = mock_handshake_server
|
||||
proto.processHandshake()
|
||||
|
||||
data = []
|
||||
|
||||
def collect(d, *args):
|
||||
data.append(d)
|
||||
proto.sendData = collect
|
||||
|
||||
proto.sendFrame(**kwargs)
|
||||
return b''.join(data)
|
||||
|
||||
# beware the evils of line-endings...
|
||||
mock_handshake_client = b'GET / HTTP/1.1\r\nUser-Agent: AutobahnPython/0.10.2\r\nHost: localhost:80\r\nUpgrade: WebSocket\r\nConnection: Upgrade\r\nPragma: no-cache\r\nCache-Control: no-cache\r\nSec-WebSocket-Key: 6Jid6RgXpH0RVegaNSs/4g==\r\nSec-WebSocket-Protocol: wamp.2.json\r\nSec-WebSocket-Version: 13\r\n\r\n'
|
||||
|
||||
mock_handshake_server = b'HTTP/1.1 101 Switching Protocols\r\nServer: AutobahnPython/0.10.2\r\nX-Powered-By: AutobahnPython/0.10.2\r\nUpgrade: WebSocket\r\nConnection: Upgrade\r\nSec-WebSocket-Protocol: wamp.2.json\r\nSec-WebSocket-Accept: QIatSt9QkZPyS4QQfdufO8TgkL0=\r\n\r\n\x81~\x02\x19[1,"crossbar",{"roles":{"subscriber":{"features":{"publisher_identification":true,"pattern_based_subscription":true,"subscription_revocation":true}},"publisher":{"features":{"publisher_identification":true,"publisher_exclusion":true,"subscriber_blackwhite_listing":true}},"caller":{"features":{"caller_identification":true,"progressive_call_results":true}},"callee":{"features":{"progressive_call_results":true,"pattern_based_registration":true,"registration_revocation":true,"shared_registration":true,"caller_identification":true}}}}]\x18'
|
||||
|
||||
class TestDeflate(unittest.TestCase):
|
||||
|
||||
def test_max_size(self):
|
||||
decoder = PerMessageDeflate(
|
||||
is_server=False,
|
||||
server_no_context_takeover=False,
|
||||
client_no_context_takeover=False,
|
||||
server_max_window_bits=15,
|
||||
client_max_window_bits=15,
|
||||
mem_level=8,
|
||||
max_message_size=10,
|
||||
)
|
||||
|
||||
# 2000 'x' characters compressed
|
||||
compressed_data = b'\xab\xa8\x18\x05\xa3`\x14\x8c\x82Q0\nF\xc1P\x07\x00\xcf@\xa9\xae'
|
||||
|
||||
decoder.start_decompress_message()
|
||||
data = decoder.decompress_message_data(compressed_data)
|
||||
|
||||
# since we set max_message_size, we should only get that
|
||||
# many bytes back.
|
||||
self.assertEqual(data, b"x" * 10)
|
||||
|
||||
def test_no_max_size(self):
|
||||
decoder = PerMessageDeflate(
|
||||
is_server=False,
|
||||
server_no_context_takeover=False,
|
||||
client_no_context_takeover=False,
|
||||
server_max_window_bits=15,
|
||||
client_max_window_bits=15,
|
||||
mem_level=None,
|
||||
)
|
||||
|
||||
# 2000 'x' characters compressed
|
||||
compressed_data = b'\xab\xa8\x18\x05\xa3`\x14\x8c\x82Q0\nF\xc1P\x07\x00\xcf@\xa9\xae'
|
||||
|
||||
decoder.start_decompress_message()
|
||||
data = decoder.decompress_message_data(compressed_data)
|
||||
|
||||
self.assertEqual(data, b"x" * 2000)
|
||||
|
||||
class TestClient(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.factory = WebSocketClientFactory(protocols=['wamp.2.json'])
|
||||
self.factory.protocol = WebSocketClientProtocol
|
||||
self.factory.doStart()
|
||||
|
||||
self.proto = self.factory.buildProtocol(IPv4Address('TCP', '127.0.0.1', 65534))
|
||||
self.transport = FakeTransport()
|
||||
self.proto.transport = self.transport
|
||||
self.proto.connectionMade()
|
||||
|
||||
def tearDown(self):
|
||||
if self.proto.openHandshakeTimeoutCall:
|
||||
self.proto.openHandshakeTimeoutCall.cancel()
|
||||
self.factory.doStop()
|
||||
# not really necessary, but ...
|
||||
del self.factory
|
||||
del self.proto
|
||||
|
||||
def test_missing_reason_raw(self):
|
||||
# we want to hit the "STATE_OPEN" case, so pretend we're there
|
||||
self.proto.echoCloseCodeReason = True
|
||||
self.proto.state = self.proto.STATE_OPEN
|
||||
self.proto.websocket_version = 1
|
||||
|
||||
self.proto.sendCloseFrame = MagicMock()
|
||||
|
||||
self.proto.onCloseFrame(1000, None)
|
||||
|
||||
def test_unclean_timeout_client(self):
|
||||
"""
|
||||
make a delayed call to drop the connection (client-side)
|
||||
"""
|
||||
|
||||
if False:
|
||||
self.proto.factory._log = print
|
||||
|
||||
# get to STATE_OPEN
|
||||
self.proto.websocket_key = b64decode('6Jid6RgXpH0RVegaNSs/4g==')
|
||||
self.proto.data = mock_handshake_server
|
||||
self.proto.processHandshake()
|
||||
self.assertEqual(self.proto.state, WebSocketServerProtocol.STATE_OPEN)
|
||||
self.assertTrue(self.proto.serverConnectionDropTimeout > 0)
|
||||
|
||||
with replace_loop(Clock()) as reactor:
|
||||
# now 'do the test' and transition to CLOSING
|
||||
self.proto.sendCloseFrame()
|
||||
self.proto.onCloseFrame(1000, b"raw reason")
|
||||
|
||||
# check we scheduled a call
|
||||
self.assertEqual(len(reactor.calls), 1)
|
||||
self.assertEqual(reactor.calls[0].func, self.proto.onServerConnectionDropTimeout)
|
||||
self.assertEqual(reactor.calls[0].getTime(), self.proto.serverConnectionDropTimeout)
|
||||
|
||||
# now, advance the clock past the call (and thereby
|
||||
# execute it)
|
||||
reactor.advance(self.proto.closeHandshakeTimeout + 1)
|
||||
|
||||
# we should have called abortConnection
|
||||
self.assertTrue(self.proto.transport.abort_called())
|
||||
# ...too "internal" for an assert?
|
||||
self.assertEqual(self.proto.state, WebSocketServerProtocol.STATE_CLOSED)
|
||||
|
||||
class TestPing(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.factory = WebSocketServerFactory(protocols=['wamp.2.json'])
|
||||
self.factory.protocol = WebSocketServerProtocol
|
||||
self.factory.doStart()
|
||||
|
||||
self.proto = self.factory.buildProtocol(IPv4Address('TCP', '127.0.0.1', 65534))
|
||||
self.transport = MagicMock()
|
||||
self.proto.transport = self.transport
|
||||
self.proto.connectionMade()
|
||||
|
||||
def tearDown(self):
|
||||
if self.proto.openHandshakeTimeoutCall:
|
||||
self.proto.openHandshakeTimeoutCall.cancel()
|
||||
self.factory.doStop()
|
||||
# not really necessary, but ...
|
||||
del self.factory
|
||||
del self.proto
|
||||
|
||||
def test_unclean_timeout(self):
|
||||
"""
|
||||
make a delayed call to drop the connection
|
||||
"""
|
||||
# first we have to drive the protocol to STATE_CLOSING
|
||||
# ... which we achieve by sendCloseFrame after we're in
|
||||
# STATE_OPEN
|
||||
# XXX double-check this is the correct code-path to get here
|
||||
# "normally"?
|
||||
|
||||
# get to STATE_OPEN
|
||||
self.proto.data = mock_handshake_client
|
||||
self.proto.processHandshake()
|
||||
self.assertTrue(self.proto.state == WebSocketServerProtocol.STATE_OPEN)
|
||||
|
||||
with replace_loop(Clock()) as reactor:
|
||||
# now 'do the test' and transition to CLOSING
|
||||
self.proto.sendCloseFrame()
|
||||
|
||||
# check we scheduled a call
|
||||
self.assertEqual(len(reactor.calls), 1)
|
||||
|
||||
# now, advance the clock past the call (and thereby
|
||||
# execute it)
|
||||
reactor.advance(self.proto.closeHandshakeTimeout + 1)
|
||||
|
||||
# we should have called abortConnection
|
||||
self.assertEqual("call.abortConnection()", str(self.proto.transport.method_calls[-1]))
|
||||
self.assertTrue(self.proto.transport.abortConnection.called)
|
||||
# ...too "internal" for an assert?
|
||||
self.assertEqual(self.proto.state, WebSocketServerProtocol.STATE_CLOSED)
|
||||
|
||||
def test_auto_pingpong_timeout(self):
|
||||
"""
|
||||
autoping and autoping-timeout timing
|
||||
"""
|
||||
# options are evaluated in succeedHandshake, called below
|
||||
self.proto.autoPingInterval = 5
|
||||
self.proto.autoPingTimeout = 2
|
||||
|
||||
with replace_loop(Clock()) as reactor:
|
||||
# get to STATE_OPEN
|
||||
self.proto.data = mock_handshake_client
|
||||
self.proto.processHandshake()
|
||||
self.assertTrue(self.proto.state == WebSocketServerProtocol.STATE_OPEN)
|
||||
|
||||
# we should have scheduled an autoPing
|
||||
self.assertEqual(1, len(reactor.calls))
|
||||
|
||||
# advance past first auto-ping timeout
|
||||
reactor.advance(5)
|
||||
|
||||
# first element from args tuple from transport.write()
|
||||
# call is our data
|
||||
self.assertTrue(self.transport.write.called)
|
||||
data = self.transport.write.call_args[0][0]
|
||||
|
||||
_data = bytes([data[0]])
|
||||
|
||||
# the opcode is the lower 7 bits of the first byte.
|
||||
(opcode,) = struct.unpack("B", _data)
|
||||
opcode = opcode & (~0x80)
|
||||
|
||||
# ... and should be "9" for ping
|
||||
self.assertEqual(9, opcode)
|
||||
|
||||
# Because we have autoPingTimeout there should be
|
||||
# another delayed-called created now
|
||||
self.assertEqual(1, len(reactor.calls))
|
||||
self.assertNotEqual(self.proto.state, self.proto.STATE_CLOSED)
|
||||
|
||||
# ...which we'll now cause to trigger, aborting the connection
|
||||
reactor.advance(3)
|
||||
self.assertEqual(self.proto.state, self.proto.STATE_CLOSED)
|
||||
|
||||
def test_auto_ping_got_pong(self):
|
||||
"""
|
||||
auto-ping with correct reply cancels timeout
|
||||
"""
|
||||
# options are evaluated in succeedHandshake, called below
|
||||
self.proto.autoPingInterval = 5
|
||||
self.proto.autoPingTimeout = 2
|
||||
|
||||
with replace_loop(Clock()) as reactor:
|
||||
# get to STATE_OPEN
|
||||
self.proto.data = mock_handshake_client
|
||||
self.proto.processHandshake()
|
||||
self.assertTrue(self.proto.state == WebSocketServerProtocol.STATE_OPEN)
|
||||
|
||||
# we should have scheduled an autoPing
|
||||
self.assertEqual(1, len(reactor.calls))
|
||||
|
||||
# advance past first auto-ping timeout
|
||||
reactor.advance(5)
|
||||
|
||||
# should have an auto-ping timeout scheduled, and we
|
||||
# save it for later (to check it got cancelled)
|
||||
self.assertEqual(1, len(reactor.calls))
|
||||
timeout_call = reactor.calls[0]
|
||||
|
||||
# elsewhere we check that we actually send an opcode-9
|
||||
# message; now we just blindly inject our own reply
|
||||
# with a PONG frame
|
||||
|
||||
frame = create_client_frame(opcode=10, payload=self.proto.autoPingPending)
|
||||
self.proto.data = frame
|
||||
# really needed twice; does header first, then rest
|
||||
self.proto.processData()
|
||||
self.proto.processData()
|
||||
|
||||
# which should have cancelled the call
|
||||
self.assertTrue(timeout_call.cancelled)
|
||||
@@ -0,0 +1,281 @@
|
||||
###############################################################################
|
||||
#
|
||||
# The MIT License (MIT)
|
||||
#
|
||||
# Copyright (c) typedef int GmbH
|
||||
#
|
||||
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
# of this software and associated documentation files (the "Software"), to deal
|
||||
# in the Software without restriction, including without limitation the rights
|
||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
# copies of the Software, and to permit persons to whom the Software is
|
||||
# furnished to do so, subject to the following conditions:
|
||||
#
|
||||
# The above copyright notice and this permission notice shall be included in
|
||||
# all copies or substantial portions of the Software.
|
||||
#
|
||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
# THE SOFTWARE.
|
||||
#
|
||||
###############################################################################
|
||||
|
||||
import os
|
||||
import unittest
|
||||
from hashlib import sha1
|
||||
from base64 import b64encode
|
||||
from unittest.mock import Mock
|
||||
|
||||
import txaio
|
||||
|
||||
from autobahn.websocket.protocol import WebSocketServerProtocol
|
||||
from autobahn.websocket.protocol import WebSocketServerFactory
|
||||
from autobahn.websocket.protocol import WebSocketClientProtocol
|
||||
from autobahn.websocket.protocol import WebSocketClientFactory
|
||||
from autobahn.websocket.protocol import WebSocketProtocol
|
||||
from autobahn.websocket.types import ConnectingRequest
|
||||
from autobahn.testutil import FakeTransport
|
||||
from autobahn.wamp.types import TransportDetails
|
||||
|
||||
|
||||
class WebSocketClientProtocolTests(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
t = FakeTransport()
|
||||
f = WebSocketClientFactory()
|
||||
f.log = txaio.make_logger()
|
||||
p = WebSocketClientProtocol()
|
||||
p.log = txaio.make_logger()
|
||||
p.factory = f
|
||||
p.transport = t
|
||||
p._transport_details = TransportDetails()
|
||||
|
||||
p._connectionMade()
|
||||
p.state = p.STATE_OPEN
|
||||
p.websocket_version = 18
|
||||
|
||||
self.protocol = p
|
||||
self.transport = t
|
||||
|
||||
def tearDown(self):
|
||||
for call in [
|
||||
self.protocol.autoPingPendingCall,
|
||||
self.protocol.autoPingTimeoutCall,
|
||||
self.protocol.openHandshakeTimeoutCall,
|
||||
self.protocol.closeHandshakeTimeoutCall,
|
||||
]:
|
||||
if call is not None:
|
||||
call.cancel()
|
||||
|
||||
def test_auto_ping(self):
|
||||
self.protocol.autoPingInterval = 1
|
||||
self.protocol.websocket_protocols = [Mock()]
|
||||
self.protocol.websocket_extensions = []
|
||||
self.protocol._onOpen = lambda: None
|
||||
self.protocol._wskey = '0' * 24
|
||||
self.protocol.peer = Mock()
|
||||
|
||||
# usually provided by the Twisted or asyncio specific
|
||||
# subclass, but we're testing the parent here...
|
||||
self.protocol._onConnect = Mock()
|
||||
self.protocol._closeConnection = Mock()
|
||||
|
||||
# set up a connection
|
||||
self.protocol._actuallyStartHandshake(
|
||||
ConnectingRequest(
|
||||
host="example.com",
|
||||
port=80,
|
||||
resource="/ws",
|
||||
)
|
||||
)
|
||||
|
||||
key = self.protocol.websocket_key + WebSocketProtocol._WS_MAGIC
|
||||
self.protocol.data = (
|
||||
b"HTTP/1.1 101 Switching Protocols\x0d\x0a"
|
||||
b"Upgrade: websocket\x0d\x0a"
|
||||
b"Connection: upgrade\x0d\x0a"
|
||||
b"Sec-Websocket-Accept: " + b64encode(sha1(key).digest()) + b"\x0d\x0a\x0d\x0a"
|
||||
)
|
||||
self.protocol.processHandshake()
|
||||
|
||||
self.assertTrue(self.protocol.autoPingPendingCall is not None)
|
||||
|
||||
|
||||
class WebSocketServerProtocolTests(unittest.TestCase):
|
||||
"""
|
||||
Tests for autobahn.websocket.protocol.WebSocketProtocol.
|
||||
"""
|
||||
def setUp(self):
|
||||
t = FakeTransport()
|
||||
f = WebSocketServerFactory()
|
||||
f.log = txaio.make_logger()
|
||||
p = WebSocketServerProtocol()
|
||||
p.log = txaio.make_logger()
|
||||
p.factory = f
|
||||
p.transport = t
|
||||
|
||||
p._connectionMade()
|
||||
p.state = p.STATE_OPEN
|
||||
p.websocket_version = 18
|
||||
|
||||
self.protocol = p
|
||||
self.transport = t
|
||||
|
||||
def tearDown(self):
|
||||
for call in [
|
||||
self.protocol.autoPingPendingCall,
|
||||
self.protocol.autoPingTimeoutCall,
|
||||
self.protocol.openHandshakeTimeoutCall,
|
||||
self.protocol.closeHandshakeTimeoutCall,
|
||||
]:
|
||||
if call is not None:
|
||||
call.cancel()
|
||||
|
||||
def test_auto_ping(self):
|
||||
proto = Mock()
|
||||
proto._get_seconds = Mock(return_value=1)
|
||||
self.protocol.autoPingInterval = 1
|
||||
self.protocol.websocket_protocols = [proto]
|
||||
self.protocol.websocket_extensions = []
|
||||
self.protocol._onOpen = lambda: None
|
||||
self.protocol._wskey = '0' * 24
|
||||
self.protocol.succeedHandshake(proto)
|
||||
|
||||
self.assertTrue(self.protocol.autoPingPendingCall is not None)
|
||||
|
||||
def test_sendClose_none(self):
|
||||
"""
|
||||
sendClose with no code or reason works.
|
||||
"""
|
||||
self.protocol.sendClose()
|
||||
|
||||
# We closed properly
|
||||
self.assertEqual(self.transport._written, b"\x88\x00")
|
||||
self.assertEqual(self.protocol.state, self.protocol.STATE_CLOSING)
|
||||
|
||||
def test_sendClose_str_reason(self):
|
||||
"""
|
||||
sendClose with a str reason works.
|
||||
"""
|
||||
self.protocol.sendClose(code=1000, reason="oh no")
|
||||
|
||||
# We closed properly
|
||||
self.assertEqual(self.transport._written[2:], b"\x03\xe8oh no")
|
||||
self.assertEqual(self.protocol.state, self.protocol.STATE_CLOSING)
|
||||
|
||||
def test_sendClose_unicode_reason(self):
|
||||
"""
|
||||
sendClose with a unicode reason works.
|
||||
"""
|
||||
self.protocol.sendClose(code=1000, reason="oh no")
|
||||
|
||||
# We closed properly
|
||||
self.assertEqual(self.transport._written[2:], b"\x03\xe8oh no")
|
||||
self.assertEqual(self.protocol.state, self.protocol.STATE_CLOSING)
|
||||
|
||||
def test_sendClose_toolong(self):
|
||||
"""
|
||||
sendClose with a too-long reason will truncate it.
|
||||
"""
|
||||
self.protocol.sendClose(code=1000, reason="abc" * 1000)
|
||||
|
||||
# We closed properly
|
||||
self.assertEqual(self.transport._written[2:],
|
||||
b"\x03\xe8" + (b"abc" * 41))
|
||||
self.assertEqual(self.protocol.state, self.protocol.STATE_CLOSING)
|
||||
|
||||
def test_sendClose_reason_with_no_code(self):
|
||||
"""
|
||||
Trying to sendClose with a reason but no code will raise an Exception.
|
||||
"""
|
||||
with self.assertRaises(Exception) as e:
|
||||
self.protocol.sendClose(reason="abc")
|
||||
|
||||
self.assertIn("close reason without close code", str(e.exception))
|
||||
|
||||
# We shouldn't have closed
|
||||
self.assertEqual(self.transport._written, b"")
|
||||
self.assertEqual(self.protocol.state, self.protocol.STATE_OPEN)
|
||||
|
||||
def test_sendClose_invalid_code_type(self):
|
||||
"""
|
||||
Trying to sendClose with a non-int code will raise an Exception.
|
||||
"""
|
||||
with self.assertRaises(Exception) as e:
|
||||
self.protocol.sendClose(code="134")
|
||||
|
||||
self.assertIn("invalid type", str(e.exception))
|
||||
|
||||
# We shouldn't have closed
|
||||
self.assertEqual(self.transport._written, b"")
|
||||
self.assertEqual(self.protocol.state, self.protocol.STATE_OPEN)
|
||||
|
||||
def test_sendClose_invalid_code_value(self):
|
||||
"""
|
||||
Trying to sendClose with a non-valid int code will raise an Exception.
|
||||
"""
|
||||
with self.assertRaises(Exception) as e:
|
||||
self.protocol.sendClose(code=10)
|
||||
|
||||
self.assertIn("invalid close code 10", str(e.exception))
|
||||
|
||||
# We shouldn't have closed
|
||||
self.assertEqual(self.transport._written, b"")
|
||||
self.assertEqual(self.protocol.state, self.protocol.STATE_OPEN)
|
||||
|
||||
def test_interpolate_server_status_template(self):
|
||||
from autobahn.websocket.protocol import _SERVER_STATUS_TEMPLATE
|
||||
s = _SERVER_STATUS_TEMPLATE.format(None, '0.0.0')
|
||||
self.assertEqual(type(s), str)
|
||||
self.assertTrue(len(s) > 0)
|
||||
|
||||
|
||||
if os.environ.get('USE_TWISTED', False):
|
||||
class TwistedProtocolTests(unittest.TestCase):
|
||||
"""
|
||||
Tests which require a specific framework's protocol class to work
|
||||
(in this case, using Twisted)
|
||||
"""
|
||||
def setUp(self):
|
||||
from autobahn.twisted.websocket import WebSocketServerProtocol
|
||||
from autobahn.twisted.websocket import WebSocketServerFactory
|
||||
t = FakeTransport()
|
||||
f = WebSocketServerFactory()
|
||||
p = WebSocketServerProtocol()
|
||||
p.factory = f
|
||||
p.transport = t
|
||||
|
||||
p._connectionMade()
|
||||
p.state = p.STATE_OPEN
|
||||
p.websocket_version = 18
|
||||
|
||||
self.protocol = p
|
||||
self.transport = t
|
||||
|
||||
def tearDown(self):
|
||||
for call in [
|
||||
self.protocol.autoPingPendingCall,
|
||||
self.protocol.autoPingTimeoutCall,
|
||||
self.protocol.openHandshakeTimeoutCall,
|
||||
self.protocol.closeHandshakeTimeoutCall,
|
||||
]:
|
||||
if call is not None:
|
||||
call.cancel()
|
||||
|
||||
def test_loseConnection(self):
|
||||
"""
|
||||
If we lose our connection before openHandshakeTimeout fires, it is
|
||||
cleaned up
|
||||
"""
|
||||
# so, I guess a little cheezy, but we depend on the asyncio or
|
||||
# twisted class to call _connectionLost at some point; faking
|
||||
# that here
|
||||
self.protocol._connectionLost(txaio.create_failure(RuntimeError("testing")))
|
||||
self.assertTrue(self.protocol.openHandshakeTimeoutCall is None)
|
||||
|
||||
def test_send_server_status(self):
|
||||
self.protocol.sendServerStatus()
|
||||
@@ -0,0 +1,125 @@
|
||||
###############################################################################
|
||||
#
|
||||
# The MIT License (MIT)
|
||||
#
|
||||
# Copyright (c) typedef int GmbH
|
||||
#
|
||||
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
# of this software and associated documentation files (the "Software"), to deal
|
||||
# in the Software without restriction, including without limitation the rights
|
||||
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
# copies of the Software, and to permit persons to whom the Software is
|
||||
# furnished to do so, subject to the following conditions:
|
||||
#
|
||||
# The above copyright notice and this permission notice shall be included in
|
||||
# all copies or substantial portions of the Software.
|
||||
#
|
||||
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
# THE SOFTWARE.
|
||||
#
|
||||
###############################################################################
|
||||
|
||||
import unittest
|
||||
|
||||
from autobahn.websocket.util import create_url, parse_url
|
||||
|
||||
|
||||
class TestCreateWsUrl(unittest.TestCase):
|
||||
|
||||
def test_create_url01(self):
|
||||
self.assertEqual(create_url("localhost"), "ws://localhost:80/")
|
||||
|
||||
def test_create_url02(self):
|
||||
self.assertEqual(create_url("localhost", port=8090), "ws://localhost:8090/")
|
||||
|
||||
def test_create_url03(self):
|
||||
self.assertEqual(create_url("localhost", path="ws"), "ws://localhost:80/ws")
|
||||
|
||||
def test_create_url04(self):
|
||||
self.assertEqual(create_url("localhost", path="/ws"), "ws://localhost:80/ws")
|
||||
|
||||
def test_create_url05(self):
|
||||
self.assertEqual(create_url("localhost", path="/ws/foobar"), "ws://localhost:80/ws/foobar")
|
||||
|
||||
def test_create_url06(self):
|
||||
self.assertEqual(create_url("localhost", isSecure=True), "wss://localhost:443/")
|
||||
|
||||
def test_create_url07(self):
|
||||
self.assertEqual(create_url("localhost", isSecure=True, port=443), "wss://localhost:443/")
|
||||
|
||||
def test_create_url08(self):
|
||||
self.assertEqual(create_url("localhost", isSecure=True, port=80), "wss://localhost:80/")
|
||||
|
||||
def test_create_url09(self):
|
||||
self.assertEqual(create_url("localhost", isSecure=True, port=9090, path="ws", params={'foo': 'bar'}), "wss://localhost:9090/ws?foo=bar")
|
||||
|
||||
def test_create_url10(self):
|
||||
wsurl = create_url("localhost", isSecure=True, port=9090, path="ws", params={'foo': 'bar', 'moo': 23})
|
||||
self.assertTrue(wsurl == "wss://localhost:9090/ws?foo=bar&moo=23" or wsurl == "wss://localhost:9090/ws?moo=23&foo=bar")
|
||||
|
||||
def test_create_url11(self):
|
||||
self.assertEqual(create_url("127.0.0.1", path="ws"), "ws://127.0.0.1:80/ws")
|
||||
|
||||
def test_create_url12(self):
|
||||
self.assertEqual(create_url("62.146.25.34", path="ws"), "ws://62.146.25.34:80/ws")
|
||||
|
||||
def test_create_url13(self):
|
||||
self.assertEqual(create_url("subsub1.sub1.something.com", path="ws"), "ws://subsub1.sub1.something.com:80/ws")
|
||||
|
||||
def test_create_url14(self):
|
||||
self.assertEqual(create_url("::1", path="ws"), "ws://::1:80/ws")
|
||||
|
||||
def test_create_url15(self):
|
||||
self.assertEqual(create_url("0:0:0:0:0:0:0:1", path="ws"), "ws://0:0:0:0:0:0:0:1:80/ws")
|
||||
|
||||
|
||||
class TestParseWsUrl(unittest.TestCase):
|
||||
|
||||
# parse_url -> (isSecure, host, port, resource, path, params)
|
||||
|
||||
def test_parse_url01(self):
|
||||
self.assertEqual(parse_url("ws://localhost"), (False, 'localhost', 80, '/', '/', {}))
|
||||
|
||||
def test_parse_url02(self):
|
||||
self.assertEqual(parse_url("ws://localhost:80"), (False, 'localhost', 80, '/', '/', {}))
|
||||
|
||||
def test_parse_url03(self):
|
||||
self.assertEqual(parse_url("wss://localhost"), (True, 'localhost', 443, '/', '/', {}))
|
||||
|
||||
def test_parse_url04(self):
|
||||
self.assertEqual(parse_url("wss://localhost:443"), (True, 'localhost', 443, '/', '/', {}))
|
||||
|
||||
def test_parse_url05(self):
|
||||
self.assertEqual(parse_url("wss://localhost/ws"), (True, 'localhost', 443, '/ws', '/ws', {}))
|
||||
|
||||
def test_parse_url06(self):
|
||||
self.assertEqual(parse_url("wss://localhost/ws?foo=bar"), (True, 'localhost', 443, '/ws?foo=bar', '/ws', {'foo': ['bar']}))
|
||||
|
||||
def test_parse_url07(self):
|
||||
self.assertEqual(parse_url("wss://localhost/ws?foo=bar&moo=23"), (True, 'localhost', 443, '/ws?foo=bar&moo=23', '/ws', {'moo': ['23'], 'foo': ['bar']}))
|
||||
|
||||
def test_parse_url08(self):
|
||||
self.assertEqual(parse_url("wss://localhost/ws?foo=bar&moo=23&moo=44"), (True, 'localhost', 443, '/ws?foo=bar&moo=23&moo=44', '/ws', {'moo': ['23', '44'], 'foo': ['bar']}))
|
||||
|
||||
def test_parse_url09(self):
|
||||
self.assertRaises(Exception, parse_url, "http://localhost")
|
||||
|
||||
def test_parse_url10(self):
|
||||
self.assertRaises(Exception, parse_url, "https://localhost")
|
||||
|
||||
def test_parse_url11(self):
|
||||
self.assertRaises(Exception, parse_url, "http://localhost:80")
|
||||
|
||||
def test_parse_url12(self):
|
||||
self.assertRaises(Exception, parse_url, "http://localhost#frag1")
|
||||
|
||||
def test_parse_url13(self):
|
||||
self.assertRaises(Exception, parse_url, "wss://")
|
||||
|
||||
def test_parse_url14(self):
|
||||
self.assertRaises(Exception, parse_url, "ws://")
|
||||
Reference in New Issue
Block a user