def test_delete_used_pack_ind(self):
"""test PackManager.delete_pack method with used pack"""
db_pack = Paquet.objects.create(
user=self.usertest,
type_cig='IND',
brand='ELPASO',
qt_paquet=5,
price=10,
)
ConsoCig.objects.create(
user=self.usertest,
date_cig=datetime.date(2020, 5, 13),
time_cig=datetime.time(13, 55),
paquet=db_pack,
)
datas ={'id_pack': db_pack.id}
pack = PackManager(self.usertest, datas)
pack.delete_pack()
filter_pack = Paquet.objects.filter(
user=self.usertest,
type_cig='IND',
brand='ELPASO',
qt_paquet=5,
price=10,
)
self.assertTrue(filter_pack.exists())
self.assertEqual(filter_pack[0].display, False)
def delete_pack(request, id_pack):
"""
Used when user click on the trash of one of the paquet
Don't delete it but change display attribute into False if already used
"""
if Paquet.objects.filter(user=request.user, id=id_pack).exists():
data = {'id_pack': id_pack}
new_pack = PackManager(request.user, data)
new_pack.delete_pack()
return redirect('QuitSoonApp:paquets')
# asked out of form so not expected request from user
raise Http404()
def test_delete_unused_pack_ind(self):
"""test PackManager.delete_pack method with unused pack"""
db_pack = Paquet.objects.create(
user=self.usertest,
type_cig='IND',
brand='CAMEL',
qt_paquet=20,
price=10,
)
data = {'id_pack': db_pack.id}
pack = PackManager(self.usertest, data)
pack.delete_pack()
self.assertFalse(Paquet.objects.filter(id=db_pack.id).exists())