Fix plate tests

Add menu tests
This commit is contained in:
littlewolf 2019-09-20 15:58:23 +03:00
parent 014931d2c4
commit 2f74c63fd9

View File

@ -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)