def test_batch_update(self): klass = self._get_target_class() from kardboard.tasks import update_daily_records update_daily_records.apply(args=[7], throw=True) self.assertEqual(7, klass.objects.count()) # update_daily_records should be idempotent update_daily_records.apply(args=[7], throw=True) self.assertEqual(7, klass.objects.count())
def main(): DailyRecord._get_collection().drop_indexes() oldest_card = Kard.objects.all().order_by('+backlog_date')[0] start_date = make_start_date(date=oldest_card.backlog_date) end_date = make_end_date(date=datetime.datetime.now()) print "Daily records: %s" % DailyRecord.objects.count() print "Creating daily records" print "%s --> %s" % (start_date, end_date) days = end_date - start_date print "Going back %s days" % days.days r = update_daily_records.apply(args=(days.days,)) r.get() print "DONE!" print "Daily records: %s" % DailyRecord.objects.count()