def test_two_lc_site(self):
self.datetime_ouverture = datetime.today()-timedelta(days=50)
self.lc1 = LettreCommande.objects.create(marche=self.marche,\
charge_insertion=self.charge_insertion,\
site = self.site, \
date_debut=self.datetime_ouverture.date(),\
date_fin=date.today()+timedelta(days=100),\
avancement=models_etat.LC_EN_COURS)
self.frein1 = Frein.objects.create(lettre_commande=self.lc1, pro1=Frein.PRO_PP)
self.lc2 = LettreCommande.objects.create(marche=self.marche,\
charge_insertion=self.charge_insertion,\
site = self.site, \
date_debut=self.datetime_ouverture.date(),\
date_fin=date.today()+timedelta(days=100),\
avancement=models_etat.LC_EN_COURS)
self.frein2 = Frein.objects.create(lettre_commande=self.lc2, pro1=Frein.PRO_PP)
expected_result = LCFreinResult()
expected_result.add_first(Frein.PRO_INTIT, Frein.PRO_PP)
expected_result.add_first(Frein.PRO_INTIT, Frein.PRO_PP)
self.assertEqual(expected_result, SiteCalc(self.site, self.date_debut, self.date_fin).calculate()[1])
self.lc1.delete()
self.frein1.delete()
self.lc2.delete()
self.frein2.delete()
def test_complicated(self):
self.frein = Frein.objects.create(lettre_commande=self.lc,
pro1=Frein.PRO_EXPE,
pro2 = Frein.PRO_PP)
expected_result = LCFreinResult()
expected_result.add_first(Frein.PRO_INTIT, Frein.PRO_EXPE)
expected_result.add(Frein.PRO_INTIT, Frein.PRO_PP)
self.assertEqual(expected_result , LCFreinCalc(self.lc).calculate())
def test_add(self):
frein_categorie = Frein.PRO_INTIT
frein_valeur = Frein.PRO_ETU
expected_result = LCFreinResult()
expected_result.frein_result_dic[frein_categorie].all_frein[frein_valeur] += 1
produced_result = LCFreinResult()
produced_result.add(frein_categorie, frein_valeur)
self.assertEqual(expected_result, produced_result)
def test_addition(self):
lc_frein_result_1 = LCFreinResult()
lc_frein_result_1.add(Frein.PRO_INTIT, Frein.PRO_ETU)
lc_frein_result_2 = LCFreinResult()
lc_frein_result_2.add(Frein.PRO_INTIT, Frein.PRO_ETU)
expected_result = LCFreinResult()
expected_result.add(Frein.PRO_INTIT, Frein.PRO_ETU)
expected_result.add(Frein.PRO_INTIT, Frein.PRO_ETU)
self.assertEqual(expected_result, lc_frein_result_1 + lc_frein_result_2)
