fix test

Author: suejung-sentry

billing/tests/test_views.py

@@ -1584,14 +1584,21 @@ def test_has_unverified_initial_payment_method_no_payment_intent(
     def test_has_unverified_initial_payment_method_payment_intent_succeeded(
         self, invoice_retrieve_mock, payment_intent_retrieve_mock
     ):
-        subscription = Mock()
-        subscription.latest_invoice = "inv_123"
-
-        class MockPaymentIntent:
-            status = "succeeded"
+        subscription = stripe.Subscription.construct_from(
+            {"latest_invoice": "inv_123"},
+            "sub_123"
+        )
 
-        invoice_retrieve_mock.return_value = Mock(payment_intent="pi_123")
-        payment_intent_retrieve_mock.return_value = MockPaymentIntent()
+        invoice_retrieve_mock.return_value = stripe.Invoice.construct_from(
+            {"payment_intent": "pi_123"},
+            "inv_123"
+        )
+        payment_intent_retrieve_mock.return_value = stripe.PaymentIntent.construct_from(
+            {
+                "status": "succeeded"
+            },
+            "payment_intent_asdf"
+        )
 
         handler = StripeWebhookHandler()
         result = handler._has_unverified_initial_payment_method(subscription)