feat: api_v2

api_v2/tests.py

@@ -0,0 +1,53 @@
+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))