"""Tests for Period Class."""

from reports.period_reports import PeriodReports


def test_period_create():
    """Test creating a period."""
    period_1 = PeriodReports(1, 'month', 'label')
    assert period_1.period_id == 1
    assert period_1.period_ids == '1'
    assert period_1.interval == 'month'
    assert period_1.account_type == 'label'
    assert not period_1.is_quarter
    assert period_1.prefix == 'schematized_files/1_label_all_all_month/'

    period_2 = PeriodReports(2, 'month', 'subaccount')
    assert period_2.period_id == 2
    assert period_2.period_ids == '2'
    assert period_2.interval == 'month'
    assert period_2.account_type == 'subaccount'
    assert not period_2.is_quarter
    assert period_2.prefix == 'schematized_files/2_subaccount_all_all_month/'

    period_3 = PeriodReports(3, 'month', 'label')
    assert period_3.period_id == 3
    assert period_3.period_ids == '3'
    assert period_3.interval == 'month'
    assert period_3.account_type == 'label'
    assert not period_3.is_quarter
    assert period_3.prefix == 'schematized_files/3_label_all_all_month/'

    period_3_quarter = PeriodReports(3, 'quarter', 'label')
    assert period_3_quarter.period_id == 3
    assert period_3_quarter.period_ids == '1,2,3'
    assert period_3_quarter.interval == 'quarter'
    assert period_3_quarter.account_type == 'label'
    assert period_3_quarter.is_quarter
    assert period_3_quarter.prefix == 'schematized_files/1,2,3_label_all_all_quarter/'  # noqa


def test_period_create_with_string():
    """Test creating a period."""
    period_1 = PeriodReports('1', 'month', 'label')
    assert period_1.period_id == 1
    assert period_1.period_ids == '1'

    period_2 = PeriodReports('2', 'month', 'label')
    assert period_2.period_id == 2
    assert period_2.period_ids == '2'

    period_3 = PeriodReports('3', 'month', 'label')
    assert period_3.period_id == 3
    assert period_3.period_ids == '3'

    period_3_quarter = PeriodReports('3', 'quarter', 'label')
    assert period_3_quarter.period_id == 3
    assert period_3_quarter.period_ids == '1,2,3'