Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions hypha/apply/activity/tests/factories.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
class ActivityFactory(factory.django.DjangoModelFactory):
class Meta:
model = Activity
skip_postgeneration_save = True

class Params:
internal = factory.Trait(visibility=TEAM)
Expand All @@ -38,6 +39,7 @@ def _get_manager(cls, model_class):
class EventFactory(factory.django.DjangoModelFactory):
class Meta:
model = Event
skip_postgeneration_save = True

type = factory.Iterator([choice[0] for choice in MESSAGES.choices])
by = factory.SubFactory(UserFactory)
Expand All @@ -47,6 +49,7 @@ class Meta:
class MessageFactory(factory.django.DjangoModelFactory):
class Meta:
model = Message
skip_postgeneration_save = True

type = "Email"
content = factory.Faker("sentence")
Expand Down
2 changes: 2 additions & 0 deletions hypha/apply/categories/tests/factories.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
class CategoryFactory(factory.django.DjangoModelFactory):
class Meta:
model = Category
skip_postgeneration_save = True

name = factory.Faker("word")
help_text = factory.Faker("sentence")
Expand All @@ -14,6 +15,7 @@ class Meta:
class OptionFactory(factory.django.DjangoModelFactory):
class Meta:
model = Option
skip_postgeneration_save = True

value = factory.Faker("word")
category = factory.SubFactory(CategoryFactory)
2 changes: 2 additions & 0 deletions hypha/apply/determinations/tests/factories.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@ def _build(cls, model_class, *args, **kwargs):
class DeterminationFactory(factory.django.DjangoModelFactory):
class Meta:
model = Determination
skip_postgeneration_save = True

class Params:
accepted = factory.Trait(outcome=ACCEPTED)
Expand Down Expand Up @@ -94,6 +95,7 @@ class Meta:
class DeterminationFormFactory(factory.django.DjangoModelFactory):
class Meta:
model = DeterminationForm
skip_postgeneration_save = True

name = factory.Faker("word")
form_fields = DeterminationFormFieldsFactory
20 changes: 20 additions & 0 deletions hypha/apply/funds/tests/factories/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,6 +77,7 @@ def workflow_for_stages(stages):
class AbstractApplicationFactory(wagtail_factories.PageFactory):
class Meta:
abstract = True
skip_postgeneration_save = True

class Params:
workflow_stages = 1
Expand All @@ -103,6 +104,7 @@ def forms(self, create, extracted, **kwargs):
application=self,
**kwargs,
)
self.save()


class FundTypeFactory(AbstractApplicationFactory):
Expand All @@ -118,6 +120,7 @@ class Meta:
class AbstractRelatedFormFactory(factory.django.DjangoModelFactory):
class Meta:
abstract = True
skip_postgeneration_save = True

form = factory.SubFactory(
"hypha.apply.funds.tests.factories.ApplicationFormFactory"
Expand All @@ -134,6 +137,7 @@ class Meta:
class ApplicationFormFactory(factory.django.DjangoModelFactory):
class Meta:
model = ApplicationForm
skip_postgeneration_save = True

name = factory.Faker("word")
form_fields = blocks.CustomFormFieldsFactory
Expand All @@ -142,6 +146,7 @@ class Meta:
class RoundFactory(wagtail_factories.PageFactory):
class Meta:
model = Round
skip_postgeneration_save = True

class Params:
now = factory.Trait(
Expand Down Expand Up @@ -183,6 +188,7 @@ def forms(self, create, extracted, **kwargs):
round=self,
**kwargs,
)
self.save()


class SealedRoundFactory(RoundFactory):
Expand Down Expand Up @@ -224,6 +230,7 @@ def forms(self, create, extracted, **kwargs):
lab=self,
**kwargs,
)
self.save()


class LabBaseFormFactory(AbstractRelatedFormFactory):
Expand All @@ -240,6 +247,7 @@ class ApplicationFormDataFactory(FormDataFactory):
class ApplicationSubmissionFactory(factory.django.DjangoModelFactory):
class Meta:
model = ApplicationSubmission
skip_postgeneration_save = True

class Params:
workflow_stages = 1
Expand Down Expand Up @@ -277,11 +285,13 @@ def reviewers(self, create, reviewers, **kwargs):
reviewer=reviewer,
submission=self,
)
self.save()


class ReviewerRoleFactory(factory.django.DjangoModelFactory):
class Meta:
model = ReviewerRole
skip_postgeneration_save = True

name = factory.Faker("word")
order = factory.Sequence(lambda n: n)
Expand All @@ -290,6 +300,7 @@ class Meta:
class AssignedReviewersFactory(factory.django.DjangoModelFactory):
class Meta:
model = AssignedReviewers
skip_postgeneration_save = True
django_get_or_create = ("submission", "reviewer")

class Params:
Expand Down Expand Up @@ -342,6 +353,7 @@ class LabSubmissionFactory(ApplicationSubmissionFactory):
class ApplicationRevisionFactory(factory.django.DjangoModelFactory):
class Meta:
model = ApplicationRevision
skip_postgeneration_save = True

submission = factory.SubFactory(
"hypha.apply.funds.tests.factories.ApplicationSubmissionFactory"
Expand All @@ -353,10 +365,16 @@ class Meta:
clean=True,
)

@factory.post_generation
def post(instance: ApplicationRevision, create: bool, extracted, **kwargs):
if create:
instance.save()


class AbstractReviewFormFactory(factory.django.DjangoModelFactory):
class Meta:
abstract = True
skip_postgeneration_save = True

form = factory.SubFactory("hypha.apply.review.tests.factories.ReviewFormFactory")

Expand Down Expand Up @@ -385,6 +403,7 @@ class Meta:
class ScreeningStatusFactory(factory.django.DjangoModelFactory):
class Meta:
model = ScreeningStatus
skip_postgeneration_save = True

title = factory.Iterator(["Bad", "Good"])
yes = factory.Iterator([True, False])
Expand All @@ -393,6 +412,7 @@ class Meta:
class ReminderFactory(factory.django.DjangoModelFactory):
class Meta:
model = Reminder
skip_postgeneration_save = True

submission = factory.SubFactory(
"hypha.apply.funds.tests.factories.ApplicationSubmissionFactory"
Expand Down
4 changes: 4 additions & 0 deletions hypha/apply/projects/reports/tests/factories.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ class ReportConfigFactory(factory.django.DjangoModelFactory):

class Meta:
model = ReportConfig
skip_postgeneration_save = True
django_get_or_create = ("project",)

class Params:
Expand All @@ -43,6 +44,7 @@ class ReportVersionFactory(factory.django.DjangoModelFactory):

class Meta:
model = ReportVersion
skip_postgeneration_save = True

@factory.post_generation
def relate(obj, create, should_relate, **kwargs):
Expand All @@ -56,6 +58,7 @@ def relate(obj, create, should_relate, **kwargs):
obj.report.current = obj
obj.report.submitted = obj.submitted
obj.report.save()
obj.save()


class ReportFactory(factory.django.DjangoModelFactory):
Expand All @@ -72,6 +75,7 @@ class ReportFactory(factory.django.DjangoModelFactory):

class Meta:
model = Report
skip_postgeneration_save = True

class Params:
past_due = factory.Trait(
Expand Down
12 changes: 12 additions & 0 deletions hypha/apply/projects/tests/factories.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,11 +61,13 @@ class DocumentCategoryFactory(factory.django.DjangoModelFactory):

class Meta:
model = DocumentCategory
skip_postgeneration_save = True


class ProjectFormFactory(factory.django.DjangoModelFactory):
class Meta:
model = ProjectForm
skip_postgeneration_save = True

name = factory.Faker("word")
form_fields = FormFieldsBlockFactory
Expand All @@ -74,6 +76,7 @@ class Meta:
class ProjectSOWFormFactory(factory.django.DjangoModelFactory):
class Meta:
model = ProjectSOWForm
skip_postgeneration_save = True

name = factory.Faker("word")
form_fields = FormFieldsBlockFactory
Expand All @@ -86,6 +89,7 @@ class ProjectFormDataFactory(FormDataFactory):
class ProjectReportFormFactory(factory.django.DjangoModelFactory):
class Meta:
model = ProjectReportForm
skip_postgeneration_save = True

name = factory.Faker("word")
form_fields = FormFieldsBlockFactory
Expand All @@ -111,6 +115,7 @@ class ProjectFactory(factory.django.DjangoModelFactory):

class Meta:
model = Project
skip_postgeneration_save = True

class Params:
in_approval = factory.Trait(
Expand All @@ -129,6 +134,7 @@ class PAFReviewerRoleFactory(factory.django.DjangoModelFactory):

class Meta:
model = PAFReviewersRole
skip_postgeneration_save = True

@factory.post_generation
def user_roles(self, create, extracted, **kwargs):
Expand All @@ -137,6 +143,7 @@ def user_roles(self, create, extracted, **kwargs):
GroupFactory(name=STAFF_GROUP_NAME),
GroupFactory(name=APPROVER_GROUP_NAME),
)
self.save()


class PAFApprovalsFactory(factory.django.DjangoModelFactory):
Expand All @@ -146,6 +153,7 @@ class PAFApprovalsFactory(factory.django.DjangoModelFactory):

class Meta:
model = PAFApprovals
skip_postgeneration_save = True


class ContractFactory(factory.django.DjangoModelFactory):
Expand All @@ -158,6 +166,7 @@ class ContractFactory(factory.django.DjangoModelFactory):

class Meta:
model = Contract
skip_postgeneration_save = True


class PacketFileFactory(factory.django.DjangoModelFactory):
Expand All @@ -169,6 +178,7 @@ class PacketFileFactory(factory.django.DjangoModelFactory):

class Meta:
model = PacketFile
skip_postgeneration_save = True


class InvoiceFactory(factory.django.DjangoModelFactory):
Expand All @@ -181,6 +191,7 @@ class InvoiceFactory(factory.django.DjangoModelFactory):

class Meta:
model = Invoice
skip_postgeneration_save = True


class SupportingDocumentFactory(factory.django.DjangoModelFactory):
Expand All @@ -190,6 +201,7 @@ class SupportingDocumentFactory(factory.django.DjangoModelFactory):

class Meta:
model = SupportingDocument
skip_postgeneration_save = True


class ApprovedProjectFactory(ProjectFactory):
Expand Down
3 changes: 3 additions & 0 deletions hypha/apply/review/tests/factories/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ class ReviewFormDataFactory(FormDataFactory):
class ReviewFactory(factory.django.DjangoModelFactory):
class Meta:
model = Review
skip_postgeneration_save = True

class Params:
recommendation_yes = factory.Trait(recommendation=YES)
Expand Down Expand Up @@ -48,6 +49,7 @@ class Params:
class ReviewOpinionFactory(factory.django.DjangoModelFactory):
class Meta:
model = ReviewOpinion
skip_postgeneration_save = True

class Params:
opinion_agree = factory.Trait(opinion=AGREE)
Expand All @@ -65,6 +67,7 @@ class Params:
class ReviewFormFactory(factory.django.DjangoModelFactory):
class Meta:
model = ReviewForm
skip_postgeneration_save = True

name = factory.Faker("word")
form_fields = blocks.ReviewFormFieldsFactory
Loading
Loading