"""Unit tests for the generate_label_digest task."""

from unittest.mock import MagicMock

from activity_detector import base_config
from activity_detector.flows.analytics_digest import utils
from activity_detector.flows.analytics_digest.tasks import \
    generate_label_digest


def test_generate_label_digest(monkeypatch):
    """Test generating the digest."""
    monkeypatch.setattr(utils, 'generate_digest', MagicMock())

    start_date = '2018-01-01'
    end_date = '2018-01-08'
    last_dates_per_store = [
        {
            'store_name': 'Spotify',
            'last_date': '2017-12-25'
        },
        {
            'store_name': 'Itunes',
            'last_date': '2017-12-26'
        }
    ]
    generate_label_digest.generate_label_digest(
        None,
        start_date,
        end_date,
        None,
        last_dates_per_store)

    utils.generate_digest.assert_called_with(
        base_config.ACCOUNT_TYPE_LABEL,
        start_date,
        end_date,
        None,
        last_dates_per_store)