Mirroristas/bigchaindb
2
0
Fork 0
mirror of https://github.com/bigchaindb/bigchaindb.git synced 2024-10-13 13:34:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make utility test function into a fixture

Browse Source
This commit is contained in:
Sylvain Bellemare 2017-04-11 18:32:21 +02:00
parent e0e997755e
commit 98e52e047e
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/web/test_websocket_server.py
View File

@ -8,7 +8,9 @@ import pytest
from bigchaindb.models import Transaction
def create_block(b, total=1):
@pytest.fixture
def _block(b, request):
total = getattr(request, 'param', 1)
transactions = [
Transaction.create(
[b.me],
@ -154,7 +156,8 @@ def test_websocket_string_event(test_client, loop):
@asyncio.coroutine
def test_websocket_block_event(b, test_client, loop):
@pytest.mark.parametrize('_block', (10,), indirect=('_block',), ids=('block',))
def test_websocket_block_event(b, _block, test_client, loop):
from bigchaindb import events
from bigchaindb.web.websocket_server import init_app, POISON_PILL, EVENTS_ENDPOINT
@ -162,7 +165,7 @@ def test_websocket_block_event(b, test_client, loop):
app = init_app(event_source, loop=loop)
client = yield from test_client(app)
ws = yield from client.ws_connect(EVENTS_ENDPOINT)
block = create_block(b, 10).to_dict()
block = _block.to_dict()
block_event = events.Event(events.EventTypes.BLOCK_VALID, block)
yield from event_source.put(block_event)