Mock Stripe Metody Pythona do testowania

Question

Więc staram się wyśmiewać wszystko stripe web hooks w sposób tak, że mogę napisać Unit test dla niego. Używam mock library dla wyśmianie metody paskiem. Oto metoda Próbuję mock:

class AddCardView(APIView): 
""" 
* Add card for the customer 
""" 

permission_classes = (
    CustomerPermission, 
) 

def post(self, request, format=None): 
    name = request.DATA.get('name', None) 
    cvc = request.DATA.get('cvc', None) 
    number = request.DATA.get('number', None) 
    expiry = request.DATA.get('expiry', None) 

    expiry_month, expiry_year = expiry.split("/") 

    customer_obj = request.user.contact.business.customer 

    customer = stripe.Customer.retrieve(customer_obj.stripe_id) 

    try: 
     card = customer.sources.create(
      source={ 
       "object": "card", 
       "number": number, 
       "exp_month": expiry_month, 
       "exp_year": expiry_year, 
       "cvc": cvc, 
       "name": name 
      } 
     ) 
     # making it the default card 
     customer.default_source = card.id 
     customer.save() 
    except CardError as ce: 
     logger.error("Got CardError for customer_id={0}, CardError={1}".format(customer_obj.pk, ce.json_body)) 
     return Response({"success": False, "error": "Failed to add card"}) 
    else: 
     customer_obj.card_last_4 = card.get('last4') 
     customer_obj.card_kind = card.get('type', '') 
     customer_obj.card_fingerprint = card.get('fingerprint') 
     customer_obj.save() 

    return Response({"success": True}) 

Jest to metoda dla unit testing:

@mock.patch('stripe.Customer.retrieve') 
@mock.patch('stripe.Customer.create') 
def test_add_card(self,create_mock,retrieve_mock): 
    response = { 
     'default_card': None, 
     'cards': { 
      "count": 0, 
      "data": [] 
     } 
    } 

    # save_mock.return_value = response 
    create_mock.return_value = response 
    retrieve_mock.return_value = response 

    self.api_client.client.login(username = self.username, password = self.password) 
    res = self.api_client.post('/biz/api/auth/card/add') 

    print res 

Teraz stripe.Customer.retrieve jest właściwie szydzili. Ale nie jestem w stanie drwić customer.sources.create. Ja naprawdę zatrzymany w tej sprawie.

Answer 1

Jest to właściwy sposób to zrobić:

@mock.patch('stripe.Customer.retrieve') 
def test_add_card_failure(self, retrieve_mock): 
    data = { 
     'name': "shubham", 
     'cvc': 123, 
     'number': "4242424242424242", 
     'expiry': "12/23", 
    } 
    e = CardError("Card Error", "", "") 
    retrieve_mock.return_value.sources.create.return_value = e 

    self.api_client.client.login(username=self.username, password=self.password) 

    res = self.api_client.post('/biz/api/auth/card/add', data=data) 

    self.assertEqual(self.deserialize(res)['success'], False) 

Answer 2

Nawet jeśli dana odpowiedź jest poprawna, nie jest to sposób bardziej komfortowe rozwiązanie wykorzystujące vcrpy. Że tworzy kasety (rekord) raz dany rekord jeszcze nie istnieje. Kiedy to robi, szyderczy odbywa się w sposób przejrzysty i płyta zostanie odtworzona. Piękny.

uwzględniając wniosek wanilia piramida, używając py.test, mój testu teraz wygląda tak:

import vcr 
# here we have some FactoryBoy fixtures 
from tests.fixtures import PaymentServiceProviderFactory, SSOUserFactory 

def test_post_transaction(sqla_session, test_app): 
    # first we need a PSP and a User existent in the DB 
    psp = PaymentServiceProviderFactory() # type: PaymentServiceProvider 
    user = SSOUserFactory() 
    sqla_session.add(psp, user) 
    sqla_session.flush() 

    with vcr.use_cassette('tests/casettes/tests.checkout.services.transaction_test.test_post_transaction.yaml'): 
     # with that PSP we create a new PSPTransaction ... 
     res = test_app.post(url='/psps/%s/transaction' % psp.id, 
          params={ 
           'token': '4711', 
           'amount': '12.44', 
           'currency': 'EUR', 
          }) 
     assert 201 == res.status_code 
     assert 'id' in res.json_body