Files
website-v2/mailing_list/tests/test_models.py
Natalia a8cd8f507a Ran pre-commit run -a
Completed runs of black and rest of pre-commit checks.
2023-05-30 23:21:53 -03:00

31 lines
1.1 KiB
Python

import pytest
from model_bakery import baker
@pytest.fixture
def mailing_list_message_factory():
def factory(**kwargs):
return baker.make("mailing_list.MailingListMessage", **kwargs)
return factory
def test_mailing_list_message_tree_structure(mailing_list_message_factory):
root = mailing_list_message_factory(subject="Root")
child1 = mailing_list_message_factory(subject="Child 1", parent=root)
child2 = mailing_list_message_factory(subject="Child 2", parent=root)
grandchild1 = mailing_list_message_factory(subject="Grandchild 1", parent=child1)
grandchild2 = mailing_list_message_factory(subject="Grandchild 2", parent=child1)
assert root.get_children().count() == 2
assert child1 in root.get_children()
assert child2 in root.get_children()
assert child1.get_children().count() == 2
assert grandchild1 in child1.get_children()
assert grandchild2 in child1.get_children()
assert child2.get_children().count() == 0
assert grandchild1.get_children().count() == 0
assert grandchild2.get_children().count() == 0