From c50bdc863745caf442a8a51f77e964a43be7bbf9 Mon Sep 17 00:00:00 2001
From: littlewolf <nickptic@code-artel.com>
Date: Tue, 24 Sep 2019 21:34:21 +0300
Subject: [PATCH] Add city tests

---
 apps/location/tests.py | 42 +++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 41 insertions(+), 1 deletion(-)

diff --git a/apps/location/tests.py b/apps/location/tests.py
index ceeb2043..d85db414 100644
--- a/apps/location/tests.py
+++ b/apps/location/tests.py
@@ -66,7 +66,7 @@ class RegionTests(BaseTestCase):
         )
 
     def test_region_CRUD(self):
-        response = self.client.get('/api/back/location/countries/', format='json')
+        response = self.client.get('/api/back/location/regions/', format='json')
         self.assertEqual(response.status_code, status.HTTP_200_OK)
 
         data = {
@@ -93,6 +93,46 @@ class RegionTests(BaseTestCase):
         self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
 
 
+class CityTests(RegionTests):
+
+    def setUp(self):
+        super().setUp()
+
+        self.region = Region.objects.create(
+            name="Test region",
+            code="812",
+            country=self.country
+        )
+
+    def test_city_CRUD(self):
+        response = self.client.get('/api/back/location/cities/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        data = {
+            'name': 'Test country',
+            'code': 'test',
+            'country_id': self.country.id,
+            'region_id': self.region.id
+        }
+
+        response = self.client.post('/api/back/location/cities/', data=data, format='json')
+        response_data = response.json()
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+
+        response = self.client.get(f'/api/back/location/cities/{response_data["id"]}/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        update_data = {
+            'name': json.dumps({"en-GB": "Test new country"})
+        }
+
+        response = self.client.patch(f'/api/back/location/cities/{response_data["id"]}/', data=update_data)
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+
+        response = self.client.delete(f'/api/back/location/cities/{response_data["id"]}/', format='json')
+        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
+
+
 class AddressTests(BaseTestCase):
 
     def setUp(self):