import logging
from unittest.mock import patch

from django.test import TestCase
from django.urls import reverse
from pydantic import BaseModel
from rest_framework import status
from rest_framework.test import APIClient

from core.apps.accounts.models import ResetToken
from django_core.models import SmsConfirm
from core.services import SmsService
from django.contrib.auth import get_user_model


class TokenModel(BaseModel):
    access: str
    refresh: str


class SmsViewTest(TestCase):
    def setUp(self):
        self.client = APIClient()
        self.phone = "998999999999"
        self.password = "password"
        self.code = "1111"
        self.token = "token"
        self.user = get_user_model().objects.create_user(
            phone=self.phone, first_name="John", last_name="Doe", password=self.password
        )
        SmsConfirm.objects.create(phone=self.phone, code=self.code)

    def test_reg_view(self):
        """Test register view."""
        data = {
            "phone": "998999999991",
            "first_name": "John",
            "last_name": "Doe",
            "password": "password",
        }
        with patch.object(SmsService, "send_confirm", return_value=True):
            response = self.client.post(reverse("auth-register"), data=data)
            self.assertEqual(response.status_code, status.HTTP_202_ACCEPTED)
            self.assertEqual(
                response.data["data"]["detail"],
                "Sms %(phone)s raqamiga yuborildi" % {"phone": data["phone"]},
            )

    def test_confirm_view(self):
        """Test confirm view."""
        data = {"phone": self.phone, "code": self.code}
        response = self.client.post(reverse("auth-confirm"), data=data)
        self.assertEqual(response.status_code, status.HTTP_202_ACCEPTED)

    def test_invalid_confirm_view(self):
        """Test confirm view."""
        data = {"phone": self.phone, "code": "1112"}
        response = self.client.post(reverse("auth-confirm"), data=data)
        self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)

    def test_reset_confirmation_code_view(self):
        """Test reset confirmation code view."""
        data = {"phone": self.phone, "code": self.code}
        response = self.client.post(reverse("auth-confirm"), data=data)
        self.assertEqual(response.status_code, status.HTTP_202_ACCEPTED)
        self.assertIn("token", response.data["data"])

    def test_reset_confirmation_code_view_invalid_code(self):
        """Test reset confirmation code view with invalid code."""
        data = {"phone": self.phone, "code": "123456"}
        response = self.client.post(reverse("auth-confirm"), data=data)
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)

    def test_reset_set_password_view(self):
        """Test reset set password view."""
        token = ResetToken.objects.create(user=self.user, token=self.token)
        data = {"token": token.token, "password": "new_password"}
        response = self.client.post(reverse("reset-password-reset-password-set"), data=data)
        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_reset_set_password_view_invalid_token(self):
        """Test reset set password view with invalid token."""
        token = "test_token"
        data = {"token": token, "password": "new_password"}
        with patch.object(get_user_model().objects, "filter", return_value=get_user_model().objects.none()):
            response = self.client.post(reverse("reset-password-reset-password-set"), data=data)
            self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
            self.assertEqual(response.data["data"]["detail"], "Invalid token")

    def test_resend_view(self):
        """Test resend view."""
        data = {"phone": self.phone}
        response = self.client.post(reverse("auth-resend"), data=data)
        logging.error(response.json())
        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_reset_password_view(self):
        """Test reset password view."""
        data = {"phone": self.phone}
        response = self.client.post(reverse("reset-password-reset-password"), data=data)
        logging.error(response.json())
        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_me_view(self):
        """Test me view."""
        self.client.force_authenticate(user=self.user)
        response = self.client.get(reverse("me-me"))
        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_me_update_view(self):
        """Test me update view."""
        self.client.force_authenticate(user=self.user)
        data = {"first_name": "Updated"}
        response = self.client.patch(reverse("me-user-update"), data=data)
        logging.error(response.json())
        self.assertEqual(response.status_code, status.HTTP_200_OK)