# -*- 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 unittest.mock import patch from django.db.utils import IntegrityError from django.test import Client, 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. """ 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') Account.objects.create(profile=p1, trigramme='BAR') CofProfile.objects.create(profile=p2) Account.objects.create(profile=p2, trigramme='BAZ') self.assertRaises(IntegrityError, CofProfile.objects.create, profile=p2) class TestProfile(TestCase): @patch('kfet.signals.messages') def test_profile(self, mock_messages): User.objects.create_superuser('root', 'foo@bar.com', 'root') c = Client() c.login(username='root', password='root') response = c.get('/') self.assertEqual(response.status_code, 200)