"""Tests for Period model."""

from models import period
import mysql
import test_utils


@test_utils.test_schema
def test_period_to_dict(period_data):
    """Test Period to_dict."""
    new_period_obj = period.Period(**period_data)
    with mysql.ar_db_session() as session:
        session.add(new_period_obj)
        period_obj = session.query(
            period.Period).first()
    assert period_obj.to_dict() == {
        'period_id': 1,
        'year': 2000,
        'quarter': 1,
        'month': 2,
        'status': 'open'
    }


@test_utils.test_schema
def test_get_all_periods(period_data):
    """Test Period get_all_periods function."""
    test_period_objects = [
        period.Period(**period_data), period.Period(**period_data),
        period.Period(**period_data)]
    with mysql.ar_db_session() as session:
        for obj in test_period_objects:
            session.add(obj)
    result = period.get_all_periods()
    assert len(result) == 3
    assert result[0] == {
        'period_id': 1,
        'year': 2000,
        'quarter': 1,
        'month': 2,
        'status': 'open'
    }