# -*- coding: utf-8 -*-
# This file is part of the cashbook-module from m-ds for Tryton.
# The COPYRIGHT file at the top level of this repository contains the
# full copyright notices and license terms.

from trytond.tests.test_tryton import ModuleTestCase, with_transaction
from trytond.pool import Pool
from trytond.transaction import Transaction
from trytond.modules.cashbook.tests import CashbookTestCase
from .qifdata import qif_category, qif_types


class CategoryTestCase(CashbookTestCase):
    'Test cashbook categoy module'
    module = 'CashbookExchangeTestCase'

    @with_transaction()
    def test_category_create_by_qif_emptydb(self):
        """ create categories by import a qif-file
        """
        pool = Pool()
        Category = pool.get('cashbook.category')

        company = self.prep_company()
        with Transaction().set_context({
            'company': company.id,
            }):
            records = Category.create_from_qif(qif_types)

            records = Category.search([], order=[('name', 'ASC')])
            self.assertEqual(len(records), 15)
            self.assertEqual(records[0].rec_name, 'Fernsehen')
            self.assertEqual(records[1].rec_name, 'Telekommunikation/Fernsehen')
            self.assertEqual(records[2].rec_name,  'Gehalt')
            self.assertEqual(records[3].rec_name, 'Fernsehen/GEZ')
            self.assertEqual(records[4].rec_name, 'Lebensmittel')
            self.assertEqual(records[5].rec_name, 'Telekommunikation/Online-Dienste')
            self.assertEqual(records[6].rec_name, 'Telefon/Telco1-Tablett')
            self.assertEqual(records[7].rec_name, 'Telefon/Telco2-Handy')
            self.assertEqual(records[8].rec_name, 'Telefon/Telco3')
            self.assertEqual(records[9].rec_name, 'Telekommunikation/Telefon')
            self.assertEqual(records[10].rec_name, 'Telefon')
            self.assertEqual(records[11].rec_name, 'Telekommunikation')
            self.assertEqual(records[12].rec_name, 'Telekommunikation/Telefon/Test1')
            self.assertEqual(records[13].rec_name, 'Fernsehen/TV-Company')
            self.assertEqual(records[14].rec_name, 'Gehalt/Zulagen')

    @with_transaction()
    def test_category_create_by_qif_existing_categories(self):
        """ create categories by import a qif-file,
            some categories exists already
        """
        pool = Pool()
        Category = pool.get('cashbook.category')

        company = self.prep_company()
        with Transaction().set_context({
            'company': company.id,
            }):
            cat1, = Category.create([{
                'name': 'Telekommunikation',
                'cattype': 'out',
                'childs': [('create', [{
                    'cattype': 'out',
                    'name': 'Telefon',
                    }])],
                }])

            records = Category.search([])
            self.assertEqual(len(records), 2)
            self.assertEqual(records[0].rec_name, 'Telekommunikation/Telefon')
            self.assertEqual(records[1].rec_name, 'Telekommunikation')

            records1 = Category.create_from_qif(qif_types)

            records = Category.search([], order=[('name', 'ASC')])
            self.assertEqual(len(records), 15)

            for rec in records:
                print('-rec:', rec.rec_name)
            self.assertEqual(records[0].rec_name, 'Telekommunikation/Fernsehen')
            self.assertEqual(records[1].rec_name, 'Fernsehen')
            self.assertEqual(records[2].rec_name,  'Gehalt')
            self.assertEqual(records[3].rec_name, 'Fernsehen/GEZ')
            self.assertEqual(records[4].rec_name, 'Lebensmittel')
            self.assertEqual(records[5].rec_name, 'Telekommunikation/Online-Dienste')
            self.assertEqual(records[6].rec_name, 'Telefon/Telco1-Tablett')
            self.assertEqual(records[7].rec_name, 'Telefon/Telco2-Handy')
            self.assertEqual(records[8].rec_name, 'Telefon/Telco3')
            self.assertEqual(records[9].rec_name, 'Telefon')
            self.assertEqual(records[10].rec_name, 'Telekommunikation/Telefon')
            self.assertEqual(records[11].rec_name, 'Telekommunikation')
            self.assertEqual(records[12].rec_name, 'Telekommunikation/Telefon/Test1')
            self.assertEqual(records[13].rec_name, 'Fernsehen/TV-Company')
            self.assertEqual(records[14].rec_name, 'Gehalt/Zulagen')

    @with_transaction()
    def test_qiftool_split_types(self):
        """ split file-content by types
        """
        QifTool = Pool().get('cashbook_dataexchange.qiftool')

        result = QifTool.split_by_type(qif_types)
        self.assertEqual(len(result.keys()), 2)
        self.assertEqual(result['Cat'], 'NGehalt\nI\n^\nNGehalt:Zulagen\n'+
            'I\n^\nNTelekommunikation\nE\n^\nNTelekommunikation:Online-Dienste\n'+
            'E\n^\nNTelekommunikation:Telefon\nE\n^\nNTelekommunikation:Telefon:Test1\n'+
            'E\n^\nNTelefon:Telco1-Tablett\n'+
            'E\n^\nNTelefon:Telco2-Handy\nE\n^\nNTelefon:Telco3\nE\n^\n'+
            'NTelekommunikation:Fernsehen\nE\n^\nNFernsehen:TV-Company\nE\n'+
            '^\nNFernsehen:GEZ\nE\n^\nNLebensmittel\nE\n^')
        self.assertEqual(result['Bank'], 'D04.12.2013\nT7,12\nCX\nPOpening Balance\n'+
            'L[Bargeld]\n^\nD05.12.2013\nCX\nM05.12/06.42UHR TT TELTOW\nT290,00\n'+
            'PGA NR00002168 BLZ10000000 0\nL[S-Giro]\n^\nD05.12.2013\nCX\nMsome food\n'+
            'T-56,37\nPFoodshop Zehlendorf\nLLebensmittel\n^\n')

    @with_transaction()
    def test_qiftool_read_categories(self):
        """ read category-data from text
        """
        QifTool = Pool().get('cashbook_dataexchange.qiftool')

        result = QifTool.qif_read_categories('NGehalt\nI\n^\nNGehalt:Zulagen\nI\n^'+
            'NTelekommunikation\nE\n^\nNTelekommunikation:Online-Dienste\nE\n^')
        self.assertEqual(result, {
            'in': {
                'Gehalt': {
                    'type': 'in',
                    'childs': {
                        'Zulagen': {
                            'type': 'in',
                            'childs': {},
                            },
                        },
                    },
                },
            'out': {
                'Telekommunikation': {
                    'type': 'out',
                    'childs': {
                        'Online-Dienste': {
                            'type': 'out',
                            'childs': {},
                            },
                        },
                    },
                },
            })

# end CategoryTestCase