class PortfolioInvitationFactory(Base): class Meta: model = PortfolioInvitation email = factory.Faker("email") status = InvitationStatus.PENDING expiration_time = PortfolioInvitations.current_expiration_time()
class ApplicationInvitationFactory(Base): class Meta: model = ApplicationInvitation email = factory.Faker("email") status = InvitationStatus.PENDING expiration_time = PortfolioInvitations.current_expiration_time() role = factory.SubFactory(ApplicationRoleFactory)
class PortfolioInvitationFactory(Base): class Meta: model = PortfolioInvitation email = factory.Faker("email") status = InvitationStatus.PENDING expiration_time = PortfolioInvitations.current_expiration_time() dod_id = factory.LazyFunction(random_dod_id) first_name = factory.Faker("first_name") last_name = factory.Faker("last_name") @classmethod def _create(cls, model_class, *args, **kwargs): inviter_id = kwargs.pop("inviter_id", UserFactory.create().id) return super()._create(model_class, inviter_id=inviter_id, *args, **kwargs)