Changeset 12683 for main/waeup.ikoba/branches

Timestamp:

7 Mar 2015, 04:18:42 (10 years ago)

Author:

uli

Message:

Add adapter to turn lists of product opitons into payment items.

Location:

main/waeup.ikoba/branches/uli-payments/src/waeup/ikoba/customers

Files:

• contracts.py (modified) (2 diffs)
• tests/test_contract.py (modified) (3 diffs)

main/waeup.ikoba/branches/uli-payments/src/waeup/ikoba/customers/contracts.py

@@ -32 +32 @@
     IContractSelectProduct, ICustomersUtils, ISampleContract,
     ISampleContractProcess, ISampleContractEdit, ISampleContractOfficialUse)
+from waeup.ikoba.payments.interfaces import IPaymentItem
+from waeup.ikoba.payments.payment import PaymentItem
 from waeup.ikoba.utils.helpers import attrs_to_fields
 
@@ -56 +58 @@
 
 ContractsContainer = attrs_to_fields(ContractsContainer)
+
+
+def payment_items_from_contract(contract):
+    """Turn contract product options into payment items.
+    """
+    result = []
+    for num, option in enumerate(contract.product_options):
+        item = PaymentItem()
+        item.item_id = u'%s' % num
+        item.title = option.title
+        item.amount = option.fee
+        result.append(item)
+    return result
 
 

main/waeup.ikoba/branches/uli-payments/src/waeup/ikoba/customers/tests/test_contract.py

@@ -19 +19 @@
 Tests for contracts.
 """
+import decimal
 from zope.interface.verify import verifyClass, verifyObject
 from zope.component import createObject
@@ -26 +27 @@
     IContractsContainer, IContract)
 from waeup.ikoba.customers.contracts import (
-    ContractsContainer, SampleContract)
+    ContractsContainer, SampleContract, payment_items_from_contract,
+    )
+from waeup.ikoba.payments.interfaces import IPaymentItem
+from waeup.ikoba.products.productoptions import ProductOption
 from waeup.ikoba.testing import (FunctionalLayer, FunctionalTestCase)
 
@@ -89 +93 @@
         self.assertTrue('Contract created by system' in messages)
         return
+
+
+class TestContractHelpers(FunctionalTestCase):
+
+    layer = FunctionalLayer
+
+    def test_payment_items_from_contract(self):
+        # we can turn contracts into lists of payment items
+        contract = SampleContract()
+        option1 = ProductOption(u"Fee 1", decimal.Decimal("31.10"), "USD")
+        option2 = ProductOption(u"Fee 2", decimal.Decimal("12.12"), "USD")
+        contract.product_options = [option1, option2]
+        payment_items = payment_items_from_contract(contract)
+        assert len(payment_items) == 2
+        item1, item2 = payment_items
+        assert IPaymentItem.providedBy(item1)
+        assert IPaymentItem.providedBy(item2)
+        assert item1.item_id == "0"
+        assert item2.item_id == "1"
+        assert item1.title == u'Fee 1'
+        assert item2.title == u'Fee 2'
+        assert item1.amount == decimal.Decimal("31.10")
+        assert item2.amount == decimal.Decimal("12.12")
+
+    def test_payment_item_from_contract_no_options(self):
+        # we cope with empty contracts
+        contract = SampleContract()
+        payment_items = payment_items_from_contract(contract)
+        assert payment_items == []