64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.contrib.auth import get_user_model
|
|
from django.contrib.auth.models import Group
|
|
|
|
from .literals import (
|
|
TEST_GROUP_2_NAME, TEST_GROUP_2_NAME_EDITED, TEST_USER_2_EMAIL,
|
|
TEST_USER_2_PASSWORD, TEST_USER_2_USERNAME, TEST_USER_2_USERNAME_EDITED
|
|
)
|
|
|
|
|
|
class UserTestMixin(object):
|
|
def _create_test_group(self):
|
|
self.test_group = Group.objects.create(name=TEST_GROUP_2_NAME)
|
|
|
|
def _edit_test_group(self):
|
|
self.test_group.name = TEST_GROUP_2_NAME_EDITED
|
|
self.test_group.save()
|
|
|
|
def _create_test_user(self):
|
|
self.test_user = get_user_model().objects.create(
|
|
username=TEST_USER_2_USERNAME, email=TEST_USER_2_EMAIL,
|
|
password=TEST_USER_2_PASSWORD
|
|
)
|
|
|
|
def _request_test_group_create_view(self):
|
|
reponse = self.post(
|
|
viewname='user_management:group_create', data={
|
|
'name': TEST_GROUP_2_NAME
|
|
}
|
|
)
|
|
self.test_group = Group.objects.filter(name=TEST_GROUP_2_NAME).first()
|
|
return reponse
|
|
|
|
def _request_test_group_edit_view(self):
|
|
return self.post(
|
|
viewname='user_management:group_edit', kwargs={
|
|
'pk': self.test_group.pk
|
|
}, data={
|
|
'name': TEST_GROUP_2_NAME_EDITED
|
|
}
|
|
)
|
|
|
|
def _request_test_user_create_view(self):
|
|
reponse = self.post(
|
|
viewname='user_management:user_create', data={
|
|
'username': TEST_USER_2_USERNAME
|
|
}
|
|
)
|
|
|
|
self.test_user = get_user_model().objects.filter(
|
|
username=TEST_USER_2_USERNAME
|
|
).first()
|
|
return reponse
|
|
|
|
def _request_test_user_edit_view(self):
|
|
return self.post(
|
|
viewname='user_management:user_edit', kwargs={
|
|
'pk': self.test_user.pk
|
|
}, data={
|
|
'username': TEST_USER_2_USERNAME_EDITED
|
|
}
|
|
)
|