45 lines
957 B
Python
45 lines
957 B
Python
# encoding: UTF-8
|
|
import unittest
|
|
|
|
from Promise import Promise
|
|
from vnpy.api.websocket import WebsocketClient
|
|
|
|
|
|
class TestWebsocketClient(WebsocketClient):
|
|
|
|
def __init__(self):
|
|
host = 'wss://echo.websocket.org'
|
|
super(TestWebsocketClient, self).__init__()
|
|
self.init(host)
|
|
self.p = Promise()
|
|
|
|
def onMessage(self, packet):
|
|
self.p.set_result(packet)
|
|
pass
|
|
|
|
def onConnect(self):
|
|
pass
|
|
|
|
def onError(self, exceptionType, exceptionValue, tb):
|
|
self.p.set_exception(exceptionValue)
|
|
pass
|
|
|
|
|
|
class WebsocketClientTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.c = TestWebsocketClient()
|
|
self.c.start()
|
|
|
|
def tearDown(self):
|
|
self.c.stop()
|
|
|
|
def test_sendReq(self):
|
|
req = {
|
|
'name': 'val'
|
|
}
|
|
self.c.sendRequest(req)
|
|
res = self.c.p.get(3)
|
|
|
|
self.assertEqual(res, req)
|