|
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