Test coverage for paypal.

diff --git a/src/paypal/tests.py b/src/paypal/tests.py
index 7522842..e38fd84 100644 (file)
--- a/src/paypal/tests.py
+++ b/src/paypal/tests.py
@@ -3,14 +3,34 @@
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 from django.contrib.auth.models import User
-from mock import Mock, patch, DEFAULT
+from mock import MagicMock, Mock, patch, DEFAULT
 from catalogue.test_utils import WLTestCase
-from .models import BillingPlan
+from .models import BillingAgreement, BillingPlan
+from .rest import user_is_subscribed
+from paypalrestsdk import ResourceNotFound
 
 
+BillingPlanMock = Mock(
+    return_value=Mock(
+        id='some-billing-plan-id'
+    )
+)
+
 BillingAgreementMock = Mock(
+    # BillingAgreement() has a .links[]
+    return_value=MagicMock(
+        links=[
+            Mock(
+                rel="approval_url",
+                href="http://paypal.test/approval/"
+            )
+        ]
+    ),
+    # BillingAgreement.execute(token)
     execute=Mock(
         return_value=Mock(
+            error=None,
+            id='some-billing-agreement-id',
             plan=Mock(
                 payment_definitions=[
                     Mock(
@@ -19,7 +39,13 @@ BillingAgreementMock = Mock(
                 ]
             )
         )
-    )
+    ),
+    # Later we can BillingAgreement.find(...).state == 'Active'
+    find=Mock(
+        return_value=Mock(
+            state='Active'
+        )
+    ),
 )
 
 
@@ -54,32 +80,129 @@ class PaypalTests(WLTestCase):
             1)
 
     @patch.multiple('paypalrestsdk',
-        BillingPlan=DEFAULT,
-        BillingAgreement=DEFAULT
+        BillingPlan=BillingPlanMock,
+        BillingAgreement=BillingAgreementMock,
     )
-    def test_paypal_form_valid(self, BillingPlan, BillingAgreement):
+    def test_paypal_form_valid(self):
+        """PayPal form created a BillingPlan."""
         self.client.login(username='test', password='test')
         response = self.client.post('/paypal/form/', {"amount": "100"})
-        self.assertEqual(response.status_code, 302)
-        # Assert: BillingPlan created? BillingAgreement created?
-        # Models created?
+        self.assertRedirects(response, 'http://paypal.test/approval/',
+            fetch_redirect_response=False)
+        self.assertEqual(BillingPlan.objects.all().count(), 1)
+
+        # Posting the form a second time does not create another plan.
+        response = self.client.post('/paypal/form/', {"amount": "100"})
+        self.assertRedirects(response, 'http://paypal.test/approval/',
+            fetch_redirect_response=False)
+        self.assertEqual(BillingPlan.objects.all().count(), 1)
+
+        # No BillingAgreement created in our DB yet.
+        self.assertEqual(BillingAgreement.objects.all().count(), 0)
+
+    @patch('paypalrestsdk.BillingPlan', BillingPlanMock)
+    def test_paypal_form_error(self):
+        """On PayPal error, plan does not get created."""
+        self.client.login(username='test', password='test')
+
+        # It can choke on BillingPlan().create().
+        with patch('paypalrestsdk.BillingPlan', Mock(
+                return_value=Mock(create=Mock(return_value=None)))):
+            response = self.client.post('/paypal/form/', {"amount": "100"})
+            self.assertEqual(response.status_code, 200)
+
+        # Or it can choke on BillingPlan().activate().
+        with patch('paypalrestsdk.BillingPlan', Mock(
+                return_value=Mock(activate=Mock(return_value=None)))):
+            response = self.client.post('/paypal/form/', {"amount": "100"})
+            self.assertEqual(response.status_code, 200)
+
+        # No plan is created yet.
+        self.assertEqual(BillingPlan.objects.all().count(), 0)
+
+        # Or it can choke later, on BillingAgreement().create()
+        with patch('paypalrestsdk.BillingAgreement', Mock(
+                return_value=Mock(create=Mock(return_value=None)))):
+            response = self.client.post('/paypal/form/', {"amount": "100"})
+            self.assertEqual(response.status_code, 200)
+
+        # But now the plan should be created.
+        self.assertEqual(BillingPlan.objects.all().count(), 1)
 
     @patch.multiple('paypalrestsdk',
-        BillingPlan=DEFAULT,
-        BillingAgreement=DEFAULT,
+        BillingPlan=BillingPlanMock,
+        BillingAgreement=BillingAgreementMock,
     )
-    def test_paypal_form_valid(self, BillingPlan, BillingAgreement):
+    def test_paypal_app_form_valid(self):
+        """App form creates a BillingPlan."""
         self.client.login(username='test', password='test')
         response = self.client.post('/paypal/app-form/', {"amount": "100"})
-        self.assertEqual(response.status_code, 302)
+        self.assertRedirects(response, 'http://paypal.test/approval/',
+            fetch_redirect_response=False)
+        self.assertEqual(BillingPlan.objects.all().count(), 1)
 
-    @patch.multiple('paypalrestsdk',
-        BillingAgreement=BillingAgreementMock
-    )
+    @patch('paypalrestsdk.BillingAgreement', BillingAgreementMock)
     def test_paypal_return(self):
         self.client.login(username='test', password='test')
         BillingPlan.objects.create(amount=100)
+
+        # No token = no agreement.
+        response = self.client.get('/paypal/return/')
+        self.assertEqual(response.status_code, 404)
+        self.assertEqual(BillingAgreement.objects.all().count(), 0)
+
         response = self.client.get('/paypal/return/?token=secret-token')
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(BillingAgreement.objects.all().count(), 1)
+
+        # Repeated returns will not generate further agreements.
+        response = self.client.get('/paypal/return/?token=secret-token')
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(BillingAgreement.objects.all().count(), 1)
+
+        self.assertTrue(user_is_subscribed(self.user))
+
+    @patch('paypalrestsdk.BillingAgreement', BillingAgreementMock)
+    def test_paypal_app_return(self):
+        self.client.login(username='test', password='test')
+        BillingPlan.objects.create(amount=100)
+        response = self.client.get('/paypal/app-return/?token=secret-token')
+        self.assertRedirects(response, 'wolnelekturyapp://paypal_return')
+
+        # Repeated returns will not generate further agreements.
+        response = self.client.get('/paypal/app-return/?token=secret-token')
+        self.assertRedirects(response, 'wolnelekturyapp://paypal_return')
+        self.assertEqual(BillingAgreement.objects.all().count(), 1)
+
+        self.assertTrue(user_is_subscribed(self.user))
+
+    def test_paypal_return_error(self):
+        self.client.login(username='test', password='test')
+        BillingPlan.objects.create(amount=100)
+
+        # It can choke on BillingAgreement.execute()
+        with patch('paypalrestsdk.BillingAgreement', Mock(
+                execute=Mock(return_value=Mock(id=None)))):
+            self.client.get('/paypal/app-return/?token=secret-token')
+            response = self.client.get('/paypal/app-return/?token=secret-token')
+            self.assertRedirects(response, 'wolnelekturyapp://paypal_error')
+
+        # No agreement created in our DB if not executed successfully.
+        self.assertEqual(BillingAgreement.objects.all().count(), 0)
+
+        # It can execute all right, but just not be findable later.
+        with patch('paypalrestsdk.BillingAgreement', Mock(
+                execute=BillingAgreementMock.execute,
+                find=Mock(side_effect=ResourceNotFound(None)))):
+            response = self.client.get('/paypal/app-return/?token=secret-token')
+            self.assertRedirects(response, 'wolnelekturyapp://paypal_return')
+
+        # Now the agreement exists in our DB, but is not active.
+        self.assertEqual([b.active for b in BillingAgreement.objects.all()], [False])
+
+        with patch('paypalrestsdk.BillingAgreement', Mock(
+                find=Mock(return_value=Mock(state='Mocked')))):
+            self.assertFalse(user_is_subscribed(self.user))
 
     def test_paypal_cancel(self):
         response = self.client.get('/paypal/cancel/')

diff --git a/src/paypal/views.py b/src/paypal/views.py
index 15a7cf6..e46be8c 100644 (file)
--- a/src/paypal/views.py
+++ b/src/paypal/views.py
@@ -42,13 +42,13 @@ def paypal_return(request, app=False):
         if resource.id:
             amount = int(Decimal(resource.plan.payment_definitions[0].amount['value']))
             plan = BillingPlan.objects.get(amount=amount)
-            active = check_agreement(resource.id)
+            active = check_agreement(resource.id) or False
             BillingAgreement.objects.create(
                 agreement_id=resource.id, user=request.user, plan=plan, active=active, token=token)
     else:
         resource = None
     if app:
-        if getattr(resource, 'error'):
+        if getattr(resource, 'error', None):
             return HttpResponseAppRedirect('wolnelekturyapp://paypal_error')
         else:
             return HttpResponseAppRedirect('wolnelekturyapp://paypal_return')