source: main/waeup.ikoba/trunk/src/waeup/ikoba/customers/tests/test_contract.py

## $Id: test_contract.py 12769 2015-03-15 13:14:08Z henrik $
##
## Copyright (C) 2014 Uli Fouquet & Henrik Bettermann
## This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation; either version 2 of the License, or
## (at your option) any later version.
##
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
## GNU General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with this program; if not, write to the Free Software
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
##
"""
Tests for contracts.
"""
import decimal
from zope.interface.verify import verifyClass, verifyObject
from zope.component import createObject, getUtility, getUtilitiesFor
from zope.component.hooks import setSite
from hurry.workflow.interfaces import (
    IWorkflowInfo, IWorkflowState, InvalidTransitionError)
from waeup.ikoba.customers.interfaces import (
    IContractsContainer, IContract)
from waeup.ikoba.customers.contracts import (
    ContractsContainer, SampleContract, ContractPayer, ContractFinder,
    PayableContract,
    )
from waeup.ikoba.app import Company
from waeup.ikoba.customers.customer import Customer
from waeup.ikoba.payments.interfaces import (
    IPaymentItem, IPayer, IPayableFinder, IPayable,
    )
from waeup.ikoba.products.productoptions import ProductOption
from waeup.ikoba.testing import (FunctionalLayer, FunctionalTestCase)


class ContractsContainerTestCase(FunctionalTestCase):

    layer = FunctionalLayer

    def test_interfaces(self):
        # Make sure the correct interfaces are implemented.
        self.assertTrue(
            verifyClass(
                IContractsContainer, ContractsContainer)
            )
        self.assertTrue(
            verifyObject(
                IContractsContainer, ContractsContainer())
            )
        self.assertTrue(
            verifyClass(
                IContract, SampleContract)
            )
        self.assertTrue(
            verifyObject(
                IContract, SampleContract())
            )
        return

    def test_addContract(self):
        container = ContractsContainer()
        contract = createObject(u'waeup.SampleContract')
        id = contract.contract_id
        container.addContract(contract)
        self.assertEqual(container[id], contract)
        self.assertRaises(TypeError, container.addContract, object())
        self.assertEqual(contract.class_name, 'SampleContract')
        return

    def test_contract_workflow(self):
        contract = createObject(u'waeup.SampleContract')
        IWorkflowInfo(contract).fireTransition('create')
        self.assertEqual(IWorkflowState(contract).getState(), 'created')
        IWorkflowInfo(contract).fireTransition('submit')
        self.assertEqual(IWorkflowState(contract).getState(), 'submitted')
        IWorkflowInfo(contract).fireTransition('approve')
        self.assertEqual(IWorkflowState(contract).getState(), 'approved')
        IWorkflowInfo(contract).fireTransition('expire')
        self.assertEqual(IWorkflowState(contract).getState(), 'expired')
        IWorkflowInfo(contract).fireTransition('reset4')
        self.assertEqual(IWorkflowState(contract).getState(), 'created')
        self.assertRaises(InvalidTransitionError,
            IWorkflowInfo(contract).fireTransition, 'approve')
        IWorkflowInfo(contract).fireTransition('submit')
        IWorkflowInfo(contract).fireTransition('reset3')
        self.assertEqual(IWorkflowState(contract).getState(), 'created')
        return

    def test_contract_history(self):
        doc = createObject(u'waeup.SampleContract')
        IWorkflowInfo(doc).fireTransition('create')
        messages = ' '.join(doc.history.messages)
        self.assertTrue('Contract created by system' in messages)
        return


class TestContractHelpers(FunctionalTestCase):

    layer = FunctionalLayer

    def test_payer_adapter(self):
        # we can adapt IContract to IPayer (i.e. create a payer)
        customer = Customer()
        customer.firstname, customer.lastname = u'Anna', u'Tester'
        contract = createObject(u'waeup.SampleContract')
        customer['contracts'] = ContractsContainer()
        customer['contracts'].addContract(contract)
        result = IPayer(contract)
        self.assertTrue(isinstance(result, ContractPayer))
        verifyObject(IPayer, result)
        self.assertEqual(result.first_name, u'Anna')
        self.assertEqual(result.last_name, u'Tester')
        self.assertEqual(result.payer_id, customer.customer_id)

    def test_contract_finder_iface(self):
        # we have a contract finder that returns IPayableFinder data.
        verifyClass(IPayableFinder, ContractFinder)

    def test_contract_finder_registered(self):
        # the contract finder is a utility registered on startup
        util = getUtility(IPayableFinder, name='contracts_finder')
        self.assertTrue(isinstance(util, ContractFinder))
        utils = [util for name, util in getUtilitiesFor(IPayableFinder)
                 if isinstance(util, ContractFinder)]
        self.assertEqual(len(utils), 1)

    def create_contract_and_site(self):
        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]
        contract.contract_id = u'CON1234'
        self.getRootFolder()['app'] = Company()
        app = self.getRootFolder()['app']
        setSite(app)
        return contract, app

    def test_contract_finder(self):
        # the contract finder can really find contracts
        contract, app = self.create_contract_and_site()
        app['mycontract'] = contract  # trigger cataloging
        finder = ContractFinder()
        result = finder.get_payable_by_id('CON1234')
        self.assertTrue(result is contract)

    def test_contract_finder_not_stored(self):
        # we get none if an id is not stored
        contract, app = self.create_contract_and_site()
        app['mycontract'] = contract  # trigger cataloging
        finder = ContractFinder()
        result = finder.get_payable_by_id('Not-a-valid-id')
        self.assertTrue(result is None)

    def test_contract_finder_no_catalog(self):
        # contract finder does not complain about missing catalog
        finder = ContractFinder()
        result = finder.get_payable_by_id('CON1234')
        self.assertTrue(result is None)


class TestContractAsPayable(FunctionalTestCase):

    layer = FunctionalLayer

    def test_adaptable(self):
        # we can turn contracts into payables.
        contract = SampleContract()
        payable = IPayable(contract)
        self.assertTrue(payable is not None)
        self.assertTrue(isinstance(payable, PayableContract))

    def test_payable_iface(self):
        # PayableContracts really provide IPayable
        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]
        payable = PayableContract(contract)
        verifyObject(IPayable, payable)
        verifyClass(IPayable, PayableContract)

    def test_payable_simple_attributes(self):
        # the simple attribs are set correctly, according to context contract
        contract = SampleContract()
        contract.title = u'the title'
        option1 = ProductOption(u"Fee 1", decimal.Decimal("31.10"), "EUR")
        option2 = ProductOption(u"Fee 2", decimal.Decimal("12.12"), "EUR")
        contract.product_options = [option1, option2]
        payable = PayableContract(contract)
        self.assertTrue(contract.contract_id, payable.payable_id)
        self.assertEqual(payable.title, contract.title)
        self.assertEqual(payable.currency, 'EUR')

    def test_payable_items(self):
        # we can get payment items from payable
        contract = SampleContract()
        option1 = ProductOption(u"Fee 1", decimal.Decimal("31.10"), "EUR")
        option2 = ProductOption(u"Fee 2", decimal.Decimal("12.12"), "EUR")
        contract.product_options = [option1, option2]
        payable = PayableContract(contract)
        items = payable.payment_items
        self.assertTrue(isinstance(items, tuple))
        self.assertEqual(len(items), 2)
        verifyObject(IPaymentItem, items[0])
        verifyObject(IPaymentItem, items[1])
        self.assertEqual(items[0].item_id, '0')
        self.assertEqual(items[0].title, u'Fee 1')
        self.assertEqual(items[0].amount, decimal.Decimal("31.10"))

    def test_payable_no_items(self):
        # payables work also with no options set on contract
        contract = SampleContract()
        payable = PayableContract(contract)
        items = payable.payment_items
        self.assertTrue(isinstance(items, tuple))
        self.assertEqual(len(items), 0)
        self.assertEqual(payable.currency, None)