import pytest from slugify import slugify from ctrack.organisations.models import Organisation pytestmark = pytest.mark.django_db def test_organisation_get_absolute_url(org): slug = slugify(org.name) assert org.get_absolute_url() == f"/organisations/{slug}/" def test_delete_organisation(org): orgs = Organisation.objects.all() assert org in orgs Organisation.delete(org) # Assert that the record has been deleted assert Organisation.objects.count() == 0 def test_update_organisation(org): # Change the name of the organisation org.name = "Tonkers Ltd" org.save() assert org.name == "Tonkers Ltd" def test_new_address(addr): # The address "has" an organisation assert addr.organisation.name