From 69927387f8e48bd796c6952e14aaaa9637c34f4d Mon Sep 17 00:00:00 2001 From: Bruno Alla Date: Wed, 21 Aug 2024 11:35:20 +0200 Subject: [PATCH] Add type hint for UserFactory (#5312) Since factory-boy 3.3.1, we can type annotate factories: https://github.com/FactoryBoy/factory_boy/pull/1059 Setting that up for the only factory of the project --- .../{{cookiecutter.project_slug}}/users/tests/factories.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/factories.py b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/factories.py index 136d0b1d..d3115d9d 100644 --- a/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/factories.py +++ b/{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/factories.py @@ -8,7 +8,7 @@ from factory.django import DjangoModelFactory from {{ cookiecutter.project_slug }}.users.models import User -class UserFactory(DjangoModelFactory): +class UserFactory(DjangoModelFactory[User]): {%- if cookiecutter.username_type == "username" %} username = Faker("user_name") {%- endif %}