from datetime import datetime
from unittest.mock import patch

import pytest

# from playlist.queries.data_availability import get_max_available_streaming_date
from playlist.queries.fetch_queries import get_max_available_streaming_date

mock_date_str = "2021-01-01"
mock_max_available_streaming_date = datetime.strptime(mock_date_str, "%Y-%m-%d")


class Result:
    def first(self):
        return [mock_max_available_streaming_date]


mock_result = Result()


@pytest.fixture
def mock_snowflake():
    with patch(
        "playlist.connectors.snowflake.SnowflakeQuery.execute",
        return_value=mock_result,
    ) as snowflake:
        yield snowflake


def test_get_max_available_streaming_date(mock_snowflake):
    assert get_max_available_streaming_date() == mock_date_str