import asyncio
import pytest
from unittest.mock import MagicMock, patch

from aitrader.config import Settings
from aitrader.notify import discord


def test_discord_enabled_with_webhook():
    s = Settings(discord=dict(enabled=True), discord_webhook_url="https://xyz")
    assert discord._enabled(s)


def test_discord_enabled_with_bot():
    s = Settings(discord=dict(enabled=True), discord_bot_token="xyz")
    assert discord._enabled(s)


def test_discord_disabled():
    s = Settings(
        discord=dict(enabled=False),
        discord_webhook_url="https://xyz",
        discord_bot_token="xyz",
    )
    assert not discord._enabled(s)


@patch("aitrader.notify.bot.get_bot_instance")
def test_send_via_bot_success(mock_get_bot):
    mock_bot = MagicMock()
    mock_bot.is_ready.return_value = True

    mock_channel = MagicMock()
    mock_bot.get_channel.return_value = mock_channel

    mock_get_bot.return_value = mock_bot

    s = Settings(discord_channel_id="12345", discord_bot_token="token")

    def run_coro_sync(coro, loop):
        asyncio.run(coro)
        fut = MagicMock()
        fut.result.return_value = None
        return fut

    with patch("asyncio.run_coroutine_threadsafe", side_effect=run_coro_sync) as mock_run:
        res = discord.send_via_bot(s, {"title": "Test"}, channel_type="trades")
        assert res
        mock_run.assert_called_once()
        mock_bot.get_channel.assert_called_once_with(12345)