<reponame>lmacken/binance-chain-python<filename>test/test_websocket.py
# Copyright 2019, <NAME>, <NAME>, and the binance-chain-python contributors
# SPDX-License-Identifier: MIT
"""
Binance DEX WebSocket Test Suite
"""
import asyncio

import pytest

from binancechain import HTTPClient, WebSocket


def on_error(msg):
    print(f'Error: {msg}')


@pytest.fixture
async def client():
    # If we create fresh websockets too fast it may error?
    await asyncio.sleep(1)
    client = WebSocket(testnet=True)
    yield client
    client.close()


@pytest.fixture
async def symbols():
    symbols = []
    rest = HTTPClient(testnet=True)
    markets = await rest.get_markets()
    for market in markets:
        symbol = f"{market['base_asset_symbol']}_{market['quote_asset_symbol']}"
        symbols.append(symbol)
    yield symbols
    await rest.close()


@pytest.mark.asyncio
async def test_open_close(client):
    """"Open then immediately close"""
    def on_open():
        print('opened')
        client.close()

    await client.start_async(on_open=on_open, on_error=on_error)
    print('closed')


@pytest.mark.asyncio
async def test_trades(client, symbols):
    print(symbols)
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_trades(symbols=symbols, callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'trades'


@pytest.mark.asyncio
async def test_market_diff(client, symbols):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_market_diff(symbols=symbols, callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'marketDiff'


@pytest.mark.asyncio
async def test_market_depth(client, symbols):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_market_depth(symbols=symbols, callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'marketDepth'


@pytest.mark.asyncio
async def test_kline(client, symbols):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_kline(interval='1m', symbols=symbols, callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'kline_1m'


@pytest.mark.asyncio
async def test_tickers(client, symbols):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_ticker(symbols=symbols, callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'ticker'


@pytest.mark.asyncio
async def test_all_tickers(client):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_all_tickers(callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'allTickers'


@pytest.mark.asyncio
async def test_mini_ticker(client, symbols):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_mini_ticker(symbols=symbols, callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'miniTicker'


@pytest.mark.asyncio
async def test_all_mini_ticker(client, symbols):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_all_mini_tickers(callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'allMiniTickers'


@pytest.mark.asyncio
async def test_blockheight(client):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_blockheight(callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert 'stream' in result


@pytest.mark.asyncio
async def test_keepalive(client):
    def on_open():
        client.keepalive()
        client.close()
    await client.start_async(on_open=on_open, on_error=on_error)


@pytest.mark.asyncio
async def test_unsubscribe(client):
    results = []

    def callback(msg):
        results.append(msg)
        client.unsubscribe("blockheight")
        client.close()

    def on_open():
        client.subscribe_blockheight(callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)
    assert results


@pytest.mark.asyncio
async def test_decorator(client):
    @client.on('open')
    def callback():
        client.close()
    await client.start_async()


@pytest.mark.asyncio
async def test_decorator_async(client):
    @client.on('open')
    async def callback():
        client.close()
    await client.start_async()


@pytest.mark.asyncio
async def test_decorator_sub_queue(client):
    results = []

    @client.on("allTickers", symbols=["$all"])
    async def callback(msg):
        results.append(msg)
        client.close()

    await client.start_async()
    assert results

# Copyright 2019, <NAME>, <NAME>, and the binance-chain-python contributors
# SPDX-License-Identifier: MIT
"""
Binance DEX WebSocket Test Suite
"""
import asyncio

import pytest

from binancechain import HTTPClient, WebSocket


def on_error(msg):
    print(f'Error: {msg}')


@pytest.fixture
async def client():
    # If we create fresh websockets too fast it may error?
    await asyncio.sleep(1)
    client = WebSocket(testnet=True)
    yield client
    client.close()


@pytest.fixture
async def symbols():
    symbols = []
    rest = HTTPClient(testnet=True)
    markets = await rest.get_markets()
    for market in markets:
        symbol = f"{market['base_asset_symbol']}_{market['quote_asset_symbol']}"
        symbols.append(symbol)
    yield symbols
    await rest.close()


@pytest.mark.asyncio
async def test_open_close(client):
    """"Open then immediately close"""
    def on_open():
        print('opened')
        client.close()

    await client.start_async(on_open=on_open, on_error=on_error)
    print('closed')


@pytest.mark.asyncio
async def test_trades(client, symbols):
    print(symbols)
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_trades(symbols=symbols, callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'trades'


@pytest.mark.asyncio
async def test_market_diff(client, symbols):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_market_diff(symbols=symbols, callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'marketDiff'


@pytest.mark.asyncio
async def test_market_depth(client, symbols):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_market_depth(symbols=symbols, callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'marketDepth'


@pytest.mark.asyncio
async def test_kline(client, symbols):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_kline(interval='1m', symbols=symbols, callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'kline_1m'


@pytest.mark.asyncio
async def test_tickers(client, symbols):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_ticker(symbols=symbols, callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'ticker'


@pytest.mark.asyncio
async def test_all_tickers(client):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_all_tickers(callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'allTickers'


@pytest.mark.asyncio
async def test_mini_ticker(client, symbols):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_mini_ticker(symbols=symbols, callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'miniTicker'


@pytest.mark.asyncio
async def test_all_mini_ticker(client, symbols):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_all_mini_tickers(callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert result['stream'] == 'allMiniTickers'


@pytest.mark.asyncio
async def test_blockheight(client):
    results = []

    def callback(msg):
        results.append(msg)
        client.close()

    def on_open():
        client.subscribe_blockheight(callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)

    result = results[0]
    assert 'stream' in result


@pytest.mark.asyncio
async def test_keepalive(client):
    def on_open():
        client.keepalive()
        client.close()
    await client.start_async(on_open=on_open, on_error=on_error)


@pytest.mark.asyncio
async def test_unsubscribe(client):
    results = []

    def callback(msg):
        results.append(msg)
        client.unsubscribe("blockheight")
        client.close()

    def on_open():
        client.subscribe_blockheight(callback=callback)

    await client.start_async(on_open=on_open, on_error=on_error)
    assert results


@pytest.mark.asyncio
async def test_decorator(client):
    @client.on('open')
    def callback():
        client.close()
    await client.start_async()


@pytest.mark.asyncio
async def test_decorator_async(client):
    @client.on('open')
    async def callback():
        client.close()
    await client.start_async()


@pytest.mark.asyncio
async def test_decorator_sub_queue(client):
    results = []

    @client.on("allTickers", symbols=["$all"])
    async def callback(msg):
        results.append(msg)
        client.close()

    await client.start_async()
    assert results
# Copyright 2019, <NAME>, <NAME>, and the binance-chain-python contributors # SPDX-License-Identifier: MIT Binance DEX WebSocket Test Suite # If we create fresh websockets too fast it may error? "Open then immediately close