def test_complete_callback(self): scheduled_call_request = CallRequest(itinerary_call) schedule_id = self.scheduler.add(scheduled_call_request, SCHEDULE_3_RUNS) run_call_request = scheduled_call_request.call()[0] run_call_report = CallReport.from_call_request(run_call_request) run_call_report.schedule_id = schedule_id scheduler_complete_callback(run_call_request, run_call_report) collection = ScheduledCall.get_collection() scheduled_call = collection.find_one({'_id': ObjectId(schedule_id)}) self.assertNotEqual(scheduled_call['last_run'], None)
def test_complete_callback_missing_schedule(self): scheduled_call_request = CallRequest(itinerary_call) schedule_id = self.scheduler.add(scheduled_call_request, SCHEDULE_3_RUNS) run_call_request = scheduled_call_request.call()[0] run_call_report = CallReport.from_call_request(run_call_request) run_call_report.schedule_id = schedule_id collection = ScheduledCall.get_collection() collection.remove({'_id': ObjectId(schedule_id)}, safe=True) try: scheduler_complete_callback(run_call_request, run_call_report) except: self.fail()