def runTest():
blocked = False
if account.trading_blocked:
blocked = True
if blocked == False:
print("running")
connection = StreamConn(API_KEY,
API_SECRET,
base_url=BASE_URL,
data_url="wss://data.alpaca.markets",
data_stream="alpacadatav1")
@connection.on(r'^AM.AAPL$')
async def tradeInfo(connection, channels, data):
symbol = data.symbol
print("close: ", data.close)
print("open: ", data.open)
print("low: ", data.low)
print("high: ", data.high)
if data.close > data.open and data.open - data.low > 0.01:
print("buying 1")
api.submit_order(symbol, 1, "buy", "market", "day")
print("bought")
print("remaining: ", account.buying_power)
#sell and take profit later
connection.run(["AM.AAPL"])
def runTest():
blocked = False
if account.trading_blocked:
blocked = True
if blocked == False:
print("running")
connection = StreamConn(API_KEY,
API_SECRET,
base_url=BASE_URL,
data_url="wss://data.alpaca.markets",
data_stream="alpacadatav1")
print("connecting...")
@connection.on(r'^AM.AAPL$')
async def tradeInfo(connection, channels, data):
symbol = data.symbol
print("close: ", data.close)
print("open: ", data.open)
print("low: ", data.low)
print("high: ", data.high, "\n")
now = datetime.now()
now = now.strftime("%H:%M:%S")
trades = {
"ref": now,
"symbol": symbol,
"close": data.close,
"open": data.open,
"low": data.low,
"high": data.high,
"funds": account.buying_power
}
with open("store.json") as reserve:
store = json.load(reserve)
temp = store["tradeInfo"]
temp.append(trades)
writeJSON(store, file="store.json")
# sell and take profit later
connection.run(["AM.AAPL"])
def test_stream(websockets):
# _connect
connect = AsyncMock()
websockets.connect = connect
ws = connect.mock()
ws.send = AsyncMock()
ws.recv = AsyncMock(return_value=json.dumps({
'stream': 'authentication',
'data': {
'status': 'authenticated',
}
}).encode())
conn = StreamConn('key-id', 'secret-key')
conn._consume_msg = AsyncMock()
@conn.on('authenticated')
async def on_auth(conn, stream, msg):
on_auth.msg = msg
_run(conn._connect())
assert on_auth.msg.status == 'authenticated'
assert conn._consume_msg.mock.called
conn.deregister('authenticated')
assert len(conn._handlers) == 0
with pytest.raises(ValueError):
conn.register('nonasync', lambda x: x)
# _consume_msg
conn = StreamConn('key-id', 'secret-key')
ws = mock.Mock()
conn._ws = ws
ws.recv = AsyncMock(return_value=json.dumps({
'stream': 'raise',
'data': {
'key': 'value',
}
}))
ws.close = AsyncMock()
class TestException(Exception):
pass
@conn.on('raise')
async def on_raise(conn, stream, msg):
raise TestException()
with pytest.raises(TestException):
_run(conn._consume_msg())
assert ws.close.mock.called
# _ensure_nats
conn = StreamConn('key-id', 'secret-key')
with mock.patch('alpaca_trade_api.stream2.polygon') as polygon:
polygon.Stream().connect = AsyncMock()
_run(conn._ensure_nats())
assert conn.polygon is not None
assert conn.polygon.connect.mock.called
# _ensure_ws
conn = StreamConn('key-id', 'secret-key')
conn._connect = AsyncMock()
_run(conn._ensure_ws())
assert conn._connect.mock.called
assert conn._ws is not None
# subscribe
conn = StreamConn('key-id', 'secret-key')
conn._ensure_ws = AsyncMock()
conn._ws = mock.Mock()
conn._ws.send = AsyncMock()
conn._ensure_nats = AsyncMock()
conn.polygon = mock.Mock()
conn.polygon.subscribe = AsyncMock()
_run(conn.subscribe(['Q.*', 'account_updates']))
assert conn._ws.send.mock.called
assert conn.polygon.subscribe.mock.called
# close
conn = StreamConn('key-id', 'secret-key')
conn._ws = mock.Mock()
conn._ws.close = AsyncMock()
conn.polygon = mock.Mock()
conn.polygon.close = AsyncMock()
_run(conn.close())
assert conn._ws.close.mock.called
assert conn.polygon.close.mock.called
# _cast
conn = StreamConn('key-id', 'secret-key')
ent = conn._cast('account_updates', {})
assert isinstance(ent, Account)
ent = conn._cast('other', {'key': 'value'})
assert ent.key == 'value'
# _dispatch_nats
conn = StreamConn('key-id', 'secret-key')
@conn.on('^Q.')
async def on_q(conn, subject, data):
on_q.data = data
_run(conn._dispatch_nats(conn, 'Q.SPY', PolyEntity({'key': 'value'})))
assert on_q.data.key == 'value'