mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-10-13 13:34:05 +00:00
77 lines
2.0 KiB
Python
77 lines
2.0 KiB
Python
import pytest
|
|
import asyncio
|
|
|
|
|
|
class MockWebSocket:
|
|
def __init__(self):
|
|
self.received = []
|
|
|
|
def send_str(self, s):
|
|
self.received.append(s)
|
|
|
|
|
|
@asyncio.coroutine
|
|
@pytest.mark.skipif(reason='This test raises a RuntimeError, dunno how to solve it now.')
|
|
def test_dispatcher(loop):
|
|
from bigchaindb.web.websocket_server import Dispatcher, POISON_PILL
|
|
|
|
ws0 = MockWebSocket()
|
|
ws1 = MockWebSocket()
|
|
|
|
event_source = asyncio.Queue(loop=loop)
|
|
dispatcher = Dispatcher(event_source)
|
|
|
|
dispatcher.subscribe(0, ws0)
|
|
dispatcher.subscribe(1, ws1)
|
|
|
|
yield from event_source.put('hack')
|
|
yield from event_source.put('the')
|
|
|
|
yield from event_source.put('planet!')
|
|
yield from event_source.put(POISON_PILL)
|
|
|
|
loop.run_until_complete(dispatcher.publish())
|
|
|
|
assert ws0.received == ['hack', 'the', 'planet!']
|
|
assert ws1.received == ['planet!']
|
|
|
|
|
|
@asyncio.coroutine
|
|
def test_websocket(test_client, loop):
|
|
from bigchaindb.web.websocket_server import init_app, POISON_PILL
|
|
|
|
event_source = asyncio.Queue(loop=loop)
|
|
app = init_app(event_source, loop=loop)
|
|
client = yield from test_client(app)
|
|
ws = yield from client.ws_connect('/')
|
|
|
|
yield from event_source.put('hack')
|
|
yield from event_source.put('the')
|
|
yield from event_source.put('planet!')
|
|
|
|
result = yield from ws.receive()
|
|
assert result.data == 'hack'
|
|
|
|
result = yield from ws.receive()
|
|
assert result.data == 'the'
|
|
|
|
result = yield from ws.receive()
|
|
assert result.data == 'planet!'
|
|
|
|
yield from event_source.put(POISON_PILL)
|
|
|
|
|
|
@asyncio.coroutine
|
|
@pytest.mark.skipif(reason="Still don't understand how to trigger custom errors.")
|
|
def test_websocket_error(test_client, loop):
|
|
from bigchaindb.web.websocket_server import init_app, POISON_PILL
|
|
|
|
event_source = asyncio.Queue(loop=loop)
|
|
app = init_app(event_source, loop=loop)
|
|
client = yield from test_client(app)
|
|
ws = yield from client.ws_connect('/')
|
|
|
|
yield from ws.close()
|
|
|
|
yield from event_source.put(POISON_PILL)
|