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
from pyblackbird_cc.payments.models import SubscriptionPlan

User = get_user_model()


@pytest.fixture
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
    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