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.contact.models.contact import ContactMessage


class ContactAPITestCase(APITestCase):
    def setUp(self):
        self.country_1 = ContactMessage.objects.create(name="Bangladesh", email="BD@gmail.com", phone="0987655544", message="Hello")
        self.list_url = reverse("contact-message")

    def test_create_contact(self):
        data = {"name": "Nepal", "email":"BD@gmail.com", "phone":"0987655544", "message":"Hello"}
        response = self.client.post(self.list_url, data)
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        self.assertTrue(ContactMessage.objects.filter(name="Nepal").exists())