def test_run_tasks(self): ma = PeriodicTaskAdmin(PeriodicTask, self.site) self.request = self.patch_request(self.request_factory.get('/')) ma.run_tasks(self.request, PeriodicTask.objects.all()) assert len(self.request._messages._queued_messages) == 1 queued_message = self.request._messages._queued_messages[0].message assert queued_message == '2 tasks were successfully run'
def test_toggle_action_all_disabled(self): PeriodicTask.objects.create(name='name1', task='task1', enabled=False) PeriodicTask.objects.create(name='name2', task='task2', enabled=False) PeriodicTask.objects.create(name='name3', task='task3', enabled=False) qs = PeriodicTask.objects.all() PeriodicTaskAdmin(PeriodicTask, None)._toggle_tasks_activity(qs) e1 = PeriodicTask.objects.get(name='name1', task='task1').enabled e2 = PeriodicTask.objects.get(name='name2', task='task2').enabled e3 = PeriodicTask.objects.get(name='name3', task='task3').enabled self.assertTrue(e1) self.assertTrue(e2) self.assertTrue(e3)