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

import pytest
from slugify import slugify

from ..models import Organisation, Address

pytestmark = pytest.mark.django_db


def test_organisation_get_absolute_url(org: Organisation):
    slug = slugify(org.name)
    assert org.get_absolute_url() == f"/organisations/{slug}/"


def test_create_organisation():
    Organisation(name="Big Bad OES Corporation").save()
    # The organisation is saved in the db
    assert Organisation.objects.get(name="Big Bad OES Corporation")


def test_delete_organisation(org: Organisation):
    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: Organisation):
    # Change the name of the organisation
    org.name = "Tonkers Ltd"
    org.save()
    assert org.name == "Tonkers Ltd"


def test_new_address(addr: Address):
    # The address "has" an organisation
    assert addr.organisation.name