"""Unit tests for Campaign model."""

from podcast.models import campaign as campaign_model


def test_get_all_campaigns():
    """Test get all campaigns."""
    results = campaign_model.get_all_campaigns()
    assert len(results) == 2
    assert results[0]['name'] == 'Random Campaign title'
    assert results[1]['name'] == 'Another Campaign title'


def test_delete_campaign_by_megaphone_ids():
    """Test soft delete campaigns by megaphone ids."""
    results = campaign_model.delete_campaign_by_megaphone_ids(['1', '2'])
    assert results[0]['is_deleted'] is True
    assert results[1]['is_deleted'] is True