import json import pytest from django.contrib import auth from django.urls import reverse from django_scopes import scopes_disabled from cookbook.models import Storage, Sync LIST_URL = 'api:sync-list' DETAIL_URL = 'api:sync-detail' @pytest.fixture() def obj_1(space_1, u1_s1): s = Storage.objects.create( name='Test Storage', username='test', password='password', token='token', url='url', created_by=auth.get_user(u1_s1), space=space_1, ) return Sync.objects.create( storage=s, path='path', space=space_1, ) @pytest.fixture def obj_2(space_1, u1_s1): s = Storage.objects.create( name='Test Storage', username='test', password='password', token='token', url='url', created_by=auth.get_user(u1_s1), space=space_1, ) return Sync.objects.create( storage=s, path='path', space=space_1, ) @pytest.mark.parametrize("arg", [ ['a_u', 403], ['g1_s1', 403], ['u1_s1', 403], ['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, a1_s1, a1_s2, space_2): assert json.loads(a1_s1.get(reverse(LIST_URL)).content)['count'] == 2 assert json.loads(a1_s2.get(reverse(LIST_URL)).content)['count'] == 0 obj_1.space = space_2 obj_1.save() assert json.loads(a1_s1.get(reverse(LIST_URL)).content)['count'] == 1 assert json.loads(a1_s2.get(reverse(LIST_URL)).content)['count'] == 1 @pytest.mark.parametrize("arg", [ ['a_u', 403], ['g1_s1', 403], ['u1_s1', 403], ['a1_s1', 200], ['g1_s2', 403], ['u1_s2', 403], ['a1_s2', 404], ]) def test_update(arg, request, obj_1): c = request.getfixturevalue(arg[0]) print(reverse(DETAIL_URL, args={obj_1.id})) r = c.patch(reverse(DETAIL_URL, args={obj_1.id}), {'path': 'new'}, content_type='application/json') response = json.loads(r.content) assert r.status_code == arg[1] if r.status_code == 200: assert response['path'] == 'new' @pytest.mark.parametrize("arg", [ ['a_u', 403], ['g1_s1', 403], ['u1_s1', 403], ['a1_s1', 201], ]) def test_add(arg, request, a1_s2, obj_1): c = request.getfixturevalue(arg[0]) r = c.post(reverse(LIST_URL), { 'storage': obj_1.storage.pk, 'path': 'test' }, content_type='application/json') response = json.loads(r.content) print(r.content) assert r.status_code == arg[1] if r.status_code == 201: assert response['path'] == 'test' r = c.get(reverse(DETAIL_URL, args={response['id']})) assert r.status_code == 200 r = a1_s2.get(reverse(DETAIL_URL, args={response['id']})) assert r.status_code == 404 def test_delete(a1_s1, a1_s2, obj_1): r = a1_s2.delete(reverse(DETAIL_URL, args={obj_1.id})) assert r.status_code == 404 r = a1_s1.delete(reverse(DETAIL_URL, args={obj_1.id})) assert r.status_code == 204 with scopes_disabled(): assert Sync.objects.count() == 0