"""Unit tests for utils of YouTube Monthly Ingestion Workflow."""
from datetime import datetime

from feed_ingestion.flows.youtube_monthly import util


def test_get_first_last_day():
    """Make sure the first and last day are correct."""
    given_date = datetime.strptime('2018-02-25', '%Y-%m-%d')
    first_day, last_day = util.get_first_last_day(given_date)

    assert first_day.day == 1
    assert last_day.day == 28