blob: 342ff14d03ed8a46f1f96d16762d067a90940661 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
import pytest
from allauth.account.signals import user_signed_up
from django.contrib.auth import get_user_model
from django.test import RequestFactory
from pyblackbird_cc.payments.models import SubscriptionPlan, Subscription
User = get_user_model()
@pytest.fixture
def user_data():
return {
'email': 'testuser@example.com',
'password': 'testpassword123'
}
@pytest.mark.django_db
def test_user_signup_assigns_free_subscription(user_data):
# Ensure the free plan exists
free_plan, _ = SubscriptionPlan.objects.get_or_create(
name="Free Plan",
defaults={
"price": 0,
"description": "Free plan description",
"allowed_downloads": 10,
}
) # Create a new user
user = User.objects.create_user(**user_data)
# Manually trigger the user_signed_up signal
request = RequestFactory().get('/')
user_signed_up.send(sender=user.__class__, request=request, user=user)
# Check if a SubscriptionPlan was created for the user
subscription = Subscription.objects.filter(user=user).first()
assert subscription is not None
# Check if the assigned plan is the free plan
assert subscription.plan == free_plan
# Additional assertions can be added here to check other properties
# of the SubscriptionPlan or Subscription as needed
|
