25 lines
988 B
Python
25 lines
988 B
Python
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()
|
|
tasks = Task.query.filter(Task.execute_after <= datetime.now()).filter_by(completed=False).all()
|
|
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, [])
|
|
|