4 files changed, 141 insertions, 0 deletions
diff --git a/alphabetlearning/payments/tests/__init__.py b/alphabetlearning/payments/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/alphabetlearning/payments/tests/__init__.py
diff --git a/alphabetlearning/payments/tests/conftest.py b/alphabetlearning/payments/tests/conftest.py
new file mode 100644
index 0000000..7919170
--- /dev/null
+++ b/alphabetlearning/payments/tests/conftest.py
@@ -0,0 +1,6 @@
+import pytest
+
+
+@pytest.fixture
+def user_data():
+    return {"email": "testuser@example.com", "password": "testpassword123"}
diff --git a/alphabetlearning/payments/tests/test_models.py b/alphabetlearning/payments/tests/test_models.py
new file mode 100644
index 0000000..8c509f5
--- /dev/null
+++ b/alphabetlearning/payments/tests/test_models.py
@@ -0,0 +1,116 @@
+import pytest
+from allauth.account.signals import user_signed_up
+from django.contrib.auth import get_user_model
+from django.core.exceptions import ValidationError
+from django.db import IntegrityError
+from django.test import RequestFactory
+
+from alphabetlearning.payments.models import ShoppingCart
+from alphabetlearning.payments.models import Subscription
+from alphabetlearning.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)  # type: ignore
+
+    # Create a subscription for the user
+    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)  # type: ignore
+    # 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 = user.subscription
+    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
+
+
+@pytest.mark.django_db
+def test_shopping_cart_is_created_when_user_is_created(user_data):
+    user = User.objects.create_user(**user_data)  # type: ignore
+    request = RequestFactory().get("/")
+    user_signed_up.send(sender=user.__class__, request=request, user=user)
+    shopping_cart = ShoppingCart.objects.get(user=user)
+    assert shopping_cart is not None
+    assert str(shopping_cart) == "Shopping cart for {}".format(user.email)
+
+
+# When the user adds a `Resource` to their cart, create a new `CartItem` instance and associate it with the user's `ShoppingCart` and the selected `Resource`.
+
+
+@pytest.mark.django_db
+def test_cart_item_is_created_when_resource_is_added_to_cart(user_data, resource):
+    user = User.objects.create_user(**user_data)  # type: ignore
+    request = RequestFactory().get("/")
+    user_signed_up.send(sender=user.__class__, request=request, user=user)
+    users_cart = ShoppingCart.objects.get(user=user)
+    users_cart.add_resource(resource)
+    assert users_cart.items.count() == 1  # type: ignore
+
+
+@pytest.mark.django_db
+def test_cannot_add_the_same_resource_to_cart_twice(user_data, resource):
+    user = User.objects.create_user(**user_data)  # type: ignore
+    request = RequestFactory().get("/")
+    user_signed_up.send(sender=user.__class__, request=request, user=user)
+    users_cart = ShoppingCart.objects.get(user=user)
+    users_cart.add_resource(resource)
+    assert users_cart.items.count() == 1  # type: ignore
+    with pytest.raises(ValidationError):
+        users_cart.add_resource(resource)
+
+
+@pytest.mark.django_db
+def test_can_add_multiple_different_items_to_cart(user_data, resources):
+    user = User.objects.create_user(**user_data)  # type: ignore
+    request = RequestFactory().get("/")
+    user_signed_up.send(sender=user.__class__, request=request, user=user)
+    users_cart = ShoppingCart.objects.get(user=user)
+    users_cart.add_resource(resources[0])
+    users_cart.add_resource(resources[1])
+    users_cart.add_resource(resources[2])
+    users_cart.add_resource(resources[3])
+    assert users_cart.items.count() == 4  # type: ignore
diff --git a/alphabetlearning/payments/tests/test_views.py b/alphabetlearning/payments/tests/test_views.py
new file mode 100644
index 0000000..7d0f8b5
--- /dev/null
+++ b/alphabetlearning/payments/tests/test_views.py
@@ -0,0 +1,19 @@
+import pytest
+from django.urls import reverse
+
+
+@pytest.mark.django_db
+def test_cart_view(client, user):
+    url = reverse("payments:cart_detail")
+    client.force_login(user)
+    response = client.get(url)
+    assert response.status_code == 200
+    assert "My basket" in str(response.content)
+
+
+@pytest.mark.django_db
+def test_add_resource_to_cart(client, resource, user):
+    url = reverse("payments:add_to_cart", kwargs={"resource_id": resource.id})
+    client.force_login(user)
+    response = client.get(url)
+    assert response.status_code == 200