|
5 | 5 | import warnings |
6 | 6 |
|
7 | 7 | import django |
| 8 | +from django.contrib.auth import get_user_model |
| 9 | +from django.core.files.base import ContentFile |
8 | 10 | from django.db import models |
9 | 11 | from django.db.models.fields.proxy import OrderWrt |
10 | 12 | from django.test import TestCase |
11 | | -from django.core.files.base import ContentFile |
12 | 13 |
|
13 | 14 | from simple_history.models import HistoricalRecords, convert_auto_field |
14 | 15 | from ..models import ( |
|
28 | 29 | from django.db.models import get_model |
29 | 30 | else: |
30 | 31 | get_model = apps.get_model |
31 | | -try: |
32 | | - from unittest import skipUnless |
33 | | -except ImportError: |
34 | | - from unittest2 import skipUnless |
35 | | -try: |
36 | | - from django.contrib.auth import get_user_model |
37 | | - User = get_user_model() |
38 | | -except ImportError: # django 1.4 compatibility |
39 | | - from django.contrib.auth.models import User |
40 | 32 |
|
| 33 | +User = get_user_model() |
41 | 34 | today = datetime(2021, 1, 1, 10, 0) |
42 | 35 | tomorrow = today + timedelta(days=1) |
43 | 36 | yesterday = today - timedelta(days=1) |
@@ -499,7 +492,7 @@ def test_string_related(self): |
499 | 492 | related_model = field_object.related.model |
500 | 493 | self.assertEqual(related_model, HistoricalState) |
501 | 494 |
|
502 | | - @skipUnless(django.get_version() >= "1.7", "Requires 1.7 migrations") |
| 495 | + @unittest.skipUnless(django.get_version() >= "1.7", "Requires 1.7 migrations") |
503 | 496 | def test_state_serialization_of_customfk(self): |
504 | 497 | from django.db.migrations import state |
505 | 498 | state.ModelState.from_model(HistoricalCustomFKError) |
@@ -651,7 +644,7 @@ def test_restore_object_with_changed_order(self): |
651 | 644 | self.assertEqual(order[5], self.w_chair.pk) |
652 | 645 | self.assertEqual(order[6], self.w_battle.pk) |
653 | 646 |
|
654 | | - @skipUnless(django.get_version() >= "1.7", "Requires 1.7 migrations") |
| 647 | + @unittest.skipUnless(django.get_version() >= "1.7", "Requires 1.7 migrations") |
655 | 648 | def test_migrations_include_order(self): |
656 | 649 | from django.db.migrations import state |
657 | 650 | model_state = state.ModelState.from_model(SeriesWork.history.model) |
|
0 commit comments