from django.contrib.gis.geos import GEOSGeometry
from django.test import TestCase
from django.urls import reverse
from copy import copy

from apps.cadastral.models import Number
from apps.cadastral.tests.factory import NumberFactory
from apps.core._tests_mixin import CompanyRequiredTestMixin
from apps.users.models import User


class NumberCreateTest(CompanyRequiredTestMixin, TestCase):
    @classmethod
    def setUp(cls):
        cls.number_instance = NumberFactory()
        cls.create_credentials = {
            'number_instance': '1',
            'code': 807000541,
            'owner': 'Костишин Михайло Онуфрович',
            'koatuu': 4623380700,
            'area': 0.4216,
        }
        cls.log_credentials = {
            'email': 'testuser@hello.com',
            'password': 'secret'
        }
        cls.login_credentials = copy(cls.log_credentials)
        cls.login_credentials['username'] = cls.log_credentials['email']
        User.objects.create_user(**cls.log_credentials)

    def test_create(self):
        client = self.get_company_client()
        client.post(reverse('users:login'), self.login_credentials, follow=True)
        response = client.post(reverse('cadastral:number-create'), self.create_credentials, follow=True)

        number = Number.objects.filter(code=807000541).first()

        self.assertEqual(response.status_code, 200)
        self.assertTrue(number)
        self.assertEqual(number.koatuu, 4623380700)
        self.assertEqual(number.owner, 'Костишин Михайло Онуфрович')
        self.assertEqual(self.number_instance.id, number.id)