"""Base artifacts for spotify_artificial_streams flow tasks' tests."""

__all__ = [
    'BaseSpotifyArtificialStreamsTestTask',
]

from datetime import date
from unittest import TestCase
from unittest.mock import MagicMock


class BaseSpotifyArtificialStreamsTestTask(TestCase):
    """Base class for spotify_artificial_streams flow tasks' test cases."""

    def setUp(self):
        """Initialize common properties."""
        self.activity = MagicMock()
        self.string_date = date.today().isoformat()
        self.sfdb_params = {
            'account': 'account',
            'warehouse': 'warehouse',
            'database': 'db',
            'schema': 'schema',
            'role': 'role',
        }