def test_keep_alive_error(self):
mock_context = mock.Mock()
mock_flumine = mock.Mock()
mock_flumine.client.betting_client.session_token = None
mock_flumine.client.betting_client.keep_alive.side_effect = BetfairError
worker.keep_alive(mock_context, mock_flumine)
mock_flumine.client.login.assert_called_with()
def test_keep_alive_failure(self):
mock_context = mock.Mock()
mock_flumine = mock.Mock()
mock_flumine.client.betting_client.session_token = None
mock_response = mock.Mock()
mock_response.status = "FAILURE"
mock_flumine.client.betting_client.keep_alive.return_value = mock_response
worker.keep_alive(mock_context, mock_flumine)
mock_flumine.client.login.assert_called_with()
def test_keep_alive(self):
mock_context = mock.Mock()
mock_flumine = mock.Mock()
mock_flumine.client.betting_client.session_token = None
worker.keep_alive(mock_context, mock_flumine)
mock_flumine.client.login.assert_called_with()
mock_flumine.client.betting_client.session_token = 1
mock_flumine.client.betting_client.session_expired = True
worker.keep_alive(mock_context, mock_flumine)
mock_flumine.client.keep_alive.assert_called_with()
def test_keep_alive(self):
mock_context = mock.Mock()
mock_client_simulated = mock.Mock(EXCHANGE=ExchangeType.SIMULATED)
mock_client_betfair = mock.Mock(EXCHANGE=ExchangeType.BETFAIR)
mock_client_betfair.betting_client.session_token = None
mock_client_betfair_st = mock.Mock(EXCHANGE=ExchangeType.BETFAIR)
mock_client_betfair_st.betting_client.session_token = "test"
mock_client_betconnect = mock.Mock(EXCHANGE=ExchangeType.BETCONNECT)
mock_client_betconnect.keep_alive.return_value = None
mock_flumine = mock.Mock(clients=[
mock_client_simulated,
mock_client_betfair,
mock_client_betfair_st,
mock_client_betconnect,
])
worker.keep_alive(mock_context, mock_flumine)
mock_client_simulated.login.assert_called_with()
mock_client_betfair.login.assert_called_with()
mock_client_betfair_st.keep_alive.assert_called_with()
mock_client_betfair_st.login.assert_called_with()
mock_client_betconnect.keep_alive.assert_called_with()
mock_client_betconnect.login.assert_called_with()
