test case for truncate query is added and for current working cursor …

…method changed to sqlite sql code

heroes_and_monsters/entities/models.py

@@ -17,7 +17,7 @@ class Meta:
     @classmethod
     def truncate(cls):
         with connection.cursor() as cursor:
-            cursor.execute('TRUNCATE TABLE "{0}" CASCADE'.format(cls._meta.db_table))
+            cursor.execute('DELETE FROM "{0}"'.format(cls._meta.db_table))
 
     def __str__(self):
         return self.name

heroes_and_monsters/entities/tests.py

@@ -163,3 +163,15 @@ def test_hero_count(self):
         category = Category.objects.get(id=2)
 
         self.assertEqual(category.hero_count, hero_count + 1)
+
+
+class TestTruncateQuery(GlobalCategoryTestData, TestCase):
+    def test_truncate_data(self):
+        self.assertEqual(Category.objects.all().count(), 3)
+        Category.objects.all().delete()
+        self.assertEqual(Category.objects.all().count(), 0)
+
+    def test_truncate_data_using_cursor(self):
+        self.assertEqual(Category.objects.all().count(), 3)
+        Category.truncate()
+        self.assertEqual(Category.objects.all().count(), 0)