def test_task_error_repr(self): pr = ReportStore(status=celery.states.PENDING) self.session.add(pr) self.session.commit() te = TaskErrorStore(task_type='report', task_id=pr.id, message='m', traceback='t', count=1) self.session.add(te) self.session.commit() row = self.session.query(TaskErrorStore).first() assert_true(str(row).find('TaskErrorStore') >= 0)
def test_task_error(self): pr = ReportStore(status=celery.states.PENDING) self.session.add(pr) self.session.commit() te = TaskErrorStore(task_type='report', task_id=pr.id, message='m', traceback='t', count=1) self.session.add(te) self.session.commit() row = self.session.query(TaskErrorStore).first() assert_equal(row.task_type, 'report') assert_equal(row.task_id, pr.id) assert_equal(row.message, 'm') assert_equal(row.traceback, 't') assert_equal(row.count, 1)
def test_add_existing(self): # If the failing report has previous errors, # the existing task error should be updated. t1 = datetime.now() te = TaskErrorStore(task_type='report', task_id=self.report.id, count=1, timestamp=t1, message='message', traceback='traceback') self.session.add(te) self.session.commit() TaskErrorStore.add('report', self.report.id, 'message2', 'traceback2') t2 = datetime.now() row = self.session.query(TaskErrorStore).first() print t1, row.timestamp, t2 assert_equal(row.task_type, 'report') assert_equal(row.task_id, self.report.id) assert_true(row.timestamp > t1 and row.timestamp < t2) assert_equal(row.message, 'message2') assert_equal(row.traceback, 'traceback2') assert_equal(row.count, 2)