def test_without_lead_assignee(self):
"""
Test that get_lead_assignee() returns None if there are assignees
but none of them is a lead.
"""
order = OrderFactory(assignees=[])
OrderAssigneeFactory(order=order, is_lead=False)
assert not order.get_lead_assignee()
def test_with_lead_assignee(self):
"""
Test that get_lead_assignee() returns the lead assignee if present.
"""
order = OrderFactory(assignees=[])
lead_assignee = OrderAssigneeFactory(order=order, is_lead=True)
OrderAssigneeFactory(order=order, is_lead=False)
assert order.get_lead_assignee() == lead_assignee
def test_without_assignees(self):
"""
Test that get_lead_assignee() returns None if there are no assignees.
"""
order = OrderFactory(assignees=[])
assert not order.get_lead_assignee()
