164 lines
4.4 KiB
Python
164 lines
4.4 KiB
Python
import json
|
|
|
|
import pytest
|
|
from django.contrib import auth
|
|
from django.urls import reverse
|
|
from django_scopes import scopes_disabled
|
|
|
|
from cookbook.models import MealType, UnitConversion
|
|
from cookbook.tests.conftest import get_random_food, get_random_unit
|
|
|
|
LIST_URL = 'api:unitconversion-list'
|
|
DETAIL_URL = 'api:unitconversion-detail'
|
|
|
|
|
|
@pytest.fixture()
|
|
def obj_1(space_1, u1_s1):
|
|
return UnitConversion.objects.get_or_create(
|
|
food=get_random_food(space_1, u1_s1),
|
|
base_amount=100,
|
|
base_unit=get_random_unit(space_1, u1_s1),
|
|
converted_amount=100,
|
|
converted_unit=get_random_unit(space_1, u1_s1),
|
|
created_by=auth.get_user(u1_s1),
|
|
space=space_1
|
|
)[0]
|
|
|
|
|
|
@pytest.fixture
|
|
def obj_2(space_1, u1_s1):
|
|
return UnitConversion.objects.get_or_create(
|
|
food=get_random_food(space_1, u1_s1),
|
|
base_amount=100,
|
|
base_unit=get_random_unit(space_1, u1_s1),
|
|
converted_amount=100,
|
|
converted_unit=get_random_unit(space_1, u1_s1),
|
|
created_by=auth.get_user(u1_s1),
|
|
space=space_1
|
|
)[0]
|
|
|
|
|
|
@pytest.mark.parametrize("arg", [
|
|
['a_u', 403],
|
|
['g1_s1', 403],
|
|
['u1_s1', 200],
|
|
['a1_s1', 200],
|
|
])
|
|
def test_list_permission(arg, request):
|
|
c = request.getfixturevalue(arg[0])
|
|
assert c.get(reverse(LIST_URL)).status_code == arg[1]
|
|
|
|
|
|
def test_list_space(obj_1, obj_2, u1_s1, u1_s2, space_2):
|
|
assert len(json.loads(u1_s1.get(reverse(LIST_URL)).content)) == 2
|
|
assert len(json.loads(u1_s2.get(reverse(LIST_URL)).content)) == 0
|
|
|
|
obj_1.space = space_2
|
|
obj_1.save()
|
|
|
|
assert len(json.loads(u1_s1.get(reverse(LIST_URL)).content)) == 1
|
|
assert len(json.loads(u1_s2.get(reverse(LIST_URL)).content)) == 1
|
|
|
|
|
|
@pytest.mark.parametrize("arg", [
|
|
['a_u', 403],
|
|
['g1_s1', 403],
|
|
['u1_s1', 200],
|
|
['a1_s1', 200],
|
|
['g1_s2', 403],
|
|
['u1_s2', 404],
|
|
['a1_s2', 404],
|
|
])
|
|
def test_update(arg, request, obj_1):
|
|
c = request.getfixturevalue(arg[0])
|
|
r = c.patch(
|
|
reverse(
|
|
DETAIL_URL,
|
|
args={obj_1.id}
|
|
),
|
|
{'base_amount': 1000},
|
|
content_type='application/json'
|
|
)
|
|
response = json.loads(r.content)
|
|
assert r.status_code == arg[1]
|
|
if r.status_code == 200:
|
|
assert response['base_amount'] == 1000
|
|
|
|
|
|
@pytest.mark.parametrize("arg", [
|
|
['a_u', 403],
|
|
['g1_s1', 403],
|
|
['u1_s1', 201],
|
|
['a1_s1', 201],
|
|
])
|
|
def test_add(arg, request, u1_s2, space_1, u1_s1):
|
|
with scopes_disabled():
|
|
c = request.getfixturevalue(arg[0])
|
|
random_unit_1 = get_random_unit(space_1, u1_s1)
|
|
random_unit_2 = get_random_unit(space_1, u1_s1)
|
|
random_food_1 = get_random_unit(space_1, u1_s1)
|
|
r = c.post(
|
|
reverse(LIST_URL),
|
|
{
|
|
'food': {'id': random_food_1.id, 'name': random_food_1.name},
|
|
'base_amount': 100,
|
|
'base_unit': {'id': random_unit_1.id, 'name': random_unit_1.name},
|
|
'converted_amount': 100,
|
|
'converted_unit': {'id': random_unit_2.id, 'name': random_unit_2.name}
|
|
|
|
},
|
|
content_type='application/json'
|
|
)
|
|
|
|
response = json.loads(r.content)
|
|
print(response)
|
|
assert r.status_code == arg[1]
|
|
if r.status_code == 201:
|
|
assert response['base_amount'] == 100
|
|
r = c.get(reverse(DETAIL_URL, args={response['id']}))
|
|
assert r.status_code == 200
|
|
r = u1_s2.get(reverse(DETAIL_URL, args={response['id']}))
|
|
assert r.status_code == 404
|
|
|
|
|
|
# TODO make name in space unique
|
|
# def test_add_duplicate(u1_s1, u1_s2, obj_1):
|
|
# r = u1_s1.post(
|
|
# reverse(LIST_URL),
|
|
# {'name': obj_1.name},
|
|
# content_type='application/json'
|
|
# )
|
|
# response = json.loads(r.content)
|
|
# assert r.status_code == 201
|
|
# assert response['id'] == obj_1.id
|
|
#
|
|
# r = u1_s2.post(
|
|
# reverse(LIST_URL),
|
|
# {'name': obj_1.name},
|
|
# content_type='application/json'
|
|
# )
|
|
# response = json.loads(r.content)
|
|
# assert r.status_code == 201
|
|
# assert response['id'] != obj_1.id
|
|
|
|
|
|
def test_delete(u1_s1, u1_s2, obj_1):
|
|
r = u1_s2.delete(
|
|
reverse(
|
|
DETAIL_URL,
|
|
args={obj_1.id}
|
|
)
|
|
)
|
|
assert r.status_code == 404
|
|
|
|
r = u1_s1.delete(
|
|
reverse(
|
|
DETAIL_URL,
|
|
args={obj_1.id}
|
|
)
|
|
)
|
|
|
|
assert r.status_code == 204
|
|
with scopes_disabled():
|
|
assert MealType.objects.count() == 0
|