Python MockWebsocketConnection示例

示例#1

文件： test_subscribe_candlestick_event.py 项目： shede333/huobi-client

    def test_request(self):
        impl = WebsocketRequestImpl("")
        symbols = list()
        symbols.append("btcusdt")

        def callback(event):
            pass

        request = impl.subscribe_candlestick_event(symbols, CandlestickInterval.MIN1, callback)
        mock_connection = MockWebsocketConnection(request)
        request.subscription_handler(mock_connection)
        subscription = mock_connection.pop_output_message()
        self.assertTrue(subscription.find("market.btcusdt.kline.1min") != -1)

示例#2

文件： test_subscribe_trade_event.py 项目： SingletC/huobi_bot

    def test_request(self):
        impl = WebsocketRequestImpl("")
        symbols = list()
        symbols.append("btcusdt")

        def callback(event):
            pass

        request = impl.subscribe_trade_event(symbols, callback)
        mock_connection = MockWebsocketConnection(request)
        request.subscription_handler(mock_connection)
        subscription = mock_connection.pop_output_message()
        self.assertTrue(subscription.find("market.btcusdt.trade.detail") != -1)

示例#3

文件： test_subscribe_account_event.py 项目： SingletC/huobi_bot

    def test_request(self):
        impl = WebsocketRequestImpl("12345")
        symbols = list()
        symbols.append("btcusdt")

        def callback(event):
            pass

        request = impl.subscribe_account_event(BalanceMode.AVAILABLE, callback)
        mock_connection = MockWebsocketConnection(request)
        request.subscription_handler(mock_connection)
        subscription = mock_connection.pop_output_message()
        self.assertTrue(subscription.find("accounts") != -1)