tests/universite/tests.py

   1 # encoding: utf-8
   2
   3 from __future__ import absolute_import
   4
   5 from django.core import management
   6 from django.db import models
   7 from django.test import TestCase
   8
   9 from auf.django.references import models as ref
  10 from tests.universite.models import Universite
  11
  12
  13 class EtablissementBaseTestCase(TestCase):
  14     fixtures = ['tests.yaml']
  15
  16     def test_add_universite(self):
  17         canada = ref.Pays.objects.get(code='CA')
  18         uqam = Universite.objects.create(nom='UQAM', pays=canada, recteur='Claude Corbo')
  19         self.assertEqual(uqam.ref, None)
  20         self.assertEqual(Universite.objects.count(), 1)
  21
  22     def test_create_universite_from_etablissement(self):
  23         etablissement = ref.Etablissement.objects.get(nom='Université de Montréal')
  24         udem = Universite.objects.create(ref=etablissement, recteur='Guy Breton')
  25         self.assertEqual(udem.nom, u'Université de Montréal')
  26         self.assertEqual(udem.pays.nom, u'Canada')
  27         self.assertEqual(udem.recteur, u'Guy Breton')
  28         self.assertEqual(udem.ref, etablissement)
  29
  30     def test_sync_references(self):
  31         etablissement = ref.Etablissement.objects.get(nom='Université de Montréal')
  32         udem = Universite.objects.create(ref=etablissement, recteur='Guy Breton')
  33         self.assertEqual(udem.nom, u'Université de Montréal')
  34         self.assertEqual(udem.pays.nom, u'Canada')
  35         udem.save()
  36         etablissement.nom = 'UdeM'
  37         etablissement.pays = ref.Pays.objects.get(code='FR')
  38         etablissement.save()
  39         management.call_command('sync_references')
  40         udem2 = Universite.objects.get(pk=udem.pk)
  41         self.assertEqual(udem2.nom, u'UdeM')
  42         self.assertEqual(udem2.pays.nom, u'France')
  43         self.assertEqual(udem2.recteur, u'Guy Breton')