Make Stripe Just Work™

2 files changed, 2 insertions, 5 deletions

diff --git a/payments/stripe/tests.py b/payments/stripe/tests.py
index b757e78..6d17158 100644
--- a/payments/stripe/tests.py
+++ b/payments/stripe/tests.py
@@ -47,4 +47,4 @@ class TestStripeProvider(TestCase):
             secret_key='123', public_key='abc')
         form = provider.get_form()
         self.assertTrue(
-            '<script data-amount="10000" data-currency="USD" data-description="payment" data-image="" data-key="abc" data-name="Example.com store" src="https://checkout.stripe.com/checkout.js"></script>' in str(form))
+            '<script class="stripe-button" data-amount="10000" data-currency="USD" data-description="payment" data-image="" data-key="abc" data-name="Example.com store" src="https://checkout.stripe.com/checkout.js"></script>' in str(form))
diff --git a/payments/stripe/widgets.py b/payments/stripe/widgets.py
index 8ef0299..e8774a2 100644
--- a/payments/stripe/widgets.py
+++ b/payments/stripe/widgets.py
@@ -13,7 +13,7 @@ class StripeWidget(Input):
     def __init__(self, provider, payment, *args, **kwargs):
         attrs = kwargs.get('attrs', {})
         kwargs['attrs'] = {
-            'id': 'stripe-id',
+            'class': 'stripe-button',
             'data-key': provider.public_key,
             'data-image': provider.image,
             'data-name': provider.name,
@@ -35,6 +35,3 @@ class StripeWidget(Input):
             # Only add the 'value' attribute if a value is non-empty.
             final_attrs['value'] = force_text(self._format_value(value))
         return format_html('<script{0}></script>', flatatt(final_attrs))
-
-    class Media:
-        js = ['https://js.stripe.com/v2/']