"""Unit test for util functions for generic os tools."""
import os
import shutil
from tempfile import gettempdir

from feed_ingestion.util import os_tools


def test_date_not_in_temp_dir():
    """Test creating temporary directory for particular flow."""
    feed_name = 'test_date_not_in_temp_dir'
    date = '2020-04-01'
    wrong_date = '2021-04-01'

    os_tools.create_temp_dir(feed_name=feed_name, date=date)
    test_dir_list = os.listdir(os.path.join(gettempdir(), feed_name))

    assert date in test_dir_list
    assert wrong_date not in test_dir_list
    shutil.rmtree(os.path.join(gettempdir(), feed_name))


def test_create_temp_dir():
    """Test creating temporary directory for particular flow."""
    feed_name = 'test_create_temp_dir'
    date1 = '2020-04-01'
    date2 = '2020-05-09'
    date3 = '2020-03-21'
    date4 = '2021-02-11'
    date5 = '2020-04-01'
    os_tools.create_temp_dir(feed_name=feed_name, date=date1)
    os_tools.create_temp_dir(feed_name=feed_name, date=date2)
    os_tools.create_temp_dir(feed_name=feed_name, date=date3)
    os_tools.create_temp_dir(feed_name=feed_name, date=date4)
    os_tools.create_temp_dir(feed_name=feed_name, date=date5)
    test_dir_list = os.listdir(os.path.join(gettempdir(), feed_name))

    assert date1 in test_dir_list
    assert date2 in test_dir_list
    assert date3 in test_dir_list
    assert date4 in test_dir_list
    assert date4 in test_dir_list
    shutil.rmtree(os.path.join(gettempdir(), feed_name))