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())