378 lines
10 KiB
ReStructuredText
378 lines
10 KiB
ReStructuredText
===============================
|
|
Account Payment Stripe Scenario
|
|
===============================
|
|
|
|
Imports::
|
|
|
|
>>> import datetime as dt
|
|
>>> import os
|
|
>>> import time
|
|
>>> from decimal import Decimal
|
|
|
|
>>> import stripe
|
|
|
|
>>> from proteus import Model
|
|
>>> from trytond.modules.account.tests.tools import create_chart, create_fiscalyear
|
|
>>> from trytond.modules.company.tests.tools import create_company, get_company
|
|
>>> from trytond.tests.tools import activate_modules
|
|
|
|
>>> today = dt.date.today()
|
|
|
|
>>> FETCH_SLEEP, MAX_SLEEP = 1, 100
|
|
|
|
Activate modules::
|
|
|
|
>>> config = activate_modules(
|
|
... 'account_payment_stripe', create_company, create_chart)
|
|
|
|
Create fiscal year::
|
|
|
|
>>> fiscalyear = create_fiscalyear(today=today)
|
|
>>> fiscalyear.click('create_period')
|
|
|
|
Create Stripe account::
|
|
|
|
>>> StripeAccount = Model.get('account.payment.stripe.account')
|
|
>>> stripe_account = StripeAccount(name="Stripe")
|
|
>>> stripe_account.secret_key = os.getenv('STRIPE_SECRET_KEY')
|
|
>>> stripe_account.publishable_key = os.getenv('STRIPE_PUBLISHABLE_KEY')
|
|
>>> stripe_account.save()
|
|
>>> stripe.api_key = os.getenv('STRIPE_SECRET_KEY')
|
|
|
|
Create webhook identifier::
|
|
|
|
>>> stripe_account.click('new_identifier')
|
|
>>> len(stripe_account.webhook_identifier)
|
|
32
|
|
|
|
Remove webhook::
|
|
|
|
>>> stripe_account.click('new_identifier')
|
|
>>> stripe_account.webhook_identifier
|
|
|
|
Setup fetch events cron::
|
|
|
|
>>> Cron = Model.get('ir.cron')
|
|
>>> cron_fetch_events, = Cron.find([
|
|
... ('method', '=', 'account.payment.stripe.account|fetch_events'),
|
|
... ])
|
|
>>> cron_fetch_events.companies.append(get_company())
|
|
|
|
Create payment journal::
|
|
|
|
>>> PaymentJournal = Model.get('account.payment.journal')
|
|
>>> payment_journal = PaymentJournal(name="Stripe",
|
|
... process_method='stripe', stripe_account=stripe_account)
|
|
>>> payment_journal.save()
|
|
|
|
Create party::
|
|
|
|
>>> Party = Model.get('party.party')
|
|
>>> Lang = Model.get('ir.lang')
|
|
>>> customer = Party(name='Customer')
|
|
>>> customer.lang, = Lang.find([('code', '=', 'en')])
|
|
>>> customer.save()
|
|
|
|
Create submitted payment::
|
|
|
|
>>> Payment = Model.get('account.payment')
|
|
>>> payment = Payment()
|
|
>>> payment.journal = payment_journal
|
|
>>> payment.kind = 'receivable'
|
|
>>> payment.party = customer
|
|
>>> payment.amount = Decimal('42')
|
|
>>> payment.reference = 'Testing'
|
|
>>> payment.click('submit')
|
|
>>> payment.state
|
|
'submitted'
|
|
|
|
Checkout the payment::
|
|
|
|
>>> checkout = payment.click('stripe_checkout')
|
|
>>> bool(payment.stripe_checkout_id)
|
|
True
|
|
|
|
>>> token = stripe.Token.create(
|
|
... card={
|
|
... 'number': '4242424242424242',
|
|
... 'exp_month': 12,
|
|
... 'exp_year': today.year + 1,
|
|
... 'cvc': '123',
|
|
... },
|
|
... )
|
|
>>> Payment.write([payment.id], {
|
|
... 'stripe_token': token.id,
|
|
... 'stripe_chargeable': True,
|
|
... 'stripe_payment_intent_id': None, # Remove intent from checkout
|
|
... }, config.context)
|
|
|
|
Process the payment::
|
|
|
|
>>> process_payment = payment.click('process_wizard')
|
|
>>> payment.state
|
|
'processing'
|
|
|
|
>>> for _ in range(MAX_SLEEP):
|
|
... cron_fetch_events.click('run_once')
|
|
... payment.reload()
|
|
... if payment.state == 'succeeded':
|
|
... break
|
|
... time.sleep(FETCH_SLEEP)
|
|
>>> payment.state
|
|
'succeeded'
|
|
>>> bool(payment.stripe_captured)
|
|
True
|
|
|
|
Create failing payment::
|
|
|
|
>>> previous_idempotency_key = payment.stripe_idempotency_key
|
|
>>> payment, = payment.duplicate()
|
|
>>> payment.stripe_idempotency_key != previous_idempotency_key
|
|
True
|
|
>>> payment.click('submit')
|
|
>>> payment.state
|
|
'submitted'
|
|
>>> checkout = payment.click('stripe_checkout')
|
|
>>> bool(payment.stripe_checkout_id)
|
|
True
|
|
>>> token = stripe.Token.create(
|
|
... card={
|
|
... 'number': '4000000000000002',
|
|
... 'exp_month': 12,
|
|
... 'exp_year': today.year + 1,
|
|
... 'cvc': '123',
|
|
... },
|
|
... )
|
|
>>> Payment.write([payment.id], {
|
|
... 'stripe_token': token.id,
|
|
... 'stripe_chargeable': True,
|
|
... 'stripe_payment_intent_id': None, # Remove intent from checkout
|
|
... }, config.context)
|
|
>>> process_payment = payment.click('process_wizard')
|
|
>>> payment.state
|
|
'failed'
|
|
>>> payment.stripe_error_code
|
|
'card_declined'
|
|
|
|
Create a customer::
|
|
|
|
>>> Customer = Model.get('account.payment.stripe.customer')
|
|
>>> stripe_customer = Customer()
|
|
>>> stripe_customer.party = customer
|
|
>>> stripe_customer.stripe_account = stripe_account
|
|
|
|
Checkout the customer::
|
|
|
|
>>> checkout = stripe_customer.click('stripe_checkout')
|
|
>>> bool(stripe_customer.stripe_checkout_id)
|
|
True
|
|
|
|
>>> token = stripe.Token.create(
|
|
... card={
|
|
... 'number': '4012888888881881',
|
|
... 'exp_month': 12,
|
|
... 'exp_year': today.year + 1,
|
|
... 'cvc': '123',
|
|
... },
|
|
... )
|
|
>>> Customer.write(
|
|
... [stripe_customer.id], {'stripe_token': token.id}, config.context)
|
|
|
|
Run cron::
|
|
|
|
>>> cron_customer_create, = Cron.find([
|
|
... ('method', '=', 'account.payment.stripe.customer|stripe_create'),
|
|
... ])
|
|
>>> cron_customer_create.companies.append(get_company())
|
|
>>> cron_customer_create.click('run_once')
|
|
|
|
>>> stripe_customer.reload()
|
|
>>> bool(stripe_customer.stripe_customer_id)
|
|
True
|
|
|
|
Update customer::
|
|
|
|
>>> contact = customer.contact_mechanisms.new()
|
|
>>> contact.type = 'email'
|
|
>>> contact.value = 'customer@example.com'
|
|
>>> customer.save()
|
|
|
|
>>> cus = stripe.Customer.retrieve(stripe_customer.stripe_customer_id)
|
|
>>> cus.email
|
|
'customer@example.com'
|
|
>>> cus.preferred_locales
|
|
['en']
|
|
|
|
Make payment with customer::
|
|
|
|
>>> payment, = payment.duplicate()
|
|
>>> payment.stripe_customer = stripe_customer
|
|
>>> payment.save()
|
|
>>> _, source = Payment.get_stripe_customer_sources(payment.id, config.context)
|
|
>>> source_id, source_name = source
|
|
>>> source_name
|
|
'Visa ****1881 12/...'
|
|
>>> payment.stripe_customer_source = source_id
|
|
>>> payment.click('submit')
|
|
>>> payment.state
|
|
'submitted'
|
|
>>> process_payment = payment.click('process_wizard')
|
|
>>> payment.state
|
|
'processing'
|
|
|
|
>>> for _ in range(MAX_SLEEP):
|
|
... cron_fetch_events.click('run_once')
|
|
... payment.reload()
|
|
... if payment.state == 'succeeded':
|
|
... break
|
|
... time.sleep(FETCH_SLEEP)
|
|
>>> payment.state
|
|
'succeeded'
|
|
|
|
Detach source::
|
|
|
|
>>> detach = stripe_customer.click('detach_source')
|
|
>>> detach.form.source = source_id
|
|
>>> detach.execute('detach')
|
|
|
|
>>> cus = stripe.Customer.retrieve(
|
|
... stripe_customer.stripe_customer_id, expand=['sources'])
|
|
>>> len(cus.sources)
|
|
0
|
|
>>> len(stripe.PaymentMethod.list(customer=cus.id, type='card'))
|
|
0
|
|
|
|
Delete customer::
|
|
|
|
>>> stripe_customer.delete()
|
|
>>> bool(stripe_customer.active)
|
|
False
|
|
|
|
Run cron::
|
|
|
|
>>> cron_customer_delete, = Cron.find([
|
|
... ('method', '=', 'account.payment.stripe.customer|stripe_delete'),
|
|
... ])
|
|
>>> cron_customer_delete.companies.append(get_company())
|
|
>>> cron_customer_delete.click('run_once')
|
|
|
|
>>> stripe_customer.reload()
|
|
>>> stripe_customer.stripe_token
|
|
>>> stripe_customer.stripe_customer_id
|
|
|
|
Create capture payment::
|
|
|
|
>>> payment, = payment.duplicate()
|
|
>>> payment.stripe_capture = False
|
|
>>> payment.click('submit')
|
|
>>> payment.state
|
|
'submitted'
|
|
|
|
Checkout the capture payment::
|
|
|
|
>>> token = stripe.Token.create(
|
|
... card={
|
|
... 'number': '4242424242424242',
|
|
... 'exp_month': 12,
|
|
... 'exp_year': today.year + 1,
|
|
... 'cvc': '123',
|
|
... },
|
|
... )
|
|
>>> Payment.write([payment.id], {
|
|
... 'stripe_token': token.id,
|
|
... }, config.context)
|
|
|
|
Process the capture payment::
|
|
|
|
>>> process_payment = payment.click('process_wizard')
|
|
>>> payment.state
|
|
'processing'
|
|
>>> bool(payment.stripe_captured)
|
|
False
|
|
|
|
Capture lower amount::
|
|
|
|
>>> payment.amount = Decimal('40')
|
|
>>> payment.click('stripe_do_capture')
|
|
>>> payment.state
|
|
'processing'
|
|
|
|
>>> for _ in range(MAX_SLEEP):
|
|
... cron_fetch_events.click('run_once')
|
|
... payment.reload()
|
|
... if payment.state == 'succeeded':
|
|
... break
|
|
... time.sleep(FETCH_SLEEP)
|
|
>>> payment.state
|
|
'succeeded'
|
|
>>> bool(payment.stripe_captured)
|
|
True
|
|
|
|
Refund some amount::
|
|
|
|
>>> Refund = Model.get('account.payment.stripe.refund')
|
|
>>> refund = Refund()
|
|
>>> refund.payment = payment
|
|
>>> refund.amount = Decimal('38')
|
|
>>> refund.click('submit')
|
|
>>> refund.click('approve')
|
|
>>> cron_refund_create, = Cron.find([
|
|
... ('method', '=', 'account.payment.stripe.refund|stripe_create'),
|
|
... ])
|
|
>>> cron_refund_create.click('run_once')
|
|
|
|
>>> for _ in range(MAX_SLEEP):
|
|
... cron_fetch_events.click('run_once')
|
|
... payment.reload()
|
|
... if payment.amount == Decimal('2.00'):
|
|
... break
|
|
... time.sleep(FETCH_SLEEP)
|
|
>>> payment.amount
|
|
Decimal('2.00')
|
|
>>> payment.state
|
|
'succeeded'
|
|
>>> refund.reload()
|
|
>>> refund.state
|
|
'succeeded'
|
|
|
|
Simulate charge.refunded event with full amount::
|
|
|
|
>>> refund = Refund()
|
|
>>> refund.payment = payment
|
|
>>> refund.amount = Decimal('2')
|
|
>>> refund.click('submit')
|
|
>>> refund.click('approve')
|
|
>>> cron_refund_create.click('run_once')
|
|
|
|
>>> for _ in range(MAX_SLEEP):
|
|
... cron_fetch_events.click('run_once')
|
|
... payment.reload()
|
|
... if payment.amount == Decimal('0.00'):
|
|
... break
|
|
... time.sleep(FETCH_SLEEP)
|
|
>>> payment.amount
|
|
Decimal('0.00')
|
|
>>> payment.state
|
|
'failed'
|
|
>>> refund.reload()
|
|
>>> refund.state
|
|
'succeeded'
|
|
|
|
Try to refund more::
|
|
|
|
>>> refund = Refund()
|
|
>>> refund.payment = payment
|
|
>>> refund.amount = Decimal('10')
|
|
>>> refund.click('submit')
|
|
>>> refund.click('approve')
|
|
>>> cron_refund_create.click('run_once')
|
|
|
|
>>> for _ in range(MAX_SLEEP):
|
|
... cron_fetch_events.click('run_once')
|
|
... refund.reload()
|
|
... if refund.state == 'failed':
|
|
... break
|
|
... time.sleep(FETCH_SLEEP)
|
|
>>> refund.state
|
|
'failed'
|