diff options
Diffstat (limited to 'tests/basic')
| -rw-r--r-- | tests/basic/tests.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/basic/tests.py b/tests/basic/tests.py index 5eada343e1..e186958400 100644 --- a/tests/basic/tests.py +++ b/tests/basic/tests.py @@ -5,7 +5,7 @@ from unittest import mock from django.core.exceptions import MultipleObjectsReturned, ObjectDoesNotExist from django.db import DEFAULT_DB_ALIAS, DatabaseError, connections, models from django.db.models.manager import BaseManager -from django.db.models.query import MAX_GET_RESULTS, EmptyQuerySet, QuerySet +from django.db.models.query import MAX_GET_RESULTS, EmptyQuerySet from django.test import ( SimpleTestCase, TestCase, TransactionTestCase, skipUnlessDBFeature, ) @@ -316,7 +316,7 @@ class ModelTest(TestCase): # A hacky test for custom QuerySet subclass - refs #17271 Article.objects.create(headline='foo', pub_date=datetime.now()) - class CustomQuerySet(QuerySet): + class CustomQuerySet(models.QuerySet): def do_something(self): return 'did something' @@ -607,7 +607,7 @@ class ManagerTest(SimpleTestCase): `Manager` will need to be added to `ManagerTest.QUERYSET_PROXY_METHODS`. """ self.assertEqual( - sorted(BaseManager._get_queryset_methods(QuerySet)), + sorted(BaseManager._get_queryset_methods(models.QuerySet)), sorted(self.QUERYSET_PROXY_METHODS), ) @@ -640,7 +640,7 @@ class SelectOnSaveTests(TestCase): orig_class = Article._base_manager._queryset_class - class FakeQuerySet(QuerySet): + class FakeQuerySet(models.QuerySet): # Make sure the _update method below is in fact called. called = False |
