2019-01-29 14:21:06 -08:00
|
|
|
from datetime import datetime
|
|
|
|
|
|
|
|
from grant.task.models import Task
|
|
|
|
|
|
|
|
from ..config import BaseProposalCreatorConfig
|
|
|
|
|
|
|
|
|
|
|
|
class TestTaskAPI(BaseProposalCreatorConfig):
|
|
|
|
|
|
|
|
def test_proposal_reminder_task_is_created(self):
|
|
|
|
tasks = Task.query.filter(Task.execute_after <= datetime.now()).filter_by(completed=False).all()
|
|
|
|
self.assertEqual(tasks, [])
|
|
|
|
self.make_proposal_reminder_task()
|
2019-01-29 14:43:04 -08:00
|
|
|
tasks = Task.query.filter(Task.execute_after <= datetime.now()).filter_by(completed=False).all()
|
2019-01-29 14:21:06 -08:00
|
|
|
self.assertEqual(len(tasks), 1)
|
|
|
|
|
|
|
|
def test_proposal_reminder_task_is_marked_completed_after_call(self):
|
|
|
|
self.make_proposal_reminder_task()
|
|
|
|
tasks = Task.query.filter(Task.execute_after <= datetime.now()).filter_by(completed=False).all()
|
|
|
|
self.assertEqual(len(tasks), 1)
|
|
|
|
self.app.get("/api/v1/task")
|
|
|
|
tasks = Task.query.filter(Task.execute_after <= datetime.now()).filter_by(completed=False).all()
|
|
|
|
self.assertEqual(tasks, [])
|
|
|
|
|