Beispiel #1
0
    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)
Beispiel #2
0
    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()