Fix location tests

apps/location/tests.py

@@ -6,6 +6,7 @@ from rest_framework import status
 from http.cookies import SimpleCookie
 
 from location.models import City, Region, Country, Language
+from django.contrib.gis.geos import Point
 from account.models import Role, UserRole
 
 class BaseTestCase(APITestCase):
@@ -34,6 +35,23 @@ class BaseTestCase(APITestCase):
         )
         self.lang.save()
 
+        self.country_ru = Country.objects.create(
+            name='{"ru-RU":"Russia"}',
+            code='23',
+            low_price=15,
+            high_price=150000,
+        )
+        self.country_ru.languages.add(self.lang)
+        self.country_ru.save()
+
+        self.role = Role.objects.create(role=Role.COUNTRY_ADMIN,
+                                        country=self.country_ru)
+        self.role.save()
+
+        self.user_role = UserRole.objects.create(user=self.user, role=self.role)
+
+        self.user_role.save()
+
         # role = Role.objects.create(role=Role.COUNTRY_ADMIN)
 
 
@@ -154,6 +172,7 @@ class AddressTests(BaseTestCase):
     def setUp(self):
         super().setUp()
 
+
         self.country = Country.objects.create(
             name=json.dumps({"en-GB": "Test country"}),
             code="test"
@@ -172,6 +191,13 @@ class AddressTests(BaseTestCase):
             country=self.country
         )
 
+        role = Role.objects.create(role=Role.COUNTRY_ADMIN, country=self.country)
+        role.save()
+
+        user_role = UserRole.objects.create(user=self.user, role=role)
+
+        user_role.save()
+
     def test_address_CRUD(self):
         response = self.client.get('/api/back/location/addresses/', format='json')
         self.assertEqual(response.status_code, status.HTTP_200_OK)
@@ -179,10 +205,8 @@ class AddressTests(BaseTestCase):
         data = {
             'city_id': self.city.id,
             'number': '+79999999',
-            "coordinates": {
-                "latitude": 37.0625,
-                "longitude": -95.677068
-            },
+            "latitude": 37.0625,
+            "longitude": -95.677068,
             "geo_lon": -95.677068,
             "geo_lat": 37.0625
         }