From 0c3252fb974c39877d717f706674c98616f7ccdb Mon Sep 17 00:00:00 2001 From: Seth Yastrov Date: Mon, 28 Jun 2021 12:23:20 +0200 Subject: [PATCH] Check whether reported issues actually fails currently (#653) --- tests/typecheck/managers/test_managers.yml | 41 ++++++++++++++++++++++ 1 file changed, 41 insertions(+) diff --git a/tests/typecheck/managers/test_managers.yml b/tests/typecheck/managers/test_managers.yml index 5f8acec..1548afe 100644 --- a/tests/typecheck/managers/test_managers.yml +++ b/tests/typecheck/managers/test_managers.yml @@ -357,3 +357,44 @@ class ModelB(models.Model): movie = models.TextField() + +- case: override_manager_create1 + main: | + from myapp.models import MyModel + MyModel.objects.create() + installed_apps: + - myapp + files: + - path: myapp/__init__.py + - path: myapp/models.py + content: | + from django.db import models + class MyModelManager(models.Manager): + + def create(self, *args, **kwargs) -> 'MyModel': + return super().create(*args, **kwargs) + + + class MyModel(models.Model): + + objects = MyModelManager() + +- case: override_manager_create2 + main: | + from myapp.models import MyModel + MyModel.objects.create() + installed_apps: + - myapp + files: + - path: myapp/__init__.py + - path: myapp/models.py + content: | + from django.db import models + class MyModelManager(models.Manager['MyModel']): + + def create(self, *args, **kwargs) -> 'MyModel': + return super().create(*args, **kwargs) + + class MyModel(models.Model): + + objects = MyModelManager()