40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
from datetime import datetime
|
|
|
|
import factory
|
|
|
|
from .models import Project, Grant, GrantReport
|
|
|
|
|
|
class ProjectFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = Project
|
|
|
|
name = factory.Sequence(lambda n: 'project{}'.format(n))
|
|
code = factory.Sequence(lambda n: 'p{}'.format(n))
|
|
iacuc_number = 'abc'
|
|
description = 'lorem ipsum'
|
|
sort_order = factory.Sequence(lambda n: n)
|
|
|
|
|
|
class GrantFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = Grant
|
|
|
|
title = factory.Sequence(lambda n: 'grant{}'.format(n))
|
|
code = factory.Sequence(lambda n: 'g{}'.format(n))
|
|
description = 'lorem ipsum'
|
|
sort_order = factory.Sequence(lambda n: n)
|
|
|
|
|
|
class GrantReportFactory(factory.DjangoModelFactory):
|
|
class Meta:
|
|
model = GrantReport
|
|
|
|
grant = factory.SubFactory(GrantFactory)
|
|
title = factory.Sequence(lambda n: 'grant{}'.format(n))
|
|
report_type = 'lorem ipsum'
|
|
description = 'lorem ipsum'
|
|
due_date = factory.LazyFunction(datetime.now)
|
|
submitted_date = factory.LazyFunction(datetime.now)
|
|
sort_order = factory.Sequence(lambda n: n)
|
|
|