cookiecutter/cookiecutter-django
1
1
Fork 0
mirror of https://github.com/cookiecutter/cookiecutter-django.git synced 2025-07-06 13:23:09 +03:00
Code Issues Packages Projects Releases Wiki Activity

Override _after_postgeneration to force save in User factory (#4534)

Browse Source 
* Override _after_postgeneration to force save

Saving in _after_postgeneration was automatic in Factory Boy versions below 4.0, but must be done explicitly in newer versions.

* Clean up formatting
This commit is contained in:
John 2023-08-19 15:31:53 -04:00 committed by GitHub
parent 2ff9a12c30
commit 13b97de9d7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
{{cookiecutter.project_slug}}/{{cookiecutter.project_slug}}/users/tests/factories.py
View File

@ -29,6 +29,13 @@ class UserFactory(DjangoModelFactory):
) )
self.set_password(password) self.set_password(password)
@classmethod
def _after_postgeneration(cls, instance, create, results=None):
"""Save again the instance if creating and at least one hook ran."""
if create and results and not cls._meta.skip_postgeneration_save:
# Some post-generation hooks ran, and may have modified us.
instance.save()
class Meta: class Meta:
model = get_user_model() model = get_user_model()
django_get_or_create = ["{{cookiecutter.username_type}}"] django_get_or_create = ["{{cookiecutter.username_type}}"]