From 7550f31f604fa45f9030a32b9b8fbc2d0739bed7 Mon Sep 17 00:00:00 2001 From: Matthew Lemon Date: Mon, 9 Sep 2024 17:24:53 +0100 Subject: subscription and user is now uniquely constrained --- pyblackbird_cc/payments/tests/test_models.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) (limited to 'pyblackbird_cc/payments/tests/test_models.py') diff --git a/pyblackbird_cc/payments/tests/test_models.py b/pyblackbird_cc/payments/tests/test_models.py index 132d025..467d6e4 100644 --- a/pyblackbird_cc/payments/tests/test_models.py +++ b/pyblackbird_cc/payments/tests/test_models.py @@ -1,6 +1,7 @@ import pytest from allauth.account.signals import user_signed_up from django.contrib.auth import get_user_model +from django.db import IntegrityError from django.test import RequestFactory from pyblackbird_cc.payments.models import Subscription @@ -14,6 +15,30 @@ def user_data(): return {"email": "testuser@example.com", "password": "testpassword123"} +@pytest.mark.django_db +def test_subscription_user_unique(): + # 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_data = {"email": "testuser@example.com", "password": "testpassword123"} + user = User.objects.create_user(**user_data) + + # Create a subscription for the user + subscription = Subscription.objects.create(user=user, plan=free_plan) + + # Try to create another subscription for the same user + with pytest.raises(IntegrityError): + Subscription.objects.create(user=user, plan=free_plan) + + @pytest.mark.django_db def test_user_signup_assigns_free_subscription(user_data): # Ensure the free plan exists -- cgit v1.2.3