#
#    Copyright (C) 2008-2012  Corporation of Balclutha. All rights Reserved.
#
#    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.
#
#    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
#
#    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
#    AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
#    IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
#    ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
#    LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
#    CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
#    GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
#    HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
#    LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
#    OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#
 
from unittest import TestSuite, makeSuite
from Products.BastionLedger.tests.LedgerTestCase import LedgerTestCase
from Acquisition import aq_base
from DateTime import DateTime
from Products.BastionBanking.ZCurrency import ZCurrency
from Products.BastionLedger.utils import floor_date
 
 
class TestTaxTables(LedgerTestCase):
    """
    lightweight Zope-only tax table tests
    """
    def afterSetUp(self):
        LedgerTestCase.afterSetUp(self)
	self.controller_tool.manage_addProduct['BastionLedger'].manage_addBLTaxTable('company_tax')
        self.table = self.controller_tool.company_tax
        self.addRate = self.table.manage_addProduct['BastionLedger'].manage_addBLTaxTableRecord
 
    def testConstructed(self):
        self.assertEqual(self.table.indexes(), ['effective_date'])
 
    def testCanAddRate(self):
        self.addRate(0.35,DateTime('1980/01/01'))
 
        self.assertEqual(len(self.table.objectIds()),1)
        self.assertEqual(len(self.table._catalog), 1)
 
        table_rec = self.table.objectValues()[0]
 
        fields = table_rec.fields()
        fields.sort()
        self.assertEqual(fields, ['code', 'effective_date','rate',])
 
    def testRateOrdering(self):
        self.addRate(0.35,DateTime('1980/01/01'))
        self.addRate(0.34,DateTime('1981/01/01'))
        self.addRate(0.33,DateTime('1982/01/01'))
 
        recs = self.table.objectValues()
 
        self.assertEqual(min(recs).effective(), DateTime('1980/01/01'))
        self.assertEqual(max(recs).effective(), DateTime('1982/01/01'))
 
    def testCodes(self):
        self.addRate(0.35,DateTime('1980/01/01'), 'A')
        self.addRate(0.34,DateTime('1980/01/01'), 'B')
        self.addRate(0.35,DateTime('1981/01/01'), 'A')
        self.addRate(0.34,DateTime('1981/01/01'), 'C')
 
        self.assertEqual(self.table.taxCodes(), ['A', 'B', 'C'])
        self.assertEqual(self.table.taxCodes(DateTime('1980/01/01')), ['A', 'B'])
        self.assertEqual(len(self.table.getTableRecords(DateTime('1981/01/01'))), 2)
        self.assertEqual(len(self.table.getTableRecords(DateTime('1981/01/01'), 'A')), 1)
        self.assertEqual(len(self.table.getTableRecords(DateTime('1981/01/01'), 'D')), 0)
 
    def testMultiDimensional(self):
	self.controller_tool.manage_addProduct['BastionLedger'].manage_addBLTaxTable('personal_tax',
                                                                                     'Personal Tax Rates',
                                                                                     (('locale', 'FieldIndex'),))
        table = self.controller_tool.personal_tax
        #self.assertEqual(table.title, 'Personal Tax Rates')
        self.assertEqual(table.dimensions(), ['locale'])
 
        addRate = table.manage_addProduct['BastionLedger'].manage_addBLTaxTableRecord
        addRate(0.10,DateTime('1980/01/01'), locale='gb')
        addRate(0.20,DateTime('1980/01/01'), locale='nz')
        addRate(0.10,DateTime('1981/01/01'), locale='gb')
        addRate(0.21,DateTime('1981/01/01'), locale='nz')
 
        table_rec = table.objectValues()[0]
 
        fields = table_rec.fields()
        fields.sort()
        self.assertEqual(fields, ['code', 'effective_date', 'locale', 'rate'])
 
        self.assertEqual(len(table.getTableRecords(DateTime('1980/06/01'),
                                                   locale='nz')),
                         1)
        self.assertEqual(len(table.getTableRecords(DateTime('1981/06/01'),
                                                   locale='nz')),
                         1)
 
    def testTieredAmounts(self):
	self.controller_tool.manage_addProduct['BastionLedger'].manage_addBLTaxTable('personal_tax',
                                                                            'Personal Tax Rates',
                                                                            (('amount', 'CurrencyIndex'),))
        table = self.controller_tool.personal_tax
        # we don't want to have to set up currencies ...
        table._catalog.indexes['amount']._updateProperty('base_currency', 'GBP')
 
        self.assertEqual(table.dimensions(), ['amount'])
 
        addRate = table.manage_addProduct['BastionLedger'].manage_addBLTaxTableRecord
        addRate(0.10,DateTime('1980/01/01'), amount=ZCurrency('GBP 10000'))
        addRate(0.20,DateTime('1980/01/01'), amount=ZCurrency('GBP 20000'))
        addRate(0.30,DateTime('1981/01/01'), amount=ZCurrency('GBP 30000'))
        addRate(0.40,DateTime('1981/01/01'), amount=ZCurrency('GBP 40000'))
 
 
        self.assertEqual(len(table.getTableRecords(DateTime('1980/06/01'),
                                                   amount={'query': ZCurrency('GBP 20000'),
                                                           'range':'max'})),
                         2)
        self.assertEqual(len(table.getTableRecords(DateTime('1981/06/01'),
                                                   amount={'query': ZCurrency('GBP 20000'),
                                                           'range':'min'})),
                         2)
 
        self.assertEqual(len(table.getTableRecords(DateTime('1981/06/01'),
                                                   amount={'query': (ZCurrency('GBP 18000'),
                                                                     ZCurrency('GBP 21000')),
                                                           'range':'minmax'})),
                         1)
 
        # this is a typical tiered tax table return entry ...
        self.assertEqual(map(lambda x: x.amount,
                             table.getTableRecords(DateTime('1980/06/01'),
                                                   sort_on='amount',
                                                   sort_order='asc')),
                         [ZCurrency('GBP 10000'),ZCurrency('GBP 20000')])
 
        self.assertEqual(map(lambda x: x.amount,
                             table.getTableRecords(DateTime('1981/06/01'),
                                                   sort_on='amount',
                                                   sort_order='asc')),
                         [ZCurrency('GBP 30000'),ZCurrency('GBP 40000')])
 
        self.assertEqual(table.calculateTax(DateTime('1981/06/01'), ZCurrency('GBP 30000')),
                         ZCurrency('GBP 0.00'))
 
        self.assertEqual(table.calculateTax(DateTime('1981/06/01'), ZCurrency('GBP 30001')),
                         ZCurrency('GBP 0.30'))
 
        self.assertEqual(table.calculateTax(DateTime('1981/06/01'), ZCurrency('GBP 40000')),
                         ZCurrency('GBP 3000.00'))
 
        self.assertEqual(table.calculateTax(DateTime('1981/06/01'), ZCurrency('GBP 50000')),
                         ZCurrency('GBP 7000.00'))
 
 
    def testCodes(self):
        self.assertEqual(self.table.taxCodes(), [])
        self.addRate(0.35, DateTime('2000/01/02'), 'C')
        self.assertEqual(self.table.taxCodes(), ['C'])
        self.assertEqual(self.table.taxCodes('2000/01/01'), [])
 
 
def test_suite():
    suite = TestSuite()
    suite.addTest(makeSuite(TestTaxTables))
    return suite