From 2f74c63fd9c2be470dd550e5ef7b3bad2ec26475 Mon Sep 17 00:00:00 2001 From: littlewolf Date: Fri, 20 Sep 2019 15:58:23 +0300 Subject: [PATCH] Fix plate tests Add menu tests --- apps/establishment/tests.py | 40 ++++++++++++++++++++++++++++++++----- 1 file changed, 35 insertions(+), 5 deletions(-) diff --git a/apps/establishment/tests.py b/apps/establishment/tests.py index ee4e444a..726eee5b 100644 --- a/apps/establishment/tests.py +++ b/apps/establishment/tests.py @@ -1,8 +1,10 @@ +import json from rest_framework.test import APITestCase from account.models import User from rest_framework import status from http.cookies import SimpleCookie -from establishment.models import Establishment, EstablishmentType +from main.models import Currency +from establishment.models import Establishment, EstablishmentType, Menu # Create your tests here. @@ -138,14 +140,22 @@ class SocialTests(ChildTestCase): class PlatesTests(ChildTestCase): - def test_plates_CRD(self): + def test_plate_CRD(self): response = self.client.get('/api/back/establishments/plates/', format='json') self.assertEqual(response.status_code, status.HTTP_200_OK) + menu = Menu.objects.create( + category=json.dumps({"en-GB": "Test category"}), + establishment=self.establishment + ) + currency = Currency.objects.create(name="Test currency") + data = { - 'title': "Test social", - 'url': 'https://testsocial.com', - 'establishment': self.establishment.id + 'name': json.dumps({"en-GB": "Test plate"}), + 'establishment': self.establishment.id, + 'price': 10, + 'menu': menu.id, + 'currency_id': currency.id } response = self.client.post('/api/back/establishments/plates/', data=data) @@ -157,3 +167,23 @@ class PlatesTests(ChildTestCase): response = self.client.delete('/api/back/establishments/plates/1/') self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) + + +class MenusTests(ChildTestCase): + def test_menu_CRD(self): + response = self.client.get('/api/back/establishments/menus/', format='json') + self.assertEqual(response.status_code, status.HTTP_200_OK) + + data = { + 'category': json.dumps({"en-GB": "Test category"}), + 'establishment': self.establishment.id + } + + response = self.client.post('/api/back/establishments/menus/', data=data) + self.assertEqual(response.status_code, status.HTTP_201_CREATED) + + response = self.client.get('/api/back/establishments/menus/1/', format='json') + self.assertEqual(response.status_code, status.HTTP_200_OK) + + response = self.client.delete('/api/back/establishments/menus/1/') + self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)