150 lines
5.1 KiB
Python
150 lines
5.1 KiB
Python
import json
|
|
|
|
from cookbook.models import UserPreference
|
|
from cookbook.tests.views.test_views import TestViews
|
|
from django.contrib import auth
|
|
from django.urls import reverse
|
|
|
|
|
|
class TestApiUserPreference(TestViews):
|
|
|
|
def setUp(self):
|
|
super(TestApiUserPreference, self).setUp()
|
|
|
|
def test_preference_create(self):
|
|
r = self.user_client_1.post(reverse('api:userpreference-list'))
|
|
self.assertEqual(r.status_code, 201)
|
|
response = json.loads(r.content)
|
|
self.assertEqual(
|
|
response['user'], auth.get_user(self.user_client_1).id
|
|
)
|
|
self.assertEqual(
|
|
response['theme'],
|
|
UserPreference._meta.get_field('theme').get_default()
|
|
)
|
|
|
|
def test_preference_list(self):
|
|
UserPreference.objects.create(user=auth.get_user(self.user_client_1))
|
|
UserPreference.objects.create(user=auth.get_user(self.guest_client_1))
|
|
|
|
# users can only see own preference in list
|
|
r = self.user_client_1.get(reverse('api:userpreference-list'))
|
|
self.assertEqual(r.status_code, 200)
|
|
response = json.loads(r.content)
|
|
self.assertEqual(len(response), 1)
|
|
self.assertEqual(
|
|
response[0]['user'], auth.get_user(self.user_client_1).id
|
|
)
|
|
|
|
# superusers can see all user prefs in list
|
|
r = self.superuser_client.get(reverse('api:userpreference-list'))
|
|
self.assertEqual(r.status_code, 200)
|
|
response = json.loads(r.content)
|
|
self.assertEqual(len(response), 2)
|
|
|
|
def test_preference_retrieve(self):
|
|
UserPreference.objects.create(user=auth.get_user(self.user_client_1))
|
|
UserPreference.objects.create(user=auth.get_user(self.guest_client_1))
|
|
|
|
self.batch_requests(
|
|
[
|
|
(self.guest_client_1, 404),
|
|
(self.user_client_1, 200),
|
|
(self.user_client_2, 404),
|
|
(self.anonymous_client, 403),
|
|
(self.admin_client_1, 404),
|
|
(self.superuser_client, 200)
|
|
],
|
|
reverse(
|
|
'api:userpreference-detail',
|
|
args={auth.get_user(self.user_client_1).id}
|
|
)
|
|
)
|
|
|
|
def test_preference_update(self):
|
|
UserPreference.objects.create(user=auth.get_user(self.user_client_1))
|
|
UserPreference.objects.create(user=auth.get_user(self.guest_client_1))
|
|
|
|
# can update users preference
|
|
r = self.user_client_1.put(
|
|
reverse(
|
|
'api:userpreference-detail',
|
|
args={auth.get_user(self.user_client_1).id}
|
|
),
|
|
{'theme': UserPreference.DARKLY},
|
|
content_type='application/json'
|
|
)
|
|
response = json.loads(r.content)
|
|
self.assertEqual(r.status_code, 200)
|
|
self.assertEqual(response['theme'], UserPreference.DARKLY)
|
|
|
|
# cant set another users non existent pref
|
|
r = self.user_client_1.put(
|
|
reverse(
|
|
'api:userpreference-detail',
|
|
args={auth.get_user(self.user_client_2).id}
|
|
),
|
|
{'theme': UserPreference.DARKLY},
|
|
content_type='application/json'
|
|
)
|
|
self.assertEqual(r.status_code, 404)
|
|
|
|
# cant set another users existent pref
|
|
r = self.user_client_2.put(
|
|
reverse(
|
|
'api:userpreference-detail',
|
|
args={auth.get_user(self.user_client_1).id}
|
|
),
|
|
{'theme': UserPreference.FLATLY},
|
|
content_type='application/json'
|
|
)
|
|
self.assertEqual(r.status_code, 404)
|
|
|
|
# can set pref as superuser
|
|
r = self.superuser_client.put(
|
|
reverse(
|
|
'api:userpreference-detail',
|
|
args={auth.get_user(self.user_client_1).id}
|
|
),
|
|
{'theme': UserPreference.FLATLY},
|
|
content_type='application/json'
|
|
)
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
def test_preference_delete(self):
|
|
UserPreference.objects.create(user=auth.get_user(self.user_client_1))
|
|
|
|
# can delete own preference
|
|
r = self.user_client_1.delete(
|
|
reverse(
|
|
'api:userpreference-detail',
|
|
args={auth.get_user(self.user_client_1).id}
|
|
)
|
|
)
|
|
self.assertEqual(r.status_code, 204)
|
|
self.assertEqual(UserPreference.objects.count(), 0)
|
|
|
|
UserPreference.objects.create(user=auth.get_user(self.user_client_1
|
|
)
|
|
)
|
|
|
|
# cant delete other preference
|
|
r = self.user_client_2.delete(
|
|
reverse(
|
|
'api:userpreference-detail',
|
|
args={auth.get_user(self.user_client_1).id}
|
|
)
|
|
)
|
|
self.assertEqual(r.status_code, 404)
|
|
self.assertEqual(UserPreference.objects.count(), 1)
|
|
|
|
# superuser can delete everything
|
|
r = self.superuser_client.delete(
|
|
reverse(
|
|
'api:userpreference-detail',
|
|
args={auth.get_user(self.user_client_1).id}
|
|
)
|
|
)
|
|
self.assertEqual(r.status_code, 204)
|
|
self.assertEqual(UserPreference.objects.count(), 0)
|