bank-statement-import/bank_statement_parse_nl_ing_mt940/tests/test_import_bank_statement.py

# -*- coding: utf-8 -*-
"""Run test to import MT940 IBAN ING import."""
##############################################################################
#
#    Copyright (C) 2015 Therp BV <http://therp.nl>.
#
#    All other contributions are (C) by their respective contributors
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as
#    published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################
from openerp.tests.common import TransactionCase
from openerp.modules.module import get_module_resource


class TestStatementFile(TransactionCase):
    """Run test to import MT940 ING import."""

    def _test_statement_import(
            self, file_name, statement_name, start_balance, end_balance):
        """Test correct creation of single statement."""
        import_model = self.registry('account.bank.statement.import')
        statement_model = self.registry('account.bank.statement')
        cr, uid = self.cr, self.uid
        statement_path = get_module_resource(
            'bank_statement_parse_nl_ing_mt940',
            'test_files',
            file_name
        )
        statement_file = open(
            statement_path, 'rb').read().encode('base64')
        bank_statement_id = import_model.create(
            cr, uid,
            dict(
                data_file=statement_file,
            )
        )
        import_model.import_file(cr, uid, [bank_statement_id])
        # statement name is account number + '-' + date of last 62F line:
        ids = statement_model.search(
            cr, uid, [('name', '=', statement_name)])
        self.assertTrue(
            ids,
            'Statement %s not found after parse.' % statement_name
        )
        statement_id = ids[0]
        statement_obj = statement_model.browse(
            cr, uid, statement_id)
        self.assertTrue(
            abs(statement_obj.balance_start - start_balance) < 0.00001,
            'Start balance %f not equal to expected %f' %
            (statement_obj.balance_start, start_balance)
        )
        self.assertTrue(
            abs(statement_obj.balance_end - end_balance) < 0.00001,
            'Start balance %f not equal to expected %f' %
            (statement_obj.balance_end_real, end_balance)
        )

    def test_old_statement_import(self):
        """Test correct creation of single statement from old format."""
        self._test_statement_import(
            'test-ing-old.940', 'NL77INGB0574908765-2014-01-20',
            662.23, 564.35
        )

    def test_statement_import(self):
        """Test correct creation of single statement."""
        self._test_statement_import(
            'test-ing.940', 'NL77INGB0574908765-2014-02-20',
            662.23, 564.35
        )