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))