def test_free_category(self): menu = ProduitVendu() menu.produit = Produit.objects.get(nom="biere 50cl") menu.save() self.assertEqual(None, menu.get_free_categorie()) menu.produit = Produit.objects.get(nom="Menu Entree/Plat") cat_entrees = Categorie.objects.get(nom="Entrees") self.assertEqual(cat_entrees, menu.get_free_categorie()) entree = ProduitVendu() entree.produit = Produit.objects.get(nom="salade normande") entree.save() menu.contient.add(entree) cat_plats = Categorie.objects.get(nom="Plat") self.assertEqual(cat_plats, menu.get_free_categorie()) plat = ProduitVendu() plat.produit = Produit.objects.get(nom="entrecote") plat.save() menu.contient.add(plat) self.assertEqual(None, menu.get_free_categorie()) menu.contient.remove(entree) self.assertEqual(cat_entrees, menu.get_free_categorie()) menu.contient.add(entree) self.assertEqual(None, menu.get_free_categorie())