コード例 #1
0
 def test__get_cleared_orders_error(self, mock_events, mock_config):
     mock_flumine = mock.Mock()
     mock_betting_client = mock.Mock()
     mock_betting_client.betting.list_cleared_orders.side_effect = BetfairError
     self.assertFalse(
         worker._get_cleared_orders(mock_flumine, mock_betting_client,
                                    "1.23"))
     mock_betting_client.betting.list_cleared_orders.assert_called_with(
         bet_status="SETTLED",
         market_ids=["1.23"],
         from_record=0,
         customer_strategy_refs=[mock_config.customer_strategy_ref],
     )
コード例 #2
0
    def test__get_cleared_orders(self, mock_events, mock_config):
        mock_flumine = mock.Mock()
        mock_betting_client = mock.Mock()
        mock_cleared_orders = mock.Mock()
        mock_cleared_orders.orders = []
        mock_cleared_orders.more_available = False
        mock_betting_client.betting.list_cleared_orders.return_value = (
            mock_cleared_orders)

        self.assertTrue(
            worker._get_cleared_orders(mock_flumine, mock_betting_client,
                                       "1.23"))
        mock_betting_client.betting.list_cleared_orders.assert_called_with(
            bet_status="SETTLED",
            market_ids=["1.23"],
            from_record=0,
            customer_strategy_refs=[mock_config.customer_strategy_ref],
        )
        mock_flumine.log_control.assert_called_with(
            mock_events.ClearedOrdersEvent())
        mock_flumine.handler_queue.put.assert_called_with(
            mock_events.ClearedOrdersEvent())