path: root/ctrack/organisations/tests/test_models.py

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