"""Tests for IngestionFeed module and class methods."""
from unittest.mock import patch

from feed_status.models import feed_monitor
from feed_status.models.outgoing_feed import OutgoingFeed


mock_outgoing_feed_config = [
    OutgoingFeed(feed_id=111,
                 feed_name='test',
                 status_source_db='feed_status',
                 update_freq=1),
    OutgoingFeed(feed_id='proper_new_releases_tracks',
                 feed_name='proper_new_releases_tracks',
                 status_source_db='feed_status',
                 update_freq=1)
]


class TestGetOutgoingFeedById:
    """Unit tests for get_feed_by_id()."""

    @patch('feed_status.models.outgoing_feed.active_feeds',
           mock_outgoing_feed_config)
    def test_find_by_feed_id_int(self):
        """Test we can find Feeds by feed_id."""
        feed = feed_monitor.get_feed_by_id(111)
        assert type(feed) is OutgoingFeed

    @patch('feed_status.models.outgoing_feed.active_feeds',
           mock_outgoing_feed_config)
    def test_find_by_feed_id_str(self):
        """Test we can find Feed by feed_id."""
        feed = feed_monitor.get_feed_by_id('proper_new_releases_tracks')
        assert type(feed) is OutgoingFeed