def test_clean_no_previous(self):
     now = timezone.now()
     form = SubmitCaffeineForm(self.user,
                               DRINK_TYPES.coffee,
                               data={
                                   'date': now.date(),
                                   'time': now.time()
                               })
     self.assertTrue(form.is_valid(), str(form.errors))
 def test_clean_old_previous(self):
     Caffeine.objects.create(
         user=self.user, ctype=DRINK_TYPES.coffee,
         date=timezone.now() - timedelta(minutes=10))
     now = timezone.now()
     form = SubmitCaffeineForm(
         self.user, DRINK_TYPES.coffee,
         data={'date': now.date(), 'time': now.time()})
     self.assertTrue(form.is_valid(), str(form.errors))
 def test_submit_older(self):
     Caffeine.objects.create(
         user=self.user, ctype=DRINK_TYPES.coffee,
         date=timezone.now())
     before = timezone.now() - timedelta(
         minutes=settings.MINIMUM_DRINK_DISTANCE + 1)
     form = SubmitCaffeineForm(
         self.user, DRINK_TYPES.coffee,
         data={'date': before.date(), 'time': before.time()})
     self.assertTrue(form.is_valid(), str(form.errors))
 def test_clean_old_previous(self):
     Caffeine.objects.create(user=self.user,
                             ctype=DRINK_TYPES.coffee,
                             date=timezone.now() - timedelta(minutes=10))
     now = timezone.now()
     form = SubmitCaffeineForm(self.user,
                               DRINK_TYPES.coffee,
                               data={
                                   'date': now.date(),
                                   'time': now.time()
                               })
     self.assertTrue(form.is_valid(), str(form.errors))
 def test_submit_older(self):
     Caffeine.objects.create(user=self.user,
                             ctype=DRINK_TYPES.coffee,
                             date=timezone.now())
     before = timezone.now() - timedelta(
         minutes=settings.MINIMUM_DRINK_DISTANCE + 1)
     form = SubmitCaffeineForm(self.user,
                               DRINK_TYPES.coffee,
                               data={
                                   'date': before.date(),
                                   'time': before.time()
                               })
     self.assertTrue(form.is_valid(), str(form.errors))
 def test_submit_close_older_fails(self):
     Caffeine.objects.create(
         user=self.user, ctype=DRINK_TYPES.coffee,
         date=timezone.now())
     close_before = timezone.now() - timedelta(
         minutes=settings.MINIMUM_DRINK_DISTANCE - 1)
     form = SubmitCaffeineForm(
         self.user, DRINK_TYPES.coffee,
         data={'date': close_before.date(), 'time': close_before.time()})
     self.assertFalse(form.is_valid())
     form_errors = form.non_field_errors()
     self.assertEqual(len(form_errors), 1)
     self.assertRegexpMatches(
         form_errors[0],
         r'^Your last %(drink)s was less than %(minutes)d minutes ago' % {
             'drink': DRINK_TYPES[DRINK_TYPES.coffee],
             'minutes': settings.MINIMUM_DRINK_DISTANCE
         })
 def test_clean_empty_date_and_time(self):
     caffeine = SubmitCaffeineForm(self.user,
                                   DRINK_TYPES.coffee,
                                   data={
                                       'date': '',
                                       'time': ''
                                   }).save()
     self.assertLessEqual(caffeine.date, timezone.now())
     self.assertEqual(caffeine.ctype, DRINK_TYPES.coffee)
     self.assertEqual(caffeine.user, self.user)
     self.assertEqual(caffeine.timezone, 'Europe/Berlin')
 def test_submit_close_older_fails(self):
     Caffeine.objects.create(user=self.user,
                             ctype=DRINK_TYPES.coffee,
                             date=timezone.now())
     close_before = timezone.now() - timedelta(
         minutes=settings.MINIMUM_DRINK_DISTANCE - 1)
     form = SubmitCaffeineForm(self.user,
                               DRINK_TYPES.coffee,
                               data={
                                   'date': close_before.date(),
                                   'time': close_before.time()
                               })
     self.assertFalse(form.is_valid())
     form_errors = form.non_field_errors()
     self.assertEqual(len(form_errors), 1)
     self.assertRegexpMatches(
         form_errors[0],
         r'^Your last %(drink)s was less than %(minutes)d minutes ago' % {
             'drink': DRINK_TYPES[DRINK_TYPES.coffee],
             'minutes': settings.MINIMUM_DRINK_DISTANCE
         })
 def test_save_with_passed_data(self):
     now = timezone.now()
     caffeine = SubmitCaffeineForm(self.user,
                                   DRINK_TYPES.coffee,
                                   data={
                                       'date': now.date(),
                                       'time': now.time()
                                   }).save()
     self.assertLessEqual(caffeine.date, timezone.now())
     self.assertEqual(caffeine.ctype, DRINK_TYPES.coffee)
     self.assertEqual(caffeine.user, self.user)
     self.assertEqual(caffeine.timezone, 'Europe/Berlin')
Exemple #10
0
def add_drink(request, userinfo, messages, *args, **kwargs):
    """
    Submit a caffeinated beverage.

    :param HttpRequest request: POST request
    :param User userinfo: :py:class:`caffeine.models.User`
    :param dict messages: message dictionary
    :return: messages array or :py:class:`django.http.HttpResponseBadRequest`

    """

    ctype = request.POST.get('beverage')
    drinktime = request.POST.get('time')
    time = None
    if not ctype:
        messages.setdefault('error', []).append(
            API_ERROR_MISSING_PARAM_BEVERAGE)
    elif not hasattr(DRINK_TYPES, ctype):
        messages.setdefault('error', []).append(
            API_ERROR_INVALID_BEVERAGE)
    if not drinktime:
        messages.setdefault('error', []).append(
            API_ERROR_MISSING_PARAM_TIME)
    else:
        time = _parse_drinktime(drinktime, messages)
        if time is not None and time > datetime.now() + timedelta(minutes=1):
            messages.setdefault('error', []).append(API_ERROR_FUTURE_DATETIME)
    if 'error' in messages:
        return HttpResponseBadRequest(json.dumps(messages), 'text/json')
    data = {'date': time.date(), 'time': time.time()}
    form = SubmitCaffeineForm(userinfo, getattr(DRINK_TYPES, ctype), data)
    form.date = time.date()
    form.time = time.time()
    if not form.is_valid():
        for key in form.errors:
            messages.setdefault('error', []).extend(form.errors[key])
        return HttpResponseBadRequest(json.dumps(messages), 'text/json')
    drink = form.save()
    messages['success'] = _('Your %(drink)s has been registered!') % {
        'drink': drink
    }
    return messages
 def test_clean_no_previous(self):
     now = timezone.now()
     form = SubmitCaffeineForm(
         self.user, DRINK_TYPES.coffee,
         data={'date': now.date(), 'time': now.time()})
     self.assertTrue(form.is_valid(), str(form.errors))