Fix tests for newer versions of Django

lamby · Jun 21, 2019 · ba77822 · ba77822
1 parent 9338b5b
commit ba77822
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/tests_migrations.py b/tests/tests_migrations.py
@@ -3,6 +3,7 @@
 import tempfile
 import subprocess
 
+from django import VERSION
 from django.test import TestCase as DjangoTestCase
 
 
@@ -65,6 +66,10 @@ def test_stable_migrations(self):
         self.assertMigrationExists('0001')
 
         self.run_command('migrate')
+
+        if VERSION < (1, 10):
+            self.run_command('makemigrations tests --exit', expect_failure=True)
+        else:
+            self.run_command('makemigrations tests --check', expect_failure=True)
 
-        self.run_command('makemigrations tests --exit', expect_failure=True)
         self.assertMigrationExists('0002', exists=False)