"""test_reports_related_postings - Unit tests for RelatedPostings"""

# Copyright © 2020  Brett Smith

#

# 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 <https://www.gnu.org/licenses/>.

import collections

import datetime

import itertools

from decimal import Decimal

import pytest

from . import testutil

from conservancy_beancount import data

from conservancy_beancount.reports import core

def accruals_and_payments(acct, src_acct, dst_acct, start_date, *amounts):

    for amt, currency in amounts:

        yield testutil.Transaction(date=next(dates), postings=[

            (acct, amt, currency),

            (dst_acct if amt < 0 else src_acct, -amt, currency),

        ])

@pytest.fixture

def credit_card_cycle():

    return list(accruals_and_payments(

        'Liabilities:CreditCard',

        'Assets:Checking',

        'Expenses:Other',

        datetime.date(2020, 4, 1),

        (-110, 'USD'),

        (110, 'USD'),

        (-120, 'USD'),

        (120, 'USD'),

    ))

@pytest.fixture

def two_accruals_three_payments():

    return list(accruals_and_payments(

        'Assets:Receivable:Accounts',

        'Income:Donations',

        'Assets:Checking',

        datetime.date(2020, 4, 10),

        (440, 'USD'),

        (-230, 'USD'),

        (550, 'EUR'),

        (-210, 'USD'),

        (-550, 'EUR'),

    ))

def test_balance_empty():

    balance = core.RelatedPostings().balance()

    assert not balance

    assert balance.is_zero()

def test_balance_credit_card(credit_card_cycle):

    related = core.RelatedPostings()

    assert related.balance() == testutil.balance_map()

    expected = Decimal()

    for txn in credit_card_cycle:

        post = txn.postings[0]

        expected += post.units.number

        related.add(post)

        assert related.balance() == testutil.balance_map(USD=expected)

    assert expected == 0

# 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 <https://www.gnu.org/licenses/>.

import datetime

import itertools

import beancount.core.amount as bc_amount

import beancount.core.data as bc_data

from decimal import Decimal

from pathlib import Path

from conservancy_beancount import rtutil

EXTREME_FUTURE_DATE = datetime.date(datetime.MAXYEAR, 12, 30)

FUTURE_DATE = datetime.date.today() + datetime.timedelta(days=365 * 99)

FY_START_DATE = datetime.date(2020, 3, 1)

FY_MID_DATE = datetime.date(2020, 9, 1)

PAST_DATE = datetime.date(2000, 1, 1)

TESTS_DIR = Path(__file__).parent

def check_post_meta(txn, *expected_meta, default=None):

    assert len(txn.postings) == len(expected_meta)

    for post, expected in zip(txn.postings, expected_meta):

        if not expected:

            assert not post.meta

        else:

            actual = None if post.meta is None else {

                key: post.meta.get(key, default) for key in expected

            }

            assert actual == expected

def combine_values(*value_seqs):

    stop = 0

    for seq in value_seqs:

        try:

            stop = max(stop, len(seq))

        except TypeError:

            pass

    return itertools.islice(

        zip(*(itertools.cycle(seq) for seq in value_seqs)),

        stop,

    )

def parse_date(s, fmt='%Y-%m-%d'):

    return datetime.datetime.strptime(s, fmt).date()

def test_path(s):

    if s is None:

        return s

    s = Path(s)

    if not s.is_absolute():

        s = TESTS_DIR / s

    return s

def Amount(number, currency='USD'):

    return bc_amount.Amount(Decimal(number), currency)

def Cost(number, currency='USD', date=FY_MID_DATE, label=None):

    return bc_data.Cost(Decimal(number), currency, date, label)

def Posting(account, number,

            currency='USD', cost=None, price=None, flag=None,

            type_=bc_data.Posting, **meta):

    if cost is not None:

        cost = Cost(*cost)

    if not meta:

        meta = None

    return type_(

        account,

        Amount(number, currency),

        cost,

        price,

        flag,

        meta,

    )

LINK_METADATA_STRINGS = {

    'Invoices/304321.pdf',

    'rt:123/456',

    'rt://ticket/234',

}

NON_LINK_METADATA_STRINGS = {

    '',

    ' ',

    '     ',

}

NON_STRING_METADATA_VALUES = [

    Decimal(5),

    FY_MID_DATE,