def test_subscribe_to_channel(self, mocker):
socket = BitMEXWebsocket()
message = {
"success": "true",
"subscribe": "instrument:XBTH17",
"request": {
"op": "subscribe",
"args": ["instrument:XBTH17"]
}
}
handler = mocker.stub()
socket.on('subscribe', handler)
socket.on_message({}, json.dumps(message))
handler.assert_called_once()
def test_on_subscribe_success(self, mocker):
error = mocker.patch('bitmex_websocket.BitMEXWebsocket.on_error')
socket = BitMEXWebsocket()
message = {
"success": "true",
"subscribe": "instrument:XBTH17",
"request": {
"op": "subscribe",
"args": ["instrument:XBTH17"]
}
}
subscribe_handler = mocker.stub()
socket.on('subscribe', subscribe_handler)
socket.emit('subscribe', message)
error.assert_not_called()
subscribe_handler.assert_called_once_with(message)
