coins-demo/todo/tests/test_tracker.py

import pytest

from django.core import mail

from todo.models import Task, Comment
from todo.mail.consumers import tracker_consumer
from email.message import EmailMessage


def consumer(*args, title_format="[TEST] {subject}", **kwargs):
    return tracker_consumer(
        group="Workgroup One", task_list_slug="zip", priority=1, task_title_format=title_format
    )(*args, **kwargs)


def make_message(subject, content):
    msg = EmailMessage()
    msg.set_content(content)
    msg["Subject"] = subject
    return msg


def test_tracker_task_creation(todo_setup, django_user_model):
    msg = make_message("test1 subject", "test1 content")
    msg["From"] = "test1@example.com"
    msg["Message-ID"] = "<a@example.com>"

    # test task creation
    task_count = Task.objects.count()
    consumer([msg])

    assert task_count + 1 == Task.objects.count(), "task wasn't created"
    task = Task.objects.filter(title="[TEST] test1 subject").first()
    assert task is not None, "task was created with the wrong name"

    # test thread answers
    msg = make_message("test2 subject", "test2 content")
    msg["From"] = "test1@example.com"
    msg["Message-ID"] = "<b@example.com>"
    msg["References"] = "<nope@example.com> <a@example.com>"

    task_count = Task.objects.count()
    consumer([msg])
    assert task_count == Task.objects.count(), "comment created another task"
    Comment.objects.get(
        task=task, body__contains="test2 content", email_message_id="<b@example.com>"
    )

    # test notification answer
    msg = make_message("test3 subject", "test3 content")
    msg["From"] = "test1@example.com"
    msg["Message-ID"] = "<c@example.com>"
    msg["References"] = "<thread-{}@django-todo> <unknown@example.com>".format(task.pk)

    task_count = Task.objects.count()
    consumer([msg])
    assert task_count == Task.objects.count(), "comment created another task"
    Comment.objects.get(
        task=task, body__contains="test3 content", email_message_id="<c@example.com>"
    )