Add default settings to let new users be subscribed on trial

- Add the default user to a subscription trial
- Update associated unit tests
This commit is contained in:
sabaimran
2023-11-10 22:38:28 -08:00
parent 0a950d9382
commit e2e96f9aa4
9 changed files with 86 additions and 11 deletions

View File

@@ -9,6 +9,7 @@ from database.models import (
OpenAIProcessorConversationConfig,
UserConversationConfig,
Conversation,
Subscription,
)
@@ -68,3 +69,12 @@ class ConversationFactory(factory.django.DjangoModelFactory):
model = Conversation
user = factory.SubFactory(UserFactory)
class SubscriptionFactory(factory.django.DjangoModelFactory):
class Meta:
model = Subscription
user = factory.SubFactory(UserFactory)
type = "trial"
is_recurring = False