def smoke(request): """User smokes""" # check if packs are in parameters to fill fields with actual packs packs = Paquet.objects.filter(user=request.user, display=True) context = {'packs': packs} tz_offset = get_client_offset(request) update_dt_user_model_field(request.user, tz_offset) if packs: smoke_form = SmokeForm(request.user, tz_offset) if request.method == 'POST': smoke_form = SmokeForm(request.user, tz_offset, request.POST) if smoke_form.is_valid(): smoky = SmokeManager(request.user, smoke_form.cleaned_data, tz_offset) smoky.create_conso_cig() return redirect('QuitSoonApp:today') context['smoke_form'] = smoke_form smoke_conso = ConsoCig.objects.filter( user=request.user).order_by('-datetime_cig') context['smoke'] = smoke_conso context['nb_smoke_today'] = smoke_conso.filter( datetime_cig__date=timezone.now().date()).count() if smoke_conso: last = smoke_conso.latest('datetime_cig').datetime_cig context['lastsmoke'] = get_delta_last_event(last) return render(request, 'QuitSoonApp/smoke.html', context)
def test_get_conso_cig_new_conso(self): """test SmokeManager.conso_cig method after creation object with new conso data""" smoke = SmokeManager(self.usertest, self.new_datas_ind) smoke.create_conso_cig() self.assertEqual( smoke.get_conso_cig, ConsoCig.objects.get( user=self.usertest, datetime_cig=smoke.datetime_cig, paquet=smoke.get_pack, given=smoke.given, ))
def test_create_conso_cig(self): """test SmokeManager.create_conso_cig method with new conso data""" smoke = SmokeManager(self.usertest, self.new_datas_ind) smoke.create_conso_cig() new_smoke = ConsoCig.objects.filter(user=self.usertest, datetime_cig=datetime.datetime( 2020, 5, 17, 13, 15, tzinfo=pytz.utc), paquet=self.db_pack_ind, given=False) self.assertTrue(new_smoke.exists()) self.assertEqual(smoke.get_pack.id, self.db_pack_ind.id)
def test_create_conso_cig_given(self): """ test SmokeManager.create_conso_cig method with given=True """ new_datas_given_cig = { 'date_smoke': datetime.date(2020, 7, 15), 'time_smoke': datetime.time(20, 15), 'ind_pack_field': self.db_pack_ind.id, 'rol_pack_field': self.db_pack_rol.id, 'given_field': True, } smoke = SmokeManager(self.usertest, new_datas_given_cig) smoke.create_conso_cig() self.assertEqual(smoke.get_pack, None) new_given_smoke = ConsoCig.objects.filter( user=self.usertest, datetime_cig=datetime.datetime(2020, 7, 15, 20, 15, tzinfo=pytz.utc), paquet=None, given=True) self.assertTrue(new_given_smoke.exists())
def test_create_conso_cig_datas_id_smoke(self): """test SmokeManager.create_conso_cig method with id_smoke in request""" smoke = SmokeManager(self.usertest, self.old_smoke_ind_data) self.assertEqual(smoke.create_conso_cig(), None)