Function test_self_ref

tests/test_relations.py:209–243

(db)

@pytest.mark.asyncio
async def test_self_ref(db):
    root = await Employee.create(name="Root")
    loose = await Employee.create(name="Loose")
    _1 = await Employee.create(name="1. First H1", manager=root)
    _2 = await Employee.create(name="2. Second H1", manager=root)
    _1_1 = await Employee.create(name="1.1. First H2", manager=_1)
    _1_1_1 = await Employee.create(name="1.1.1. First H3", manager=_1_1)
    _2_1 = await Employee.create(name="2.1. Second H2", manager=_2)
    _2_2 = await Employee.create(name="2.2. Third H2", manager=_2)

    await _1.talks_to.add(_2, _1_1_1, loose)
    await _2_1.gets_talked_to.add(_2_2, _1_1, loose)

    LOOSE_TEXT = "Loose (to: 2.1. Second H2) (from: 1. First H1)"
    ROOT_TEXT = """Root (to: ) (from: )
  1. First H1 (to: 1.1.1. First H3, 2. Second H1, Loose) (from: )
    1.1. First H2 (to: 2.1. Second H2) (from: )
      1.1.1. First H3 (to: ) (from: 1. First H1)
  2. Second H1 (to: ) (from: 1. First H1)
    2.1. Second H2 (to: ) (from: 1.1. First H2, 2.2. Third H2, Loose)
    2.2. Third H2 (to: 2.1. Second H2) (from: )"""

    # Evaluated off creation objects
    assert await loose.full_hierarchy__async_for() == LOOSE_TEXT
    assert await loose.full_hierarchy__fetch_related() == LOOSE_TEXT
    assert await root.full_hierarchy__async_for() == ROOT_TEXT
    assert await root.full_hierarchy__fetch_related() == ROOT_TEXT

    # Evaluated off new objects -> Result is identical
    root2 = await Employee.get(name="Root")
    loose2 = await Employee.get(name="Loose")
    assert await loose2.full_hierarchy__async_for() == LOOSE_TEXT
    assert await loose2.full_hierarchy__fetch_related() == LOOSE_TEXT
    assert await root2.full_hierarchy__async_for() == ROOT_TEXT
    assert await root2.full_hierarchy__fetch_related() == ROOT_TEXT


@pytest.mark.asyncio