def test_useful_reraise_in_flush_endpoint(self): error_mock = Mock() error_mock.data = six.b('{"status": 0, "error": "arbitrary error"}') broken_json = '{broken JSON' consumer = mixpanel.BufferedConsumer(2) with patch('mixpanel.urllib3.PoolManager.request', return_value=error_mock): consumer.send('events', broken_json) with pytest.raises(mixpanel.MixpanelException) as excinfo: consumer.flush() assert excinfo.value.message == '[%s]' % broken_json assert excinfo.value.endpoint == 'events'
def __init__(self, botengine): """ :param token: :param request_timeout: """ import domain import mixpanel # Mixpanel Object self.mp = mixpanel.Mixpanel( domain.MIXPANEL_TOKEN, consumer=mixpanel.BufferedConsumer( request_timeout=MIXPANEL_HTTP_TIMEOUT_S)) # Total number of events tracked on this execution, never saved so always 0 on the next execution self.temp_total_tracked = 0 self._sync_user(botengine)
def setUp(self): self.MAX_LENGTH = 10 self.consumer = mixpanel.BufferedConsumer(self.MAX_LENGTH) self.mock = Mock() self.mock.read.return_value = '{"status":1, "error": null}'
def setup_class(cls): cls.MAX_LENGTH = 10 cls.consumer = mixpanel.BufferedConsumer(cls.MAX_LENGTH) cls.consumer._consumer = LogConsumer() cls.log = cls.consumer._consumer.log
def setup_class(cls): cls.MAX_LENGTH = 10 cls.consumer = mixpanel.BufferedConsumer(cls.MAX_LENGTH) cls.mock = Mock() cls.mock.read.return_value = six.b('{"status":1, "error": null}')