def test_returns_empty_string_if_no_pk(self):
"""Test that if the object has no pk, an empty link is returned."""
book = BookFactory.build()
assert get_change_link(book) == ''
def test_returns_empty_string_if_obj_is_none(self):
"""Test that if None is passed, an empty link is returned."""
assert get_change_link(None) == ''
def test_generates_links_for_saved_objects(self):
"""Test that a valid change link is generated."""
book = BookFactory()
assert get_change_link(book) == (
f'<a href="/admin/support/book/{book.pk}/change/">{book.name}</a>')
def invoice_link(self, obj):
"""Returns a link to the invoice change page."""
if obj.invoice:
return get_change_link(obj.invoice)
return ''
def transferred_to_display(self, obj):
"""Link to the company that data for this company has been transferred to."""
return get_change_link(obj.transferred_to)
def adviser_link(self, obj):
"""Returns a link to the adviser."""
return get_change_link(obj.adviser)
def order_link(self, obj):
"""Returns a link to the order change page."""
order = Order.objects.filter(reference=obj.order_reference).first()
if order:
return get_change_link(order)
return obj.order_reference