21 lines
569 B
Python
21 lines
569 B
Python
from factory import DjangoModelFactory, Sequence, SubFactory
|
|
|
|
from ..models import AdminSection, AdminEntry
|
|
|
|
|
|
class AdminSectionFactory(DjangoModelFactory):
|
|
class Meta:
|
|
model = AdminSection
|
|
|
|
name = Sequence(lambda n: 'section{}'.format(n))
|
|
sort = Sequence(lambda n: n)
|
|
|
|
|
|
class AdminEntryFactory(DjangoModelFactory):
|
|
class Meta:
|
|
model = AdminEntry
|
|
|
|
package = Sequence(lambda n: 'package{}'.format(n))
|
|
model = Sequence(lambda n: 'section{}'.format(n))
|
|
section = SubFactory(AdminSectionFactory)
|
|
sort = Sequence(lambda n: n)
|