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}')
