added food api view tests

This commit is contained in:
vabene1111
2020-07-09 22:19:24 +02:00
parent 8c118dbd3e
commit 9b9fbd5cab
4 changed files with 65 additions and 11 deletions

View File

@ -0,0 +1,53 @@
import json
from django.urls import reverse
from cookbook.models import Food
from cookbook.tests.views.test_views import TestViews
class TestApiUnit(TestViews):
def setUp(self):
super(TestApiUnit, self).setUp()
self.food_1 = Food.objects.create(
name='Beef'
)
self.food_2 = Food.objects.create(
name='Chicken'
)
def test_keyword_list(self):
# verify view permissions are applied accordingly
self.batch_requests([(self.anonymous_client, 403), (self.guest_client_1, 403), (self.user_client_1, 200), (self.admin_client_1, 200), (self.superuser_client, 200)],
reverse('api:food-list'))
# verify storage is returned
r = self.user_client_1.get(reverse('api:food-list'))
self.assertEqual(r.status_code, 200)
response = json.loads(r.content)
self.assertEqual(len(response), 2)
self.assertEqual(response[0]['name'], self.food_1.name)
r = self.user_client_1.get(f'{reverse("api:food-list")}?limit=1')
response = json.loads(r.content)
self.assertEqual(len(response), 1)
r = self.user_client_1.get(f'{reverse("api:food-list")}?query=Pork')
response = json.loads(r.content)
self.assertEqual(len(response), 0)
r = self.user_client_1.get(f'{reverse("api:food-list")}?query=Beef')
response = json.loads(r.content)
self.assertEqual(len(response), 1)
def test_keyword_update(self):
r = self.user_client_1.patch(reverse('api:food-detail', args={self.food_1.id}), {'name': 'new'}, content_type='application/json')
response = json.loads(r.content)
self.assertEqual(r.status_code, 200)
self.assertEqual(response['name'], 'new')
def test_keyword_delete(self):
r = self.user_client_1.delete(reverse('api:food-detail', args={self.food_1.id}))
self.assertEqual(r.status_code, 204)
self.assertEqual(Food.objects.count(), 1)

View File

@ -11,10 +11,10 @@ class TestApiUnit(TestViews):
def setUp(self):
super(TestApiUnit, self).setUp()
self.unit_1 = Unit.objects.create(
name='Beef'
name='kg'
)
self.unit_2 = Unit.objects.create(
name='Chicken'
name='g'
)
def test_keyword_list(self):
@ -33,11 +33,11 @@ class TestApiUnit(TestViews):
response = json.loads(r.content)
self.assertEqual(len(response), 1)
r = self.user_client_1.get(f'{reverse("api:unit-list")}?query=Pork')
r = self.user_client_1.get(f'{reverse("api:unit-list")}?query=m')
response = json.loads(r.content)
self.assertEqual(len(response), 0)
r = self.user_client_1.get(f'{reverse("api:unit-list")}?query=Beef')
r = self.user_client_1.get(f'{reverse("api:unit-list")}?query=kg')
response = json.loads(r.content)
self.assertEqual(len(response), 1)

View File

@ -16,8 +16,9 @@ router.register(r'sync', api.SyncViewSet)
router.register(r'sync-log', api.SyncLogViewSet)
router.register(r'keyword', api.KeywordViewSet)
router.register(r'unit', api.UnitViewSet)
router.register(r'food', api.FoodViewSet)
router.register(r'step', api.StepViewSet)
router.register(r'recipe', api.RecipeViewSet)
router.register(r'ingredient', api.IngredientViewSet)

View File

@ -124,6 +124,12 @@ class UnitViewSet(viewsets.ModelViewSet, StandardFilterMixin):
permission_classes = [CustomIsUser]
class FoodViewSet(viewsets.ModelViewSet, StandardFilterMixin):
queryset = Food.objects.all()
serializer_class = FoodSerializer
permission_classes = [CustomIsUser]
class RecipeBookViewSet(RetrieveModelMixin, UpdateModelMixin, ListModelMixin, viewsets.GenericViewSet):
queryset = RecipeBook.objects.all()
serializer_class = RecipeBookSerializer
@ -170,12 +176,6 @@ class MealTypeViewSet(viewsets.ModelViewSet):
return queryset
class FoodViewSet(viewsets.ModelViewSet, StandardFilterMixin):
queryset = Food.objects.all()
serializer_class = FoodSerializer
permission_classes = [CustomIsUser]
class IngredientViewSet(viewsets.ModelViewSet):
queryset = Ingredient.objects.all()
serializer_class = IngredientSerializer