first commit

modules/account_payment_braintree/tests/__init__.py

@@ -0,0 +1,2 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.

modules/account_payment_braintree/tests/scenario_account_payment_braintree.rst

@@ -0,0 +1,287 @@
+==================================
+Account Payment Braintree Scenario
+==================================
+
+Imports::
+
+    >>> import os
+    >>> import random
+    >>> from decimal import Decimal
+
+    >>> import braintree
+    >>> from braintree.test.nonces import Nonces
+
+    >>> 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, assertEqual, assertTrue
+
+Activate modules::
+
+    >>> config = activate_modules(
+    ...     'account_payment_braintree', create_company, create_chart)
+
+Get cron::
+
+    >>> Cron = Model.get('ir.cron')
+
+Get company::
+
+    >>> Company = Model.get('company.company')
+    >>> company = get_company()
+
+Create fiscal year::
+
+    >>> fiscalyear = create_fiscalyear()
+    >>> fiscalyear.click('create_period')
+
+Create Braintree account::
+
+    >>> BraintreeAccount = Model.get('account.payment.braintree.account')
+    >>> braintree_account = BraintreeAccount(name="Braintree")
+    >>> braintree_account.currency = company.currency
+    >>> braintree_account.environment = 'sandbox'
+    >>> braintree_account.merchant_id = os.getenv('BRAINTREE_MERCHANT_ID')
+    >>> braintree_account.public_key = os.getenv('BRAINTREE_PUBLIC_KEY')
+    >>> braintree_account.private_key = os.getenv('BRAINTREE_PRIVATE_KEY')
+    >>> braintree_account.save()
+
+    >>> gateway = braintree.BraintreeGateway(braintree.Configuration(
+    ...         environment=braintree.Environment.Sandbox,
+    ...         merchant_id=braintree_account.merchant_id,
+    ...         public_key=braintree_account.public_key,
+    ...         private_key=braintree_account.private_key,
+    ...         ))
+
+Create webhook identifier::
+
+    >>> braintree_account.click('new_identifier')
+    >>> len(braintree_account.webhook_identifier)
+    32
+
+Remove webhook::
+
+    >>> braintree_account.click('new_identifier')
+    >>> braintree_account.webhook_identifier
+
+Create payment journal::
+
+    >>> PaymentJournal = Model.get('account.payment.journal')
+    >>> payment_journal = PaymentJournal(name="Braintree",
+    ...     process_method='braintree', braintree_account=braintree_account)
+    >>> payment_journal.save()
+
+Create party::
+
+    >>> Party = Model.get('party.party')
+    >>> customer = Party(name="Customer")
+    >>> customer.save()
+
+Create submitted payment::
+
+    >>> Payment = Model.get('account.payment')
+    >>> payment = Payment()
+    >>> payment.journal = payment_journal
+    >>> payment.kind = 'receivable'
+    >>> payment.party = customer
+    >>> # Use random amount to prevent gateway rejection for duplicate
+    >>> payment.amount = amount = Decimal(random.randint(0, 1999))
+    >>> payment.click('submit')
+    >>> payment.state
+    'submitted'
+
+Checkout the payment::
+
+    >>> checkout = payment.click('braintree_checkout')
+    >>> bool(payment.braintree_checkout_id)
+    True
+
+    >>> Payment.write([payment.id], {
+    ...     'braintree_nonce': Nonces.Transactable,
+    ...     }, config.context)
+
+Process the payment::
+
+    >>> process_payment = payment.click('process_wizard')
+    >>> payment.state
+    'processing'
+    >>> assertTrue(payment.braintree_payment_settled)
+    >>> assertEqual(payment.amount, amount)
+
+    >>> _ = gateway.testing.settle_transaction(payment.braintree_transaction_id)
+
+Pull update::
+
+    >>> cron_payment_pull, = Cron.find([
+    ...     ('method', '=', 'account.payment|braintree_pull'),
+    ...     ])
+    >>> cron_payment_pull.companies.append(Company(company.id))
+    >>> cron_payment_pull.click('run_once')
+
+    >>> payment.reload()
+    >>> payment.state
+    'succeeded'
+    >>> assertEqual(payment.amount, amount)
+
+Create a customer::
+
+    >>> Customer = Model.get('account.payment.braintree.customer')
+    >>> braintree_customer = Customer()
+    >>> braintree_customer.party = customer
+    >>> braintree_customer.braintree_account = braintree_account
+    >>> braintree_customer.save()
+    >>> Customer.write([braintree_customer.id], {
+    ...     'braintree_nonce': Nonces.Transactable,
+    ...     }, config.context)
+
+Run cron::
+
+    >>> cron_customer_create, = Cron.find([
+    ...     ('method', '=', 'account.payment.braintree.customer|braintree_create'),
+    ...     ])
+    >>> cron_customer_create.companies.append(Company(company.id))
+    >>> cron_customer_create.click('run_once')
+
+    >>> braintree_customer.reload()
+    >>> bool(braintree_customer.braintree_customer_id)
+    True
+
+Update customer::
+
+   >>> contact = customer.contact_mechanisms.new()
+   >>> contact.type = 'email'
+   >>> contact.value = 'customer@example.com'
+   >>> customer.save()
+
+   >>> cus = gateway.customer.find(braintree_customer.braintree_customer_id)
+   >>> cus.email
+   'customer@example.com'
+
+Make payment with customer::
+
+    >>> payment, = payment.duplicate()
+    >>> payment.braintree_customer = braintree_customer
+    >>> payment.amount = amount = Decimal(random.randint(0, 1999))
+    >>> payment.save()
+    >>> _, method = Payment.get_braintree_customer_methods(payment.id, config.context)
+    >>> method_token, _ = method
+    >>> payment.braintree_customer_method = method_token
+    >>> payment.click('submit')
+    >>> payment.state
+    'submitted'
+    >>> process_payment = payment.click('process_wizard')
+    >>> payment.state
+    'processing'
+
+    >>> _ = gateway.testing.settle_transaction(payment.braintree_transaction_id)
+    >>> cron_payment_pull.click('run_once')
+    >>> payment.reload()
+    >>> payment.state
+    'succeeded'
+
+Delete customer::
+
+    >>> braintree_customer.delete()
+    >>> bool(braintree_customer.active)
+    False
+
+Run cron::
+
+    >>> cron_customer_delete, = Cron.find([
+    ...     ('method', '=', 'account.payment.braintree.customer|braintree_delete'),
+    ...     ])
+    >>> cron_customer_delete.companies.append(Company(company.id))
+    >>> cron_customer_delete.click('run_once')
+
+    >>> braintree_customer.reload()
+    >>> braintree_customer.braintree_customer_id
+
+Create payment to settle::
+
+    >>> payment, = payment.duplicate()
+    >>> payment.braintree_customer = None
+    >>> payment.braintree_settle_payment = False
+    >>> payment.amount = amount = Decimal(random.randint(0, 1999))
+    >>> payment.click('submit')
+    >>> payment.state
+    'submitted'
+
+    >>> Payment.write([payment.id], {
+    ...     'braintree_nonce': Nonces.Transactable,
+    ...     }, config.context)
+
+    >>> process_payment = payment.click('process_wizard')
+    >>> payment.state
+    'processing'
+
+    >>> _ = gateway.testing.settle_transaction(payment.braintree_transaction_id)
+    >>> cron_payment_pull.click('run_once')
+    >>> payment.reload()
+    >>> payment.state
+    'processing'
+
+Settle lower amount::
+
+    >>> payment.amount = amount = Decimal(random.randint(2, int(payment.amount)))
+    >>> payment.click('braintree_do_settle_payment')
+    >>> payment.state
+    'processing'
+
+    >>> _ = gateway.testing.settle_transaction(payment.braintree_transaction_id)
+    >>> cron_payment_pull.click('run_once')
+    >>> payment.reload()
+    >>> payment.state
+    'succeeded'
+    >>> bool(payment.braintree_payment_settled)
+    True
+
+Refund some amount::
+
+    >>> Refund = Model.get('account.payment.braintree.refund')
+    >>> refund = Refund()
+    >>> refund.payment = payment
+    >>> refund.amount = amount - 1
+    >>> refund.click('submit')
+    >>> refund.click('approve')
+    >>> cron_refund, = Cron.find([
+    ...     ('method', '=', 'account.payment.braintree.refund|braintree_refund'),
+    ...     ])
+    >>> cron_refund.companies.append(Company(company.id))
+    >>> cron_refund.click('run_once')
+
+    >>> refund.reload()
+    >>> refund.state
+    'processing'
+    >>> payment.reload()
+    >>> assertEqual(payment.amount, amount)
+
+    >>> _ = gateway.testing.settle_transaction(refund.braintree_transaction_id)
+    >>> cron_refund_pull, = Cron.find([
+    ...     ('method', '=', 'account.payment.braintree.refund|braintree_pull'),
+    ...     ])
+    >>> cron_refund_pull.companies.append(Company(company.id))
+    >>> cron_refund_pull.click('run_once')
+    >>> refund.reload()
+    >>> refund.state
+    'succeeded'
+
+    >>> payment.reload()
+    >>> payment.amount
+    Decimal('1.00')
+    >>> payment.state
+    'succeeded'
+
+Try to refund more::
+
+    >>> refund = Refund()
+    >>> refund.payment = payment
+    >>> refund.amount = Decimal('10')
+    >>> refund.click('submit')
+    >>> refund.click('approve')
+    >>> cron_refund.click('run_once')
+    >>> refund.reload()
+    >>> refund.state
+    'failed'
+    >>> payment.reload()
+    >>> payment.amount
+    Decimal('1.00')

modules/account_payment_braintree/tests/test_module.py

@@ -0,0 +1,13 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+
+from trytond.modules.party.tests import PartyCheckReplaceMixin
+from trytond.tests.test_tryton import ModuleTestCase
+
+
+class AccountPaymentBraintreeTestCase(PartyCheckReplaceMixin, ModuleTestCase):
+    'Test Account Payment Braintree module'
+    module = 'account_payment_braintree'
+
+
+del ModuleTestCase

modules/account_payment_braintree/tests/test_scenario.py

@@ -0,0 +1,16 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+
+import os
+
+from trytond.tests.test_tryton import TEST_NETWORK, load_doc_tests
+
+
+def load_tests(*args, **kwargs):
+    if (not TEST_NETWORK
+            or not (os.getenv('BRAINTREE_MERCHANT_ID')
+                and os.getenv('BRAINTREE_PUBLIC_KEY')
+                and os.getenv('BRAINTREE_PRIVATE_KEY'))):
+        kwargs.setdefault('skips', set()).add(
+            'scenario_account_payment_braintree.rst')
+    return load_doc_tests(__name__, __file__, *args, **kwargs)