mirror of
https://github.com/boostorg/website-v2.git
synced 2026-02-27 17:42:08 +00:00
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import pytest
|
|
from model_bakery import baker
|
|
|
|
from mailing_list.models import MailingListMessage
|
|
|
|
|
|
@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
|