|
1 | 1 | from unittest import mock |
2 | 2 |
|
3 | 3 | import stripe |
| 4 | +from django.contrib.sites.models import Site |
4 | 5 | from django.test import override_settings |
5 | 6 | from django.urls import reverse |
6 | 7 |
|
@@ -57,6 +58,33 @@ def test_rejects_user_with_active_subscription(self, mock_create): |
57 | 58 | self.assertEqual(response.status_code, 400) |
58 | 59 | mock_create.assert_not_called() |
59 | 60 |
|
| 61 | + @override_settings(SITE_ID=1) |
| 62 | + @mock.patch("contentcuration.views.subscription.stripe.checkout.Session.create") |
| 63 | + def test_checkout_urls_use_canonical_site_domain(self, mock_create): |
| 64 | + Site.objects.update_or_create( |
| 65 | + pk=1, defaults={"domain": "studio.learningequality.org", "name": "Studio"} |
| 66 | + ) |
| 67 | + mock_create.return_value = mock.Mock(url="https://checkout.stripe.com/test") |
| 68 | + |
| 69 | + self.client.force_authenticate(self.user) |
| 70 | + response = self.client.post( |
| 71 | + self.url, |
| 72 | + data={"storage_gb": 10}, |
| 73 | + format="json", |
| 74 | + HTTP_HOST="master.studio.learningequality.org", |
| 75 | + ) |
| 76 | + |
| 77 | + self.assertEqual(response.status_code, 200) |
| 78 | + call_kwargs = mock_create.call_args[1] |
| 79 | + for key in ("success_url", "cancel_url"): |
| 80 | + url = call_kwargs[key] |
| 81 | + self.assertNotIn( |
| 82 | + "master.studio.learningequality.org", |
| 83 | + url, |
| 84 | + f"{key} leaked internal hostname: {url}", |
| 85 | + ) |
| 86 | + self.assertIn("studio.learningequality.org", url) |
| 87 | + |
60 | 88 | @mock.patch("contentcuration.views.subscription.stripe.checkout.Session.create") |
61 | 89 | def test_user_with_canceled_subscription_can_checkout_again(self, mock_create): |
62 | 90 | """User whose subscription was canceled can create a new checkout session.""" |
@@ -124,6 +152,35 @@ def test_creates_portal_session(self, mock_create): |
124 | 152 | data = response.json() |
125 | 153 | self.assertEqual(data["portal_url"], "https://billing.stripe.com/test") |
126 | 154 |
|
| 155 | + @override_settings(SITE_ID=1) |
| 156 | + @mock.patch( |
| 157 | + "contentcuration.views.subscription.stripe.billing_portal.Session.create" |
| 158 | + ) |
| 159 | + def test_portal_return_url_uses_canonical_site_domain(self, mock_create): |
| 160 | + Site.objects.update_or_create( |
| 161 | + pk=1, defaults={"domain": "studio.learningequality.org", "name": "Studio"} |
| 162 | + ) |
| 163 | + UserSubscription.objects.create( |
| 164 | + user=self.user, |
| 165 | + stripe_customer_id="cus_test123", |
| 166 | + stripe_subscription_status="active", |
| 167 | + ) |
| 168 | + mock_create.return_value = mock.Mock(url="https://billing.stripe.com/test") |
| 169 | + |
| 170 | + self.client.force_authenticate(self.user) |
| 171 | + response = self.client.post( |
| 172 | + self.url, HTTP_HOST="master.studio.learningequality.org" |
| 173 | + ) |
| 174 | + |
| 175 | + self.assertEqual(response.status_code, 200) |
| 176 | + return_url = mock_create.call_args[1]["return_url"] |
| 177 | + self.assertNotIn( |
| 178 | + "master.studio.learningequality.org", |
| 179 | + return_url, |
| 180 | + f"return_url leaked internal hostname: {return_url}", |
| 181 | + ) |
| 182 | + self.assertIn("studio.learningequality.org", return_url) |
| 183 | + |
127 | 184 |
|
128 | 185 | @override_settings( |
129 | 186 | STRIPE_SECRET_KEY="sk_test_fake", |
|
0 commit comments