ccvpn/payments/tests/stripe.py

from datetime import timedelta

from django.test import TestCase, RequestFactory
from django.contrib.auth.models import User

from payments.models import Payment, Subscription
from payments.backends import StripeBackend


def MockSession(id):
    class MockSession(dict):
        def __init__(self, args):
            args['id'] = id
            super().__init__(args)

        @classmethod
        def create(cls, **kwargs):
            c = MockSession(kwargs)
            cls._INST = c
            return c
    return MockSession

class MockWebhook:
    @classmethod
    def construct_event(cls, payload, sig, key):
        assert sig == "hewwo", sig
        assert key == "test_wh_key", key
        assert payload == b"opak", payload
        return cls.evt


class StripeBackendTest(TestCase):
    def setUp(self):
        self.user = User.objects.create_user('test', 'test_user@example.com', None)

    def test_stripe(self):
        payment = Payment.objects.create(
            user=self.user,
            time=timedelta(days=30),
            backend_id='stripe',
            amount=300
        )
        payment.save()

        sess_id = 'cs_test_UKHmmetjaiperdu0wxGppeerrdduuhSslYaaaaaa0eU6aaaaaaaaaWSX'

        settings = dict(
            secret_key='test_secret_key',
            public_key='test_public_key',
            wh_key='test_wh_key',
            currency='EUR',
            name='Test Name',
        )

        backend = StripeBackend(settings)
        assert backend.backend_enabled

        backend.stripe = type('Stripe', (object, ), {
            'checkout': type('checkout', (object, ), {
                'Session': MockSession(sess_id),
            }),
            'Webhook': MockWebhook,

            'error': type('error', (object, ), {
                'InvalidRequestError': type('', (Exception, ), {}),
                'SignatureVerificationError': type('', (Exception, ), {}),
                'CardError': type('CardError', (Exception, ), {}),
            }),
        })

        with self.settings(ROOT_URL='root'):
            form_html = backend.new_payment(payment)

        assert 'test_public_key' in form_html

        sess = backend.stripe.checkout.Session._INST
        self.assertEqual(sess['success_url'], 'root/payments/view/%d' % payment.id)
        self.assertEqual(sess['line_items'][0]['amount'], payment.amount)

        request = RequestFactory().post('', 'opak', content_type='text/plain', HTTP_STRIPE_SIGNATURE="hewwo")

        backend.stripe.Webhook.evt = CHECKOUT_SESSION_COMPLETED__ONE
        backend.webhook(request)

        payment = Payment.objects.get(id=payment.id)
        self.assertEqual(payment.status, 'confirmed')
        self.assertEqual(payment.paid_amount, 300)
        self.assertEqual(payment.backend_extid, sess_id)

    def test_stripe_sub(self):
        sub = Subscription.objects.create(
            user=self.user,
            period='3m',
            backend_id='stripe',
        )

        settings = dict(
            secret_key='test_secret_key',
            public_key='test_public_key',
            wh_key='test_wh_key',
            currency='EUR',
            name='Test Name',
        )

        backend = StripeBackend(settings)
        assert backend.backend_enabled

        backend.stripe = type('Stripe', (object, ), {
            'checkout': type('checkout', (object, ), {
                'Session': MockSession("in_FjSMAw7ufaEBOG"),
            }),
            'Webhook': MockWebhook,

            'error': type('error', (object, ), {
                'InvalidRequestError': type('', (Exception, ), {}),
                'SignatureVerificationError': type('', (Exception, ), {}),
                'CardError': type('CardError', (Exception, ), {}),
            }),
        })

        with self.settings(ROOT_URL='root'):
            form_html = backend.new_subscription(sub)

        assert 'test_public_key' in form_html

        sess = backend.stripe.checkout.Session._INST
        self.assertEqual(sess['success_url'], 'root/payments/return_subscr/%d' % sub.id)
        self.assertEqual(sess['subscription_data']['items'][0]['plan'], 'ccvpn_3m')

        request = RequestFactory().post('', 'opak', content_type='text/plain', HTTP_STRIPE_SIGNATURE="hewwo")

        backend.stripe.Webhook.evt = checkout_session_completed__sub(sub)
        backend.webhook(request)
        backend.stripe.Webhook.evt = INVOICE_PAYMENT_SUCCEEDED
        backend.webhook(request)

        sub = Subscription.objects.get(id=sub.id)
        payment = sub.payment_set.all()[0]
        self.assertEqual(sub.status, 'active')
        self.assertEqual(payment.status, 'confirmed')
        self.assertEqual(payment.paid_amount, 900)
        self.assertEqual(payment.backend_extid, 'in_FjSMAw7ufaEBOG')


CHECKOUT_SESSION_COMPLETED__ONE = {
  "id": "evt_FkzwrsBSh0da66",
  "object": "event",
  "api_version": "2019-08-14",
  "created": 1567705834,
  "data": {
    "object": {
      "id": "cs_test_UKHmmetjaiperdu0wxGppeerrdduuhSslYaaaaaa0eU6aaaaaaaaaWSX",
      "object": "checkout.session",
      "billing_address_collection": None,
      "cancel_url": "https://test/payments/cancel/19",
      "client_reference_id": None,
      "customer": "cus_FgusdomerZ8gdP",
      "customer_email": None,
      "display_items": [
        {
          "amount": 300,
          "currency": "eur",
          "custom": {
            "description": "One month for admin",
            "images": None,
            "name": "VPN Payment"
          },
          "quantity": 1,
          "type": "custom"
        }
      ],
      "livemode": False,
      "locale": None,
      "mode": "payment",
      "payment_intent": "pi_0FmmmmmmmmmmmmmmmpyplM53",
      "payment_method_types": [
        "card"
      ],
      "setup_intent": None,
      "submit_type": None,
      "subscription": None,
      "success_url": "https://test/payments/view/19"
    }
  },
  "livemode": False,
  "pending_webhooks": 1,
  "request": {
    "id": None,
    "idempotency_key": None
  },
  "type": "checkout.session.completed"
}

def checkout_session_completed__sub(sub):
    return {
  "id": "evt_FjSIjKT3MvONr8",
  "object": "event",
  "api_version": "2019-08-14",
  "created": 1567368634,
  "data": {
    "object": {
      "id": "cs_test_03aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
      "object": "checkout.session",
      "billing_address_collection": None,
      "cancel_url": "https://test/payments/view/2",
      "client_reference_id": "sub_%d" % sub.id,
      "customer": "cus_FkSIg5BNIdrv6v",
      "customer_email": None,
      "display_items": [
        {
          "amount": 900,
          "currency": "eur",
          "plan": {
            "id": "ccvpn_3m",
            "object": "plan",
            "active": True,
            "aggregate_usage": None,
            "amount": 900,
            "amount_decimal": "900",
            "billing_scheme": "per_unit",
            "created": 1557830755,
            "currency": "eur",
            "interval": "month",
            "interval_count": 3,
            "livemode": False,
            "metadata": {
            },
            "nickname": None,
            "product": "prod_F062eB2kCukpUX",
            "tiers": None,
            "tiers_mode": None,
            "transform_usage": None,
            "trial_period_days": None,
            "usage_type": "licensed"
          },
          "quantity": 1,
          "type": "plan"
        }
      ],
      "livemode": False,
      "locale": None,
      "mode": "subscription",
      "payment_intent": None,
      "payment_method_types": [
        "card"
      ],
      "setup_intent": None,
      "submit_type": None,
      "subscription": "sub_Fjmeowmeowmeow",
      "success_url": "https://test/payments/return/2"
    }
  },
  "livemode": False,
  "pending_webhooks": 1,
  "request": {
    "id": "req_xQY51312421312",
    "idempotency_key": None
  },
  "type": "checkout.session.completed"
}

INVOICE_PAYMENT_SUCCEEDED = {
  "id": "evt_FjSMpT5zIn8G9z",
  "object": "event",
  "api_version": "2019-08-14",
  "created": 1567368905,
  "data": {
    "object": {
      "id": "in_FjSMAw7ufaEBOG",
      "object": "invoice",
      "account_country": "FR",
      "account_name": "Cognitive Cryptography",
      "amount_due": 900,
      "amount_paid": 900,
      "amount_remaining": 0,
      "application_fee_amount": None,
      "attempt_count": 1,
      "attempted": True,
      "auto_advance": False,
      "billing": "charge_automatically",
      "billing_reason": "subscription_create",
      "charge": "ch_FjSMFp6fbsc45t",
      "collection_method": "charge_automatically",
      "created": 1567368904,
      "currency": "eur",
      "custom_fields": None,
      "customer": "cus_FjSMgbAnXmL4Go",
      "customer_address": None,
      "customer_email": "test2@test.test",
      "customer_name": None,
      "customer_phone": None,
      "customer_shipping": None,
      "customer_tax_exempt": "none",
      "customer_tax_ids": [
      ],
      "default_payment_method": None,
      "default_source": None,
      "default_tax_rates": [
      ],
      "description": None,
      "discount": None,
      "due_date": None,
      "ending_balance": 0,
      "footer": None,
      "hosted_invoice_url": "https://pay.stripe.com/invoice/invst_wmhbGo7OofOofOofOofOofOofd",
      "invoice_pdf": "https://pay.stripe.com/invoice/invst_wmhbGo7OofOofOofOofOofOofd/pdf",
      "lines": {
        "object": "list",
        "data": [
          {
            "id": "sli_c20baf5dcf8472",
            "object": "line_item",
            "amount": 900,
            "currency": "eur",
            "description": "1 × VPN Subscription (3m) (at €9.00 / every 3 months)",
            "discountable": True,
            "livemode": False,
            "metadata": {
            },
            "period": {
              "end": 1575231304,
              "start": 1567368904
            },
            "plan": {
              "id": "ccvpn_3m",
              "object": "plan",
              "active": True,
              "aggregate_usage": None,
              "amount": 900,
              "amount_decimal": "900",
              "billing_scheme": "per_unit",
              "created": 1557830755,
              "currency": "eur",
              "interval": "month",
              "interval_count": 3,
              "livemode": False,
              "metadata": {
              },
              "nickname": None,
              "product": "prod_F46LeBpkCuRpUX",
              "tiers": None,
              "tiers_mode": None,
              "transform_usage": None,
              "trial_period_days": None,
              "usage_type": "licensed"
            },
            "proration": False,
            "quantity": 1,
            "subscription": "sub_Fjmeowmeowmeow",
            "subscription_item": "si_FjSMgsAsZURoWf",
            "tax_amounts": [
            ],
            "tax_rates": [
            ],
            "type": "subscription"
          }
        ],
        "has_more": False,
        "total_count": 1,
        "url": "/v1/invoices/in_FjSMAw7ufaEBOG/lines"
      },
      "livemode": False,
      "metadata": {
      },
      "next_payment_attempt": None,
      "number": "AD4E509A-0001",
      "paid": True,
      "payment_intent": "pi_0FDzS0UmBGT3Of2k4ol85K2G",
      "period_end": 1567368904,
      "period_start": 1567368904,
      "post_payment_credit_notes_amount": 0,
      "pre_payment_credit_notes_amount": 0,
      "receipt_number": None,
      "starting_balance": 0,
      "statement_descriptor": None,
      "status": "paid",
      "status_transitions": {
        "finalized_at": 1567368904,
        "marked_uncollectible_at": None,
        "paid_at": 1567368905,
        "voided_at": None
      },
      "subscription": "sub_Fjmeowmeowmeow",
      "subtotal": 900,
      "tax": None,
      "tax_percent": None,
      "total": 900,
      "total_tax_amounts": [
      ],
      "webhooks_delivered_at": None
    }
  },
  "livemode": False,
  "pending_webhooks": 1,
  "request": {
    "id": "req_6O128256512102",
    "idempotency_key": None
  },
  "type": "invoice.payment_succeeded"
}