cgiacofei/TandoorRecipes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

basic test stuff

Browse Source
This commit is contained in:
vabene1111
2021-02-20 22:56:29 +01:00
parent d7675d4b80
commit 58c6077925
15 changed files with 40 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
cookbook/tests/test_setup.py
View File

@ -2,45 +2,64 @@ from django.contrib import auth
from django.contrib.auth.models import Group, User
from django.test import Client, TestCase
from cookbook.models import Space
class TestBase(TestCase):
superuser_client = None
anonymous_client = None
guest_client_1 = None
guest_client_2 = None
user_client_1 = None
user_client_2 = None
admin_client_1 = None
admin_client_2 = None
guest_client_1 = None
guest_client_2 = None
superuser_client = None
anonymous_client = None
def create_login_user(self, name, group):
s2_guest_client_1 = None
s2_user_client_1 = None
s2_admin_client_1 = None
s2_superuser_client = None
def create_login_user(self, name, group, space, superuser=False):
client = Client()
setattr(self, name, client)
client.force_login(User.objects.get_or_create(username=name)[0])
user = auth.get_user(getattr(self, name))
user.groups.add(Group.objects.get(name=group))
self.assertTrue(user.is_authenticated)
if superuser:
user.is_superuser = True
user.userpreference.space = space
user.save()
return user
def setUp(self):
self.create_login_user('admin_client_1', 'admin')
self.create_login_user('admin_client_2', 'admin')
# users for space 1
space_1 = Space.objects.create(name='space 1')
self.create_login_user('user_client_1', 'user')
self.create_login_user('user_client_2', 'user')
self.create_login_user('admin_client_1', 'admin', space_1)
self.create_login_user('admin_client_2', 'admin', space_1)
self.create_login_user('guest_client_1', 'guest')
self.create_login_user('guest_client_2', 'guest')
self.create_login_user('user_client_1', 'user', space_1)
self.create_login_user('user_client_2', 'user', space_1)
self.create_login_user('guest_client_1', 'guest', space_1)
self.create_login_user('guest_client_2', 'guest', space_1)
self.anonymous_client = Client()
user = self.create_login_user('superuser_client', 'admin')
user.is_superuser = True
user.save()
self.create_login_user('superuser_client', 'admin', space_1, superuser=True)
def batch_requests(
self, clients, url, method='get', payload={}, content_type=''
):
# users for space 2
space_2 = Space.objects.create(name='space 2')
self.create_login_user('s2_admin_client_1', 'admin', space_2)
self.create_login_user('s2_user_client_1', 'user', space_2)
self.create_login_user('s2_guest_client_1', 'guest', space_2)
self.create_login_user('s2_superuser_client', 'admin', space_2, superuser=True)
def batch_requests(self, clients, url, method='get', payload={}, content_type=''):
for c in clients:
if method == 'get':
r = c[0].get(url)