def test_list_menu(self):
menu_date = get_now_cl().strftime("%Y-%m-%d")
menu_date_2 = (get_now_cl() + timedelta(days=1)).strftime("%Y-%m-%d")
MenuFactory(menu_date=menu_date)
MenuFactory(menu_date=menu_date_2)
url = reverse('list_menu')
response = self.client.get(f'{url}', format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(len(response.data), 2)
def test_get_menu_empty_today(self):
uuid = generate_uuid()
menu_date = get_now_cl().strftime("%Y-%m-%d")
MenuFactory(menu_date=menu_date)
url = reverse('get_menu', kwargs={'uuid': uuid})
response = self.client.get(f'{url}', format='json')
self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)
class UserMenu(models.Model):
user = models.ForeignKey(
"user.User", on_delete=models.CASCADE, null=False, blank=False)
menu_option = models.ForeignKey(
"menu.MenuOptions", on_delete=models.CASCADE, null=False, blank=False)
quantity = models.IntegerField(default=1)
customizations = models.CharField(max_length=300, blank=False, null=False)
order_date = models.DateField(default=get_now_cl().strftime("%Y-%m-%d"))
def test_get_menu_today(self):
menu_date = get_now_cl().strftime("%Y-%m-%d")
menu = MenuFactory(menu_date=menu_date)
MenuOptionsFactory(menu=menu)
url = reverse('get_menu', kwargs={'uuid': menu.uuid})
response = self.client.get(f'{url}', format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.data['menu_date'].strftime("%Y-%m-%d"),
menu_date)
def test_get_ordes_today(self):
menu_date = get_now_cl().strftime("%Y-%m-%d")
menu = MenuFactory(menu_date=menu_date)
menu_opt = MenuOptionsFactory(menu=menu)
user = UserFactory()
order = OrderFactory(menu_option=menu_opt,
user=user,
order_date=menu_date)
url = reverse('get_orders', kwargs={'user_id': order.user_id})
response = self.client.get(f'{url}', format='json')
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.data[0]['menu_option'], order.menu_option_id)
def get(self, request, user_id):
try:
User.objects.get(id=user_id)
except User.DoesNotExist:
raise ValidationError({
'user': '******'
})
today = get_now_cl()
today = today.strftime("%Y-%m-%d")
query = UserMenu.objects
#Nora(admin) is user_id=1
if user_id == settings.ADMIN_ID:
orders = query.filter(order_date=today).all()
else:
orders = query.filter(user=user_id, order_date=today).all()
orders_serializer = UserMenuSerializer(orders, many=True)
return Response(orders_serializer.data)
def validate_time(self):
now = get_now_cl()
if now.hour > settings.LIMIT_TIME:
return True
return False