forked from erp-dev/erp
54 lines
2.3 KiB
Python
54 lines
2.3 KiB
Python
from django.contrib.auth import get_user_model
|
|
from rest_framework.test import APIClient
|
|
from django.test import TestCase
|
|
|
|
from basic_info import models as basic_models
|
|
|
|
|
|
class QuickCreateEmployeeUserAPITest(TestCase):
|
|
def setUp(self):
|
|
self.client = APIClient()
|
|
self.merchant = basic_models.Merchant.objects.create(
|
|
name='测试商户',
|
|
type=basic_models.MerchantTypeEnum.FACTORY,
|
|
)
|
|
self.admin_user = get_user_model().objects.create_user(username='admin', password='pass12345')
|
|
self.admin_employee = basic_models.Employee.objects.create(
|
|
merchant=self.merchant,
|
|
sys_user=self.admin_user,
|
|
name='管理员',
|
|
)
|
|
self.client.force_authenticate(user=self.admin_user)
|
|
self.url = '/api/v2/users/quick-create/'
|
|
self.payload = {
|
|
'username': 'new_user',
|
|
'password': 'strongPass#1',
|
|
'display_name': '新员工',
|
|
'merchant_id': self.merchant.id,
|
|
'mobile': '13800000000',
|
|
}
|
|
|
|
def test_quick_create_employee_user_success(self):
|
|
response = self.client.post(self.url, self.payload, format='json')
|
|
self.assertEqual(response.status_code, 201)
|
|
data = response.data
|
|
self.assertIn('user', data)
|
|
self.assertIn('employee', data)
|
|
self.assertEqual(data['user']['username'], self.payload['username'])
|
|
|
|
created_user = get_user_model().objects.get(username=self.payload['username'])
|
|
self.assertEqual(created_user.employee.merchant, self.merchant)
|
|
self.assertEqual(created_user.employee.name, self.payload['display_name'])
|
|
|
|
def test_quick_create_employee_user_duplicate_username(self):
|
|
get_user_model().objects.create_user(username=self.payload['username'], password='pass12345')
|
|
response = self.client.post(self.url, self.payload, format='json')
|
|
self.assertEqual(response.status_code, 400)
|
|
self.assertIn('用户名已存在', str(response.data))
|
|
|
|
def test_quick_create_employee_user_invalid_merchant(self):
|
|
payload = {**self.payload, 'merchant_id': 9999}
|
|
response = self.client.post(self.url, payload, format='json')
|
|
self.assertEqual(response.status_code, 400)
|
|
self.assertIn('商户不存在', str(response.data))
|