from django.test import TestCase
from django.urls import reverse
from rest_framework import status
from rest_framework.test import APITestCase
from api.property_api.category.models.category import Category
from common.exceptions.exceptions import custom_exception_handler


class CategoryAPITestCase(APITestCase):
    def setUp(self):
        self.category_1 = Category.objects.create(name="Hospital")
        self.list_url = reverse("category-list-create")
        self.detail_url = lambda pk: reverse("category-detail", kwargs={"pk": pk})

    def test_create_category(self):
        data = {"name": "Restaurant"}
        response = self.client.post(self.list_url, data)
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        self.assertTrue(Category.objects.filter(name="Restaurant").exists())

    def test_list_active_categories(self):
        response = self.client.get(self.list_url)
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        self.assertEqual(len(response.data["results"]), 1)  # only active

    def test_get_category_detail(self):
        response = self.client.get(self.detail_url(self.category_1.id))
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        self.assertEqual(response.data["name"], self.category_1.name)

    def test_get_invalid_category_detail(self):
        response = self.client.get(self.detail_url(9999))
        self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)

    def test_update_category_detail(self):
        data = {"name": "School"}
        response = self.client.put(self.detail_url(self.category_1.id), data)
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        self.category_1.refresh_from_db()
        self.assertEqual(self.category_1.name, "School")