관리-도구

편집 파일: external_factory_boy_noerror.py

"""
Test to validate that pylint_django doesn't produce
Instance of 'SubFactory' has no 'pk' member (no-member) warnings
"""
# pylint: disable=attribute-defined-outside-init, missing-docstring, too-few-public-methods
import factory
from django import test
from django.db import models

class Author(models.Model):
    name = models.CharField()

class Book(models.Model):
    title = models.CharField()
    author = models.ForeignKey(Author, related_name="books", on_delete=models.CASCADE)

class AuthorFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = "Author"

name = factory.Sequence(lambda n: f"Author {n}")

class BookFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = "Book"

title = factory.Sequence(lambda n: f"Book {n}")
    author = factory.SubFactory(AuthorFactory)
    reviewer = factory.LazyFunction(Author.objects.first())

class BookTestCase(test.LiveServerTestCase):
    serialized_rollback = True

def _fixture_setup(self):
        super()._fixture_setup()
        self.book = BookFactory()
        _author = AuthorFactory()
        _first_book = _author.books.first()
        self.assertIsNotNone(_first_book)

def test_author_is_not_none(self):
        self.assertGreater(self.book.pk, 0)
        self.assertGreater(self.book.author.pk, 0)

self.assertIsNotNone(self.book.title)
        self.assertIsNotNone(self.book.author.name)

def test_reviewer_is_not_none(self):
        self.assertGreater(self.book.reviewer.pk, 0)