"""Test segment logic."""

from unittest.mock import MagicMock

import analytics

from notifications.logic import segment


def test_track_unsubscribe(monkeypatch):
    """Test tracking unsubscribe event."""
    monkeypatch.setattr(analytics, 'identify', MagicMock())
    monkeypatch.setattr(analytics, 'track', MagicMock())

    user_id = 'alw:123'
    feed_name = 'label_analytics_digest'
    feed_id = 'vendor_123'

    segment.track_unsubscribe(user_id, feed_name, feed_id)

    analytics.identify.assert_called_once_with(user_id)
    analytics.track.assert_called_once_with(user_id, 'Unsubscribed', {
        'feed_name': feed_name,
        'feed_id': feed_id,
        'user_id': user_id
    })