57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
from rest_framework.test import APITestCase
|
|
from rest_framework import status
|
|
from authorization.tests.tests_authorization import get_tokens_for_user
|
|
from django.urls import reverse
|
|
from http.cookies import SimpleCookie
|
|
from location.models import Country
|
|
from account.models import Role, User, UserRole
|
|
|
|
|
|
class CommentModeratorPermissionTests(APITestCase):
|
|
def setUp(self):
|
|
self.data = get_tokens_for_user()
|
|
|
|
self.client.cookies = SimpleCookie(
|
|
{'access_token': self.data['tokens'].get('access_token'),
|
|
'refresh_token': self.data['tokens'].get('access_token')})
|
|
|
|
self.country_ru = Country.objects.create(
|
|
name='{"ru-RU":"Russia"}',
|
|
code='23',
|
|
low_price=15,
|
|
high_price=150000
|
|
)
|
|
self.country_ru.save()
|
|
|
|
self.country_en = Country.objects.create(
|
|
name='{"en-GB":"England"}',
|
|
code='25',
|
|
low_price=15,
|
|
high_price=150000
|
|
)
|
|
self.country_en.save()
|
|
|
|
self.role = Role.objects.create(
|
|
role=2,
|
|
country=self.country_ru
|
|
)
|
|
self.role.save()
|
|
|
|
self.moderator = User.objects.create_user(username='moderator',
|
|
email='moderator@mail.com',
|
|
password='passwordmoderator')
|
|
|
|
self.userRole = UserRole.objects.create(
|
|
user=self.moderator,
|
|
role=self.role
|
|
)
|
|
self.userRole.save()
|
|
|
|
tokens = User.create_jwt_tokens(self.moderator)
|
|
|
|
self.client.cookies = SimpleCookie(
|
|
{'access_token': tokens.get('access_token'),
|
|
'refresh_token': tokens.get('access_token')})
|
|
|
|
def test_permission(self):
|
|
self.assertTrue(True) |