From 578c03b276e435bcd3ce9eb17b81e85135c2d3f3 Mon Sep 17 00:00:00 2001
From: sobolevn <mail@sobolevn.me>
Date: Sun, 2 Feb 2020 13:15:58 +0300
Subject: Fixed #31223 -- Added __class_getitem__() to Manager and QuerySet.

---
 tests/managers_regress/tests.py  | 7 +++++++
 tests/model_inheritance/tests.py | 8 ++++++++
 2 files changed, 15 insertions(+)

(limited to 'tests')

diff --git a/tests/managers_regress/tests.py b/tests/managers_regress/tests.py
index 3c2ba5e1f1..7fa6815959 100644
--- a/tests/managers_regress/tests.py
+++ b/tests/managers_regress/tests.py
@@ -1,7 +1,10 @@
+from unittest import skipUnless
+
 from django.db import models
 from django.template import Context, Template
 from django.test import SimpleTestCase, TestCase, override_settings
 from django.test.utils import isolate_apps
+from django.utils.version import PY37
 
 from .models import (
     AbstractBase1, AbstractBase2, AbstractBase3, Child1, Child2, Child3,
@@ -285,3 +288,7 @@ class TestManagerInheritance(SimpleTestCase):
 
         self.assertEqual(TestModel._meta.managers, (TestModel.custom_manager,))
         self.assertEqual(TestModel._meta.managers_map, {'custom_manager': TestModel.custom_manager})
+
+    @skipUnless(PY37, '__class_getitem__() was added in Python 3.7')
+    def test_manager_class_getitem(self):
+        self.assertIs(models.Manager[Child1], models.Manager)
diff --git a/tests/model_inheritance/tests.py b/tests/model_inheritance/tests.py
index 3568c538f0..60edaa3d99 100644
--- a/tests/model_inheritance/tests.py
+++ b/tests/model_inheritance/tests.py
@@ -1,9 +1,11 @@
 from operator import attrgetter
+from unittest import skipUnless
 
 from django.core.exceptions import FieldError, ValidationError
 from django.db import connection, models
 from django.test import SimpleTestCase, TestCase
 from django.test.utils import CaptureQueriesContext, isolate_apps
+from django.utils.version import PY37
 
 from .models import (
     Base, Chef, CommonInfo, GrandChild, GrandParent, ItalianRestaurant,
@@ -217,6 +219,12 @@ class ModelInheritanceTests(TestCase):
         self.assertSequenceEqual(qs, [p2, p1])
         self.assertIn(expected_order_by_sql, str(qs.query))
 
+    @skipUnless(PY37, '__class_getitem__() was added in Python 3.7')
+    def test_queryset_class_getitem(self):
+        self.assertIs(models.QuerySet[Post], models.QuerySet)
+        self.assertIs(models.QuerySet[Post, Post], models.QuerySet)
+        self.assertIs(models.QuerySet[Post, int, str], models.QuerySet)
+
 
 class ModelInheritanceDataTests(TestCase):
     @classmethod
-- 
cgit v1.3