Merge pull request learningequality#5384 from ralphiee22/tests_intege…

…r_field

Add tests for pre/post migration on bigint/integer field

kolibri/core/content/test/test_annotation.py

@@ -2,6 +2,7 @@
 import uuid
 
 from django.core.management import call_command
+from django.db import DataError
 from django.test import TestCase
 from django.test import TransactionTestCase
 from le_utils.constants import content_kinds
@@ -679,3 +680,12 @@ def test_calculate_included_languages(self):
         self.assertEqual(
             list(self.channel.included_languages.values_list("id", flat=True)), ["en"]
         )
+
+    def test_published_size_big_integer_field(self):
+        self.channel.published_size = (
+            2150000000
+        )  # out of range for integer field on postgres
+        try:
+            self.channel.save()
+        except DataError:
+            self.fail("DataError: integer out of range")

kolibri/core/content/test/test_data_migrations.py

@@ -1,5 +1,9 @@
+import os
+import unittest
 import uuid
 
+from django.db import DataError
+
 from kolibri.core.auth.test.migrationtestcase import TestMigrations
 
 
@@ -109,3 +113,16 @@ def test_channel_order(self):
         for channel in ChannelMetadata.objects.all():
             self.assertEqual(channel.order, pos)
             pos += 1
+
+    @unittest.skipIf(
+        os.environ.get("TOX_ENV") != "postgres", "Should only be run for postgres"
+    )
+    def test_channel_published_size(self):
+        # tests the integer field overflow for postgres
+        ChannelMetadata = self.apps.get_model("content", "ChannelMetadata")
+        channel = ChannelMetadata.objects.first()
+        channel.published_size = (
+            2150000000
+        )  # out of range for integer field on postgres
+        with self.assertRaises(DataError):
+            channel.save()