def update_payment_method(user, payment_method_id, address, credit_card, validate=False): profile_id = CustomerID.get_id(user._id) payment_method_id = api.update_payment_profile(profile_id, payment_method_id, address, credit_card, validate) return payment_method_id
def test_update_payment_profile(self, PaymentProfile, UpdateRequest): _request = Mock() _request.make_request.return_value = self.payment_profile_id UpdateRequest.return_value = _request # Scenario: call is successful return_value = update_payment_profile(self.customer_id, self.payment_profile_id, 'address', 1234) self.assertTrue(UpdateRequest.called) self.assertTrue(_request.make_request.called) self.assertEqual(return_value, self.payment_profile_id) # Scenario: call raises AuthorizeNetException _request.make_request.side_effect = AuthorizeNetException('') self.assertRaises(AuthorizeNetException, update_payment_profile, self.customer_id, self.payment_profile_id, 'address', 1234)