Fix test location

apps/location/tests.py

@@ -9,8 +9,8 @@ from location.models import City, Region, Country, Language
 from django.contrib.gis.geos import Point
 from account.models import Role, UserRole
 
-class BaseTestCase(APITestCase):
 
+class BaseTestCase(APITestCase):
     def setUp(self):
         self.username = 'sedragurda'
         self.password = 'sedragurdaredips19'
@@ -29,20 +29,14 @@ class BaseTestCase(APITestCase):
             {'access_token': tokkens.get('access_token'),
              'refresh_token': tokkens.get('refresh_token')})
 
-        self.lang = Language.objects.create(
+        self.lang = Language.objects.get(
             title='Russia',
             locale='ru-RU'
         )
-        self.lang.save()
 
-        self.country_ru = Country.objects.create(
-            name='{"ru-RU":"Russia"}',
-            code='23',
-            low_price=15,
-            high_price=150000,
+        self.country_ru = Country.objects.get(
+            name={"en-GB": "Russian"}
         )
-        self.country_ru.languages.add(self.lang)
-        self.country_ru.save()
 
         self.role = Role.objects.create(role=Role.COUNTRY_ADMIN,
                                         country=self.country_ru)
@@ -52,18 +46,15 @@ class BaseTestCase(APITestCase):
 
         self.user_role.save()
 
-        # role = Role.objects.create(role=Role.COUNTRY_ADMIN)
-
 
 class CountryTests(BaseTestCase):
     def setUp(self):
         super().setUp()
 
-
     def test_country_CRUD(self):
         data = {
-            'name': {"ru-RU":"Russia"},
-            'code': 'test'
+            'name': {"ru-RU": "NewCountry"},
+            'code': 'test1'
         }
 
         response = self.client.post('/api/back/location/countries/', data=data, format='json')