kpsul/gestion/tests.py

# -*- coding: utf-8 -*-
"""
This file demonstrates writing tests using the unittest module. These will pass
when you run "manage.py test".

Replace this with more appropriate tests for your application.
"""

from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals

from django.db.utils import IntegrityError
from django.test import TestCase

from gestion.models import Profile, User

def create_profile(username):
    """
    In order to create a profile, I need to create a User first.
    This is annoying in most situations, as I just want a profile.
    """
    u = User.objects.create(username=username,
                            first_name=username.title(),
                            last_name=username.title())
    p, = Profile.objects.filter(user__username=username)
    return p


class SimpleTest(TestCase):
    def test_delete_user(self):
        u = User(username='foo', first_name='foo', last_name='bar')

        # to each user there's a cofprofile associated
        u.save()
        self.assertTrue(Profile.objects.filter(user__username='foo').exists())

        # there's no point in having a cofprofile without a user associated.
        u.delete()
        self.assertFalse(Profile.objects.filter(user__username='foo').exists())

        # there's no point in having a user without a cofprofile associated.
        u.save()
        Profile.objects.get(user__username='foo').delete()
        self.assertFalse(User.objects.filter(username='foo').exists())

    def test_create_profiles(self):
        from kfet.models import Account
        from cof.models import CofProfile

        # test that two accounts can be linked to the same profile.
        p = create_profile('foo')
        pkfet = Account.objects.create(profile=p, trigramme='PIG')
        pcof = CofProfile.objects.create(profile=p)
        self.assertEqual(pkfet.profile, pcof.profile)

        # each profile is linked to at most one account of the same type.
        p1 = create_profile('bar')
        p2 = create_profile('baz')
        pkfet1 = Account.objects.create(profile=p1, trigramme='BAR')
        pcof2 = CofProfile.objects.create(profile=p2)
        pkfet2 = Account.objects.create(profile=p2, trigramme='BAZ')
        self.assertRaises(IntegrityError,
                          CofProfile.objects.create, profile=p2)