mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-27 01:05:15 +00:00
59 lines
1.3 KiB
Python
59 lines
1.3 KiB
Python
|
from concurrent import futures
|
||
|
|
|
||
|
|
import grpc
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from bbgo import BBGO
|
||
|
|
from bbgo import bbgo_pb2
|
||
|
|
from bbgo import bbgo_pb2_grpc
|
||
|
|
from tests.servicer import TestServicer
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def grpc_address(host='[::]', port=50051):
|
||
|
|
return f'{host}:{port}'
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def bbgo(host='[::]', port=50051):
|
||
|
|
return BBGO(host, port)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def grpc_channel(grpc_address):
|
||
|
|
return grpc.insecure_channel(grpc_address)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def grpc_server(grpc_address, max_workers=1):
|
||
|
|
server = grpc.server(futures.ThreadPoolExecutor(max_workers))
|
||
|
|
servicer = TestServicer()
|
||
|
|
bbgo_pb2_grpc.add_BBGOServicer_to_server(servicer, server)
|
||
|
|
server.add_insecure_port(grpc_address)
|
||
|
|
server.start()
|
||
|
|
yield server
|
||
|
|
server.stop(grace=None)
|
||
|
|
|
||
|
|
|
||
|
|
def test_submit_order(bbgo, grpc_server):
|
||
|
|
exchange = 'max'
|
||
|
|
symbol = 'BTCUSDT'
|
||
|
|
side = bbgo_pb2.Side.BUY
|
||
|
|
quantity = 0.01
|
||
|
|
order_type = bbgo_pb2.OrderType.LIMIT
|
||
|
|
|
||
|
|
response = bbgo.submit_order(
|
||
|
|
exchange=exchange,
|
||
|
|
symbol=symbol,
|
||
|
|
side=side,
|
||
|
|
quantity=quantity,
|
||
|
|
order_type=order_type,
|
||
|
|
)
|
||
|
|
|
||
|
|
order = response.order
|
||
|
|
|
||
|
|
assert order.exchange == exchange
|
||
|
|
assert order.symbol == symbol
|
||
|
|
assert order.side == side
|
||
|
|
assert order.quantity == quantity
|