def test_delete_conso_cig(self):
"""test SmokeManager.delete_conso method"""
smoke = SmokeManager(self.usertest, self.old_smoke_ind_data)
smoke.delete_conso_cig()
filter_conso = ConsoCig.objects.filter(user=self.usertest,
paquet=self.db_pack_ind.id)
self.assertFalse(filter_conso.exists())
def delete_smoke(request, id_smoke):
"""
Used when user click on the trash of one of his smocke action
"""
if ConsoCig.objects.filter(user=request.user, id=id_smoke).exists():
data = {'id_smoke': id_smoke}
smoky = SmokeManager(request.user, data)
smoky.delete_conso_cig()
return redirect('QuitSoonApp:smoke_list')
raise Http404()
def test_delete_conso_cig_given_cig(self):
"""test SmokeManager.delete_conso method with cig == given cig"""
db_smoke_given = ConsoCig.objects.create(
user=self.usertest,
datetime_cig=datetime.datetime(2020,
5,
17,
13,
15,
tzinfo=pytz.utc),
paquet=None,
)
data = {'id_smoke': db_smoke_given.id}
smoke = SmokeManager(self.usertest, data)
smoke.delete_conso_cig()
filter_conso = ConsoCig.objects.filter(user=self.usertest,
id=db_smoke_given.id)
self.assertFalse(filter_conso.exists())
def test_delete_conso_cig_fail(self):
"""test SmokeManager.delete_conso method failing cause wrong id"""
smoke = SmokeManager(self.usertest, {'id_smoke': 17364})
self.assertRaises(AttributeError, smoke.delete_conso_cig())
