def test_get_percent_complete_cache(self):
stack = helpers.create_stack()
d = Dispatcher.find_by_id_anon(stack['dispatcher_id'])
d.expire_counts()
redis.set('dispatcher_%s_percent' % d.id, 15)
self.assertEqual(d.get_percent_complete(), 15)
redis.delete('dispatcher_%s_percent' % d.id)
def test_dispatcher_send(self):
stack = helpers.create_stack()
d = Dispatcher.find_by_id_anon(stack['dispatcher_id'])
disp_count = redis.llen("dispatcher")
d.send()
self.assertEqual(redis.llen("dispatcher"), disp_count + 1)
redis.delete("dispatcher")
def test_get_percent_complete_not_ready(self):
stack = helpers.create_stack()
d = Dispatcher.find_by_id_anon(stack['dispatcher_id'])
d.expire_counts()
[d.incr_sent() for _ in range(5)]
[d.incr_skipped() for _ in range(2)]
self.assertEqual(d.get_percent_complete(), 0)
redis.delete('dispatcher_%s_percent' % d.id)
def test_get_percent_complete_under_90(self):
stack = helpers.create_stack()
d = Dispatcher.find_by_id_anon(stack['dispatcher_id'])
d.expire_counts()
[d.incr_queued() for _ in range(10)]
[d.incr_sent() for _ in range(5)]
[d.incr_skipped() for _ in range(2)]
self.assertEqual(d.get_percent_complete(), 70)
self.assertEqual(redis.exists('dispatcher_%s_percent' % d.id), True)
d.expire_counts()
redis.delete('dispatcher_%s_percent' % d.id)
def test_process_skip_bad_state(self):
stack = helpers.create_stack()
send = Send.query.first()
redis.delete('dispatcher_%s_sent' % send.dispatcher_id)
send.state = -1
send.process()
self.assertEqual(send.attempts, 1)
self.assertEqual(send.dispatcher.get_sent(), '1')
send.process()
self.assertEqual(send.attempts, 2)
self.assertEqual(send.dispatcher.get_sent(), '1')
redis.delete('dispatcher_%s_sent' % send.dispatcher_id)
def test_dispatcher_queue_emails(self):
stack = helpers.create_stack()
d = Dispatcher.find_by_id_anon(stack['dispatcher_id'])
d.expire_counts()
d.prep_data()
count = len(d.adjusted_data)
mail_count = redis.llen("mail")
d.queue_emails()
self.assertEqual(d.state, 3)
self.assertEqual(redis.llen("mail"), mail_count + count)
self.assertEqual(d.get_queued(), str(count))
self.assertEqual(d.queued_count, count)
redis.delete("dispatcher")
def test_dispatcher_prep_data(self):
stack = helpers.create_stack()
d = Dispatcher.find_by_id_anon(stack['dispatcher_id'])
orig_data = d.import_data
disp_count = redis.llen("dispatcher")
d.prep_data()
self.assertEqual(d.state, 1)
self.assertEqual(redis.llen("dispatcher"), disp_count + 1)
self.assertEqual(d.import_data, orig_data)
self.assertEqual(d.adjusted_data[0].keys(), ['name', 'hash_id'])
self.assertEqual(len(d.adjusted_data[1].get('hash_id')), 64)
self.assertEqual(len(d.adjusted_data), len(orig_data))
redis.delete("dispatcher")
def test_get_percent_complete_done(self):
stack = helpers.create_stack()
d = Dispatcher.find_by_id_anon(stack['dispatcher_id'])
d.update(queued_count=10)
d.expire_counts()
[d.incr_queued() for _ in range(10)]
[d.incr_sent() for _ in range(6)]
[d.incr_skipped() for _ in range(4)]
self.assertEqual(d.get_percent_complete(), 100)
self.assertEqual(d.percent_complete, 100)
self.assertEqual(d.state, 10)
self.assertEqual(d.sent_count, 6)
self.assertEqual(d.skipped_count, 4)
redis.delete('dispatcher_%s_percent' % d.id)
def expire_counts(self, timed=None):
if timed:
redis.expire('dispatcher_%s_percent' % self.id, timed)
redis.expire('dispatcher_%s_queued' % self.id, timed)
redis.expire('dispatcher_%s_sent' % self.id, timed)
redis.expire('dispatcher_%s_skipped' % self.id, timed)
else:
redis.delete('dispatcher_%s_percent' % self.id)
redis.delete('dispatcher_%s_queued' % self.id)
redis.delete('dispatcher_%s_sent' % self.id)
redis.delete('dispatcher_%s_skipped' % self.id)
def del_cache(key):
if check_redis():
return redis.delete(key)
def test_campaign_selector_send_count(self):
stack = helpers.create_stack()
c = Campaign.find_by_id_anon(stack['campaign_id'])
redis.delete('selector_send_count_%s' % c.id)
self.assertEqual(c.selector_send_count(),1)
