import logging

import pytest
#import unittest
from billing_report.utility import date_range_split_montly


@pytest.fixture
def date_range_1():
    return [
        ('2021-08', '2021-08-01', '2021-09-01'),
        ('2021-09', '2021-09-01', '2021-10-01'),
        ('2021-10', '2021-10-01', '2021-11-01'),
        ('2021-11', '2021-11-01', '2021-12-01'),
        ('2021-12', '2021-12-01', '2022-01-01'),
        ('2022-01', '2022-01-01', '2022-02-01'),
        ('2022-02', '2022-02-01', '2022-03-01'),
        ('2022-03', '2022-03-01', '2022-04-01'),
    ]

@pytest.fixture
def date_range_2():
    return [
        ('2021-08', '2021-08-01', '2021-09-01'),
        ('2021-09', '2021-09-01', '2021-10-01'),
        ('2021-10', '2021-10-01', '2021-11-01'),
        ('2021-11', '2021-11-01', '2021-12-01'),
        ('2021-12', '2021-12-01', '2022-01-01'),
        ('2022-01', '2022-01-01', '2022-02-01'),
        ('2022-02', '2022-02-01', '2022-03-01'),
        ('2022-03', '2022-03-01', '2022-03-15'),
    ]

@pytest.fixture
def logger(caplog):
    logger = logging.getLogger('billing_report')
    logger.setLevel(logging.DEBUG)
    caplog.set_level(logging.DEBUG, logger='billing_report')
    return logger

def test_range_split_montly_1(date_range_1, logger):
    date_range = []
    for k, y, m in date_range_split_montly(logger, '2021-08-01', '2022-04-01'):
        logger.info((k,y,m))
        date_range.append((k,y,m))

    assert date_range == date_range_1

def test_range_split_montly_2(date_range_2, logger):
    date_range = []
    for k, y, m in date_range_split_montly(logger, '2021-08-01', '2022-03-15'):
        logger.info((k,y,m))
        date_range.append((k,y,m))

    assert date_range == date_range_2