discord-rss-bot/tests/test_feeds.py

from __future__ import annotations

import asyncio
import os
import tempfile
from pathlib import Path
from typing import LiteralString
from unittest.mock import MagicMock
from unittest.mock import patch

import pytest
from reader import EntryNotFoundError
from reader import Feed
from reader import FeedNotFoundError
from reader import Reader
from reader import StorageError
from reader import make_reader

from discord_rss_bot import feeds
from discord_rss_bot.feeds import capture_full_page_screenshot
from discord_rss_bot.feeds import create_feed
from discord_rss_bot.feeds import create_screenshot_webhook
from discord_rss_bot.feeds import execute_webhook
from discord_rss_bot.feeds import extract_domain
from discord_rss_bot.feeds import get_entry_delivery_mode
from discord_rss_bot.feeds import get_screenshot_layout
from discord_rss_bot.feeds import get_webhook_url
from discord_rss_bot.feeds import is_youtube_feed
from discord_rss_bot.feeds import screenshot_filename_for_entry
from discord_rss_bot.feeds import send_discord_quest_notification
from discord_rss_bot.feeds import send_entry_to_discord
from discord_rss_bot.feeds import send_to_discord
from discord_rss_bot.feeds import set_entry_as_read
from discord_rss_bot.feeds import should_send_embed_check
from discord_rss_bot.feeds import truncate_webhook_message


def test_send_to_discord() -> None:
    """Test sending to Discord."""
    # Skip early if no webhook URL is configured to avoid a real network request.
    webhook_url: str | None = os.environ.get("TEST_WEBHOOK_URL")
    if not webhook_url:
        pytest.skip("No webhook URL provided.")

    with tempfile.TemporaryDirectory() as temp_dir:
        # Create the temp directory.
        Path.mkdir(Path(temp_dir), exist_ok=True)
        assert Path.exists(Path(temp_dir)), f"The directory '{temp_dir}' should exist."

        # Create a temporary reader.
        reader: Reader = make_reader(url=str(Path(temp_dir) / "test_db.sqlite"))
        assert reader is not None, "The reader should not be None."

        # Add a feed to the reader.
        reader.add_feed("https://www.reddit.com/r/Python/.rss")

        # Update the feed to get the entries.
        reader.update_feeds()

        # Get the feed.
        feed: Feed = reader.get_feed("https://www.reddit.com/r/Python/.rss")
        assert feed is not None, f"The feed should not be None. Got: {feed}"

        assert webhook_url is not None, f"The webhook URL should not be None. Got: {webhook_url}"

        # Add tag to the feed and check if it is there.
        reader.set_tag(feed, "webhook", webhook_url)  # pyright: ignore[reportArgumentType]
        assert reader.get_tag(feed, "webhook") == webhook_url, f"The webhook URL should be '{webhook_url}'."

        # Send the feed to Discord.
        send_to_discord(reader=reader, feed=feed, do_once=True)

        # Close the reader, so we can delete the directory.
        reader.close()


def test_truncate_webhook_message_short_message():
    message = "This is a short message."
    assert_msg = "The message should remain unchanged if it's less than 4000 characters."
    assert truncate_webhook_message(message) == message, assert_msg


def test_truncate_webhook_message_exact_length():
    message: LiteralString = "A" * 4000  # Exact length of max_content_length
    assert_msg: str = f"The message should remain unchanged if it's exactly {4000} characters."
    assert truncate_webhook_message(message) == message, assert_msg


def test_truncate_webhook_message_long_message():
    message: str = "A" * 4100  # Exceeds max_content_length
    truncated_message: str = truncate_webhook_message(message)

    # Ensure the truncated message length is correct
    assert_msg = "The length of the truncated message should be between 3999 and 4000."
    assert 3999 <= len(truncated_message) <= 4000, assert_msg

    # Calculate half length for the truncated parts
    half_length = (4000 - 3) // 2

    # Test the beginning of the message
    assert_msg = "The beginning of the truncated message should match the original message."
    assert truncated_message[:half_length] == "A" * half_length, assert_msg

    # Test the end of the message
    assert_msg = "The end of the truncated message should be '...' to indicate truncation."
    assert truncated_message[-half_length:] == "A" * half_length, assert_msg


def test_is_youtube_feed():
    """Test the is_youtube_feed function."""
    # YouTube feed URLs
    assert is_youtube_feed("https://www.youtube.com/feeds/videos.xml?channel_id=123456") is True
    assert is_youtube_feed("https://www.youtube.com/feeds/videos.xml?user=username") is True

    # Non-YouTube feed URLs
    assert is_youtube_feed("https://www.example.com/feed.xml") is False
    assert is_youtube_feed("https://www.youtube.com/watch?v=123456") is False
    assert is_youtube_feed("https://www.reddit.com/r/Python/.rss") is False


@patch("discord_rss_bot.feeds.logger")
def test_should_send_embed_check_youtube_feeds(mock_logger: MagicMock) -> None:
    """Test should_send_embed_check returns False for YouTube feeds regardless of settings."""
    # Create mocks
    mock_reader = MagicMock()
    mock_entry = MagicMock()

    # Configure a YouTube feed
    mock_entry.feed.url = "https://www.youtube.com/feeds/videos.xml?channel_id=123456"

    # Set reader to return True for should_send_embed (would normally create an embed)
    mock_reader.get_tag.return_value = True

    # Result should be False, overriding the feed settings
    result = should_send_embed_check(mock_reader, mock_entry)
    assert result is False, "YouTube feeds should never use embeds"

    # Function should not even call get_tag for YouTube feeds
    mock_reader.get_tag.assert_not_called()


@patch("discord_rss_bot.feeds.logger")
def test_should_send_embed_check_normal_feeds(mock_logger: MagicMock) -> None:
    """Test should_send_embed_check returns feed settings for non-YouTube feeds."""
    # Create mocks
    mock_reader = MagicMock()
    mock_entry = MagicMock()

    # Configure a normal feed
    mock_entry.feed.url = "https://www.example.com/feed.xml"

    # Test with should_send_embed set to True
    mock_reader.get_tag.return_value = True
    result = should_send_embed_check(mock_reader, mock_entry)
    assert result is True, "Normal feeds should use embeds when enabled"

    # Test with should_send_embed set to False
    mock_reader.get_tag.return_value = False
    result = should_send_embed_check(mock_reader, mock_entry)
    assert result is False, "Normal feeds should not use embeds when disabled"


def test_get_entry_delivery_mode_prefers_delivery_mode_tag() -> None:
    reader = MagicMock()
    entry = MagicMock()
    entry.feed.url = "https://example.com/feed.xml"

    reader.get_tag.side_effect = lambda resource, key, default=None: {  # noqa: ARG005
        "delivery_mode": "screenshot",
        "should_send_embed": True,
    }.get(key, default)

    result = get_entry_delivery_mode(reader, entry)

    assert result == "screenshot"


def test_get_entry_delivery_mode_falls_back_to_legacy_embed_flag() -> None:
    reader = MagicMock()
    entry = MagicMock()
    entry.feed.url = "https://example.com/feed.xml"

    reader.get_tag.side_effect = lambda resource, key, default=None: {  # noqa: ARG005
        "delivery_mode": "",
        "should_send_embed": False,
    }.get(key, default)

    result = get_entry_delivery_mode(reader, entry)

    assert result == "text"


@patch("discord_rss_bot.feeds.execute_webhook")
@patch("discord_rss_bot.feeds.create_text_webhook")
@patch("discord_rss_bot.feeds.create_hoyolab_webhook")
@patch("discord_rss_bot.feeds.fetch_hoyolab_post")
def test_send_entry_to_discord_hoyolab_text_mode_uses_text_webhook(
    mock_fetch_hoyolab_post: MagicMock,
    mock_create_hoyolab_webhook: MagicMock,
    mock_create_text_webhook: MagicMock,
    mock_execute_webhook: MagicMock,
) -> None:
    entry = MagicMock()
    entry.id = "entry-1"
    entry.feed.url = "https://feeds.c3kay.de/hoyolab.xml"
    entry.feed_url = "https://feeds.c3kay.de/hoyolab.xml"
    entry.link = "https://www.hoyolab.com/article/38588239"

    reader = MagicMock()
    reader.get_tag.side_effect = lambda resource, key, default=None: {  # noqa: ARG005
        "webhook": "https://discord.test/webhook",
        "delivery_mode": "text",
    }.get(key, default)

    text_webhook = MagicMock()
    mock_create_text_webhook.return_value = text_webhook

    result = send_entry_to_discord(entry, reader)

    assert result is None
    mock_fetch_hoyolab_post.assert_not_called()
    mock_create_hoyolab_webhook.assert_not_called()
    mock_create_text_webhook.assert_called_once_with(
        "https://discord.test/webhook",
        entry,
        reader=reader,
        use_default_message_on_empty=False,
    )
    mock_execute_webhook.assert_called_once_with(text_webhook, entry, reader=reader)


@patch("discord_rss_bot.feeds.execute_webhook")
@patch("discord_rss_bot.feeds.create_screenshot_webhook")
@patch("discord_rss_bot.feeds.create_hoyolab_webhook")
@patch("discord_rss_bot.feeds.fetch_hoyolab_post")
def test_send_entry_to_discord_hoyolab_screenshot_mode_uses_screenshot_webhook(
    mock_fetch_hoyolab_post: MagicMock,
    mock_create_hoyolab_webhook: MagicMock,
    mock_create_screenshot_webhook: MagicMock,
    mock_execute_webhook: MagicMock,
) -> None:
    entry = MagicMock()
    entry.id = "entry-2"
    entry.feed.url = "https://feeds.c3kay.de/hoyolab.xml"
    entry.feed_url = "https://feeds.c3kay.de/hoyolab.xml"
    entry.link = "https://www.hoyolab.com/article/38588239"

    reader = MagicMock()
    reader.get_tag.side_effect = lambda resource, key, default=None: {  # noqa: ARG005
        "webhook": "https://discord.test/webhook",
        "delivery_mode": "screenshot",
    }.get(key, default)

    screenshot_webhook = MagicMock()
    mock_create_screenshot_webhook.return_value = screenshot_webhook

    result = send_entry_to_discord(entry, reader)

    assert result is None
    mock_fetch_hoyolab_post.assert_not_called()
    mock_create_hoyolab_webhook.assert_not_called()
    mock_create_screenshot_webhook.assert_called_once_with(
        "https://discord.test/webhook",
        entry,
        reader=reader,
    )
    mock_execute_webhook.assert_called_once_with(screenshot_webhook, entry, reader=reader)


@patch("discord_rss_bot.feeds.execute_webhook")
@patch("discord_rss_bot.feeds.create_embed_webhook")
@patch("discord_rss_bot.feeds.create_hoyolab_webhook")
@patch("discord_rss_bot.feeds.fetch_hoyolab_post")
def test_send_entry_to_discord_hoyolab_embed_mode_uses_hoyolab_webhook(
    mock_fetch_hoyolab_post: MagicMock,
    mock_create_hoyolab_webhook: MagicMock,
    mock_create_embed_webhook: MagicMock,
    mock_execute_webhook: MagicMock,
) -> None:
    entry = MagicMock()
    entry.id = "entry-3"
    entry.feed.url = "https://feeds.c3kay.de/hoyolab.xml"
    entry.feed_url = "https://feeds.c3kay.de/hoyolab.xml"
    entry.link = "https://www.hoyolab.com/article/38588239"

    reader = MagicMock()
    reader.get_tag.side_effect = lambda resource, key, default=None: {  # noqa: ARG005
        "webhook": "https://discord.test/webhook",
        "delivery_mode": "embed",
    }.get(key, default)

    mock_fetch_hoyolab_post.return_value = {"post": {"subject": "News"}}
    hoyolab_webhook = MagicMock()
    mock_create_hoyolab_webhook.return_value = hoyolab_webhook

    result = send_entry_to_discord(entry, reader)

    assert result is None
    mock_fetch_hoyolab_post.assert_called_once_with("38588239")
    mock_create_hoyolab_webhook.assert_called_once_with(
        "https://discord.test/webhook",
        entry,
        {"post": {"subject": "News"}},
    )
    mock_create_embed_webhook.assert_not_called()
    mock_execute_webhook.assert_called_once_with(hoyolab_webhook, entry, reader=reader)


def test_get_screenshot_layout_prefers_mobile_tag() -> None:
    reader = MagicMock()
    feed = MagicMock()
    feed.url = "https://example.com/feed.xml"
    reader.get_tag.return_value = "mobile"

    result = get_screenshot_layout(reader, feed)

    assert result == "mobile"


def test_get_screenshot_layout_defaults_to_desktop() -> None:
    reader = MagicMock()
    feed = MagicMock()
    feed.url = "https://example.com/feed.xml"
    reader.get_tag.return_value = "unknown"

    result = get_screenshot_layout(reader, feed)

    assert result == "desktop"


def test_create_feed_inherits_global_screenshot_layout() -> None:
    reader = MagicMock()
    reader.get_tag.side_effect = lambda resource, key, default=None: {  # noqa: ARG005
        "webhooks": [{"name": "Main", "url": "https://discord.com/api/webhooks/123/abc"}],
        "screenshot_layout": "mobile",
    }.get(key, default)

    create_feed(reader, "https://example.com/feed.xml", "Main")

    reader.set_tag.assert_any_call("https://example.com/feed.xml", "screenshot_layout", "mobile")


def test_create_feed_inherits_global_text_delivery_mode() -> None:
    reader = MagicMock()
    reader.get_tag.side_effect = lambda resource, key, default=None: {  # noqa: ARG005
        "webhooks": [{"name": "Main", "url": "https://discord.com/api/webhooks/123/abc"}],
        "screenshot_layout": "desktop",
        "delivery_mode": "text",
    }.get(key, default)

    create_feed(reader, "https://example.com/feed.xml", "Main")

    reader.set_tag.assert_any_call("https://example.com/feed.xml", "delivery_mode", "text")
    reader.set_tag.assert_any_call("https://example.com/feed.xml", "should_send_embed", False)


def test_create_feed_falls_back_to_embed_when_global_delivery_mode_is_invalid() -> None:
    reader = MagicMock()
    reader.get_tag.side_effect = lambda resource, key, default=None: {  # noqa: ARG005
        "webhooks": [{"name": "Main", "url": "https://discord.com/api/webhooks/123/abc"}],
        "screenshot_layout": "desktop",
        "delivery_mode": "invalid",
    }.get(key, default)

    create_feed(reader, "https://example.com/feed.xml", "Main")

    reader.set_tag.assert_any_call("https://example.com/feed.xml", "delivery_mode", "embed")
    reader.set_tag.assert_any_call("https://example.com/feed.xml", "should_send_embed", True)


@patch("discord_rss_bot.feeds.capture_full_page_screenshot")
@patch("discord_rss_bot.feeds.DiscordWebhook")
def test_create_screenshot_webhook_adds_image_file(
    mock_discord_webhook: MagicMock,
    mock_capture: MagicMock,
) -> None:
    mock_capture.return_value = b"png-bytes"
    webhook = MagicMock()
    mock_discord_webhook.return_value = webhook

    entry = MagicMock()
    entry.id = "entry-abc"
    entry.link = "https://example.com/article"
    reader = MagicMock()
    reader.get_tag.side_effect = lambda resource, key, default=None: {  # noqa: ARG005
        "screenshot_layout": "mobile",
    }.get(key, default)

    result = create_screenshot_webhook("https://discord.com/api/webhooks/123/abc", entry, reader)

    assert result == webhook
    mock_discord_webhook.assert_called_once_with(
        url="https://discord.com/api/webhooks/123/abc",
        content="<https://example.com/article>",
        rate_limit_retry=True,
    )
    mock_capture.assert_called_once_with(
        "https://example.com/article",
        screenshot_layout="mobile",
        screenshot_type="png",
    )
    webhook.add_file.assert_called_once()


@patch("discord_rss_bot.feeds.capture_full_page_screenshot")
@patch("discord_rss_bot.feeds.DiscordWebhook")
def test_create_screenshot_webhook_retries_jpeg_when_png_too_large(
    mock_discord_webhook: MagicMock,
    mock_capture: MagicMock,
) -> None:
    oversized_png = b"x" * (8 * 1024 * 1024 + 1024)
    compressed_jpeg = b"y" * (7 * 1024 * 1024)
    mock_capture.side_effect = [oversized_png, compressed_jpeg]

    webhook = MagicMock()
    mock_discord_webhook.return_value = webhook

    entry = MagicMock()
    entry.id = "entry-large"
    entry.link = "https://example.com/large-article"
    reader = MagicMock()
    reader.get_tag.side_effect = lambda resource, key, default=None: {  # noqa: ARG005
        "screenshot_layout": "desktop",
    }.get(key, default)

    result = create_screenshot_webhook("https://discord.com/api/webhooks/123/abc", entry, reader)

    assert result == webhook
    assert mock_capture.call_count == 2
    assert mock_capture.call_args_list[0].kwargs == {
        "screenshot_layout": "desktop",
        "screenshot_type": "png",
    }
    assert mock_capture.call_args_list[1].kwargs == {
        "screenshot_layout": "desktop",
        "screenshot_type": "jpeg",
        "jpeg_quality": 85,
    }
    webhook.add_file.assert_called_once()


@patch("discord_rss_bot.feeds.create_text_webhook")
@patch("discord_rss_bot.feeds.capture_full_page_screenshot")
def test_create_screenshot_webhook_falls_back_when_all_formats_too_large(
    mock_capture: MagicMock,
    mock_create_text_webhook: MagicMock,
) -> None:
    oversized_bytes = b"z" * (9 * 1024 * 1024)
    # 1 PNG attempt + 4 JPEG quality attempts
    mock_capture.side_effect = [oversized_bytes, oversized_bytes, oversized_bytes, oversized_bytes, oversized_bytes]
    fallback_webhook = MagicMock()
    mock_create_text_webhook.return_value = fallback_webhook

    entry = MagicMock()
    entry.id = "entry-too-large"
    entry.link = "https://example.com/very-large"
    reader = MagicMock()
    reader.get_tag.side_effect = lambda resource, key, default=None: {  # noqa: ARG005
        "screenshot_layout": "desktop",
    }.get(key, default)

    result = create_screenshot_webhook("https://discord.com/api/webhooks/123/abc", entry, reader)

    assert result == fallback_webhook
    assert mock_capture.call_count == 5
    mock_create_text_webhook.assert_called_once_with(
        "https://discord.com/api/webhooks/123/abc",
        entry,
        reader=reader,
        use_default_message_on_empty=True,
    )


@patch("discord_rss_bot.feeds.capture_full_page_screenshot")
@patch("discord_rss_bot.feeds.create_text_webhook")
def test_create_screenshot_webhook_falls_back_when_entry_has_no_link(
    mock_create_text_webhook: MagicMock,
    mock_capture: MagicMock,
) -> None:
    entry = MagicMock()
    entry.id = "entry-no-link"
    entry.link = None
    reader = MagicMock()
    fallback_webhook = MagicMock()
    mock_create_text_webhook.return_value = fallback_webhook

    result = create_screenshot_webhook("https://discord.com/api/webhooks/123/abc", entry, reader)

    assert result == fallback_webhook
    mock_capture.assert_not_called()
    mock_create_text_webhook.assert_called_once_with(
        "https://discord.com/api/webhooks/123/abc",
        entry,
        reader=reader,
        use_default_message_on_empty=True,
    )


def test_screenshot_filename_for_entry_custom_extension() -> None:
    entry = MagicMock()
    entry.id = "hello/world?id=123"

    filename = screenshot_filename_for_entry(entry, extension="JPG")

    assert filename.endswith(".jpg")
    assert "/" not in filename
    assert "?" not in filename


@patch("discord_rss_bot.feeds._capture_full_page_screenshot_sync", return_value=b"jpeg-bytes")
def test_capture_full_page_screenshot_forwards_jpeg_options(mock_capture_sync: MagicMock) -> None:
    result = capture_full_page_screenshot(
        "https://example.com/article",
        screenshot_layout="mobile",
        screenshot_type="jpeg",
        jpeg_quality=55,
    )

    assert result == b"jpeg-bytes"
    mock_capture_sync.assert_called_once_with(
        "https://example.com/article",
        screenshot_layout="mobile",
        screenshot_type="jpeg",
        jpeg_quality=55,
    )


@patch("discord_rss_bot.feeds.create_text_webhook")
@patch("discord_rss_bot.feeds.capture_full_page_screenshot")
def test_create_screenshot_webhook_falls_back_to_text_on_failure(
    mock_capture: MagicMock,
    mock_create_text_webhook: MagicMock,
) -> None:
    mock_capture.return_value = None
    fallback_webhook = MagicMock()
    mock_create_text_webhook.return_value = fallback_webhook

    entry = MagicMock()
    entry.id = "entry-def"
    entry.link = "https://example.com/article"
    reader = MagicMock()

    result = create_screenshot_webhook("https://discord.com/api/webhooks/123/abc", entry, reader)

    assert result == fallback_webhook
    mock_create_text_webhook.assert_called_once_with(
        "https://discord.com/api/webhooks/123/abc",
        entry,
        reader=reader,
        use_default_message_on_empty=True,
    )


def test_capture_full_page_screenshot_uses_thread_when_loop_running() -> None:
    """Capture should offload sync Playwright work when called from an active event loop."""
    with patch("discord_rss_bot.feeds._capture_full_page_screenshot_sync", return_value=b"png") as mock_capture_sync:

        async def run_capture() -> bytes | None:
            return feeds.capture_full_page_screenshot(
                "https://example.com/article",
                screenshot_layout="desktop",
                screenshot_type="png",
            )

        result = asyncio.run(run_capture())

    assert result == b"png"
    mock_capture_sync.assert_called_once_with(
        "https://example.com/article",
        screenshot_layout="desktop",
        screenshot_type="png",
        jpeg_quality=85,
    )


@patch("discord_rss_bot.feeds.get_entry_delivery_mode")
@patch("discord_rss_bot.feeds.create_screenshot_webhook")
@patch("discord_rss_bot.feeds.execute_webhook")
def test_send_entry_to_discord_uses_screenshot_mode(
    mock_execute_webhook: MagicMock,
    mock_create_screenshot_webhook: MagicMock,
    mock_get_entry_delivery_mode: MagicMock,
) -> None:
    reader = MagicMock()
    entry = MagicMock()
    entry.feed.url = "https://example.com/feed.xml"
    entry.feed_url = "https://example.com/feed.xml"

    reader.get_tag.side_effect = lambda resource, key, default=None: {  # noqa: ARG005
        "webhook": "https://discord.com/api/webhooks/123/abc",
    }.get(key, default)

    mock_get_entry_delivery_mode.return_value = "screenshot"
    screenshot_webhook = MagicMock()
    mock_create_screenshot_webhook.return_value = screenshot_webhook

    send_entry_to_discord(entry, reader)

    mock_create_screenshot_webhook.assert_called_once_with(
        "https://discord.com/api/webhooks/123/abc",
        entry,
        reader=reader,
    )
    mock_execute_webhook.assert_called_once_with(screenshot_webhook, entry, reader=reader)


@patch("discord_rss_bot.feeds.get_reader")
@patch("discord_rss_bot.feeds.get_custom_message")
@patch("discord_rss_bot.feeds.replace_tags_in_text_message")
@patch("discord_rss_bot.feeds.create_embed_webhook")
@patch("discord_rss_bot.feeds.DiscordWebhook")
@patch("discord_rss_bot.feeds.execute_webhook")
def test_send_entry_to_discord_youtube_feed(
    mock_execute_webhook: MagicMock,
    mock_discord_webhook: MagicMock,
    mock_create_embed: MagicMock,
    mock_replace_tags: MagicMock,
    mock_get_custom_message: MagicMock,
    mock_get_reader: MagicMock,
):
    """Test send_entry_to_discord function with YouTube feeds."""
    # Set up mocks
    mock_reader = MagicMock()
    mock_get_reader.return_value = mock_reader
    mock_entry = MagicMock()
    mock_feed = MagicMock()

    # Configure a YouTube feed
    mock_entry.feed = mock_feed
    mock_entry.feed.url = "https://www.youtube.com/feeds/videos.xml?channel_id=123456"
    mock_entry.feed_url = "https://www.youtube.com/feeds/videos.xml?channel_id=123456"

    # Mock the tags
    mock_reader.get_tag.side_effect = lambda feed, tag, default=None: {  # noqa: ARG005
        "webhook": "https://discord.com/api/webhooks/123/abc",
        "should_send_embed": True,  # This should be ignored for YouTube feeds
    }.get(tag, default)

    # Mock custom message
    mock_get_custom_message.return_value = "Custom message"
    mock_replace_tags.return_value = "Formatted message with {{entry_link}}"

    # Mock webhook
    mock_webhook = MagicMock()
    mock_discord_webhook.return_value = mock_webhook

    # Call the function
    send_entry_to_discord(mock_entry, mock_reader)

    # Assertions
    mock_create_embed.assert_not_called()
    mock_discord_webhook.assert_called_once()

    # Check webhook was created with the right message
    webhook_call_kwargs = mock_discord_webhook.call_args[1]
    assert "content" in webhook_call_kwargs, "Webhook should have content"
    assert webhook_call_kwargs["url"] == "https://discord.com/api/webhooks/123/abc"

    # Verify execute_webhook was called
    mock_execute_webhook.assert_called_once_with(mock_webhook, mock_entry, reader=mock_reader)


def test_extract_domain_youtube_feed() -> None:
    """Test extract_domain for YouTube feeds."""
    url: str = "https://www.youtube.com/feeds/videos.xml?channel_id=123456"
    assert extract_domain(url) == "YouTube", "YouTube feeds should return 'YouTube' as the domain."


def test_extract_domain_reddit_feed() -> None:
    """Test extract_domain for Reddit feeds."""
    url: str = "https://www.reddit.com/r/Python/.rss"
    assert extract_domain(url) == "Reddit", "Reddit feeds should return 'Reddit' as the domain."


def test_extract_domain_github_feed() -> None:
    """Test extract_domain for GitHub feeds."""
    url: str = "https://www.github.com/user/repo"
    assert extract_domain(url) == "GitHub", "GitHub feeds should return 'GitHub' as the domain."


def test_extract_domain_custom_domain() -> None:
    """Test extract_domain for custom domains."""
    url: str = "https://www.example.com/feed"
    assert extract_domain(url) == "Example", "Custom domains should return the capitalized first part of the domain."


def test_extract_domain_no_www_prefix() -> None:
    """Test extract_domain removes 'www.' prefix."""
    url: str = "https://www.example.com/feed"
    assert extract_domain(url) == "Example", "The 'www.' prefix should be removed from the domain."


def test_extract_domain_no_tld() -> None:
    """Test extract_domain for domains without a TLD."""
    url: str = "https://localhost/feed"
    assert extract_domain(url) == "Localhost", "Domains without a TLD should return the capitalized domain."


def test_extract_domain_invalid_url() -> None:
    """Test extract_domain for invalid URLs."""
    url: str = "not-a-valid-url"
    assert extract_domain(url) == "Other", "Invalid URLs should return 'Other' as the domain."


def test_extract_domain_empty_url() -> None:
    """Test extract_domain for empty URLs."""
    url: str = ""
    assert extract_domain(url) == "Other", "Empty URLs should return 'Other' as the domain."


def test_extract_domain_special_characters() -> None:
    """Test extract_domain for URLs with special characters."""
    url: str = "https://www.ex-ample.com/feed"
    assert extract_domain(url) == "Ex-ample", "Domains with special characters should return the capitalized domain."


@pytest.mark.parametrize(
    argnames=("url", "expected"),
    argvalues=[
        ("https://blog.something.com", "Something"),
        ("https://www.something.com", "Something"),
        ("https://subdomain.example.co.uk", "Example"),
        ("https://github.com/user/repo", "GitHub"),
        ("https://youtube.com/feeds/videos.xml?channel_id=abc", "YouTube"),
        ("https://reddit.com/r/python/.rss", "Reddit"),
        ("", "Other"),
        ("not a url", "Other"),
        ("https://www.example.com", "Example"),
        ("https://foo.bar.baz.com", "Baz"),
    ],
)
def test_extract_domain(url: str, expected: str) -> None:
    assert extract_domain(url) == expected


@patch("discord_rss_bot.feeds.execute_webhook")
def test_send_discord_quest_notification_text_match(mock_execute_webhook: MagicMock) -> None:
    """Send a quest link as a separate notification when plain text content contains one."""
    entry = MagicMock()
    entry.id = "entry-1"
    entry.content = [MagicMock(type="text", value="Check this https://discord.com/quests/12345 now")]
    reader = MagicMock()

    send_discord_quest_notification(entry, "https://discord.com/api/webhooks/123/abc", reader)

    mock_execute_webhook.assert_called_once()
    webhook_sent = mock_execute_webhook.call_args[0][0]
    assert webhook_sent.content == "https://discord.com/quests/12345"


@patch("discord_rss_bot.feeds.execute_webhook")
def test_send_discord_quest_notification_html_match(mock_execute_webhook: MagicMock) -> None:
    """Send a quest link when it is found inside HTML content."""
    entry = MagicMock()
    entry.id = "entry-2"
    entry.content = [
        MagicMock(
            type="text/html",
            value='<p>Click <a href="https://discord.com/quests/777">here</a></p>',
        ),
    ]
    reader = MagicMock()

    send_discord_quest_notification(entry, "https://discord.com/api/webhooks/123/abc", reader)

    mock_execute_webhook.assert_called_once()
    webhook_sent = mock_execute_webhook.call_args[0][0]
    assert webhook_sent.content == "https://discord.com/quests/777"


@patch("discord_rss_bot.feeds.execute_webhook")
def test_send_discord_quest_notification_no_match(mock_execute_webhook: MagicMock) -> None:
    """Do nothing when no quest URL exists in entry content."""
    entry = MagicMock()
    entry.id = "entry-3"
    entry.content = [MagicMock(type="text", value="No quest link here")]
    reader = MagicMock()

    send_discord_quest_notification(entry, "https://discord.com/api/webhooks/123/abc", reader)

    mock_execute_webhook.assert_not_called()


def test_get_webhook_url_returns_value() -> None:
    reader = MagicMock()
    entry = MagicMock()
    entry.feed_url = "https://example.com/feed.xml"
    entry.feed.url = "https://example.com/feed.xml"
    reader.get_tag.return_value = "https://discord.com/api/webhooks/123/abc"

    result = get_webhook_url(reader, entry)

    assert result == "https://discord.com/api/webhooks/123/abc"


def test_get_webhook_url_returns_empty_on_storage_error() -> None:
    reader = MagicMock()
    entry = MagicMock()
    entry.feed_url = "https://example.com/feed.xml"
    entry.feed.url = "https://example.com/feed.xml"
    reader.get_tag.side_effect = StorageError("db error")

    result = get_webhook_url(reader, entry)

    assert not result


def test_set_entry_as_read_handles_entry_not_found_error() -> None:
    reader = MagicMock()
    entry = MagicMock(id="entry-4")
    reader.set_entry_read.side_effect = EntryNotFoundError("https://example.com/feed.xml", "entry-4")

    set_entry_as_read(reader, entry)

    reader.set_entry_read.assert_called_once_with(entry, True)


def test_set_entry_as_read_handles_storage_error() -> None:
    reader = MagicMock()
    entry = MagicMock(id="entry-5")
    reader.set_entry_read.side_effect = StorageError("db error")

    set_entry_as_read(reader, entry)

    reader.set_entry_read.assert_called_once_with(entry, True)


def test_execute_webhook_skips_when_feed_paused() -> None:
    webhook = MagicMock()
    reader = MagicMock()
    entry = MagicMock()
    entry.id = "entry-6"
    entry.feed.url = "https://example.com/feed.xml"
    entry.feed.updates_enabled = False

    execute_webhook(webhook, entry, reader)

    reader.get_feed.assert_not_called()
    webhook.execute.assert_not_called()


def test_execute_webhook_skips_when_feed_missing() -> None:
    webhook = MagicMock()
    reader = MagicMock()
    reader.get_feed.side_effect = FeedNotFoundError("missing")
    entry = MagicMock()
    entry.id = "entry-7"
    entry.feed.url = "https://example.com/feed.xml"
    entry.feed.updates_enabled = True

    execute_webhook(webhook, entry, reader)

    webhook.execute.assert_not_called()


@patch.object(feeds, "logger")
def test_execute_webhook_logs_error_on_bad_status(mock_logger: MagicMock) -> None:
    webhook = MagicMock()
    webhook.json = {"content": "test"}
    webhook.execute.return_value = MagicMock(status_code=500, text="fail")
    reader = MagicMock()
    entry = MagicMock()
    entry.id = "entry-8"
    entry.feed.url = "https://example.com/feed.xml"
    entry.feed.updates_enabled = True

    execute_webhook(webhook, entry, reader)

    mock_logger.error.assert_called_once()


@patch.object(feeds, "logger")
def test_execute_webhook_logs_info_on_success(mock_logger: MagicMock) -> None:
    webhook = MagicMock()
    webhook.execute.return_value = MagicMock(status_code=204, text="")
    reader = MagicMock()
    entry = MagicMock()
    entry.id = "entry-9"
    entry.feed.url = "https://example.com/feed.xml"
    entry.feed.updates_enabled = True

    execute_webhook(webhook, entry, reader)

    mock_logger.info.assert_called_once_with("Sent entry to Discord: %s", "entry-9")