"""Test dispatcher helpers."""
import asyncio

from homeassistant.core import callback
from homeassistant.helpers.dispatcher import (
    async_dispatcher_connect,
    dispatcher_send,
    dispatcher_connect,
)

from tests.common import get_test_home_assistant


class TestHelpersDispatcher:
    """Tests for discovery helper methods."""

    def setup_method(self, method):
        """Set up things to be run when tests are started."""
        self.hass = get_test_home_assistant()

    def teardown_method(self, method):
        """Stop everything that was started."""
        self.hass.stop()

    def test_simple_function(self):
        """Test simple function (executor)."""
        calls = []

        def test_funct(data):
            """Test function."""
            calls.append(data)

        dispatcher_connect(self.hass, "test", test_funct)
        dispatcher_send(self.hass, "test", 3)
        self.hass.block_till_done()

        assert calls == [3]

        dispatcher_send(self.hass, "test", "bla")
        self.hass.block_till_done()

        assert calls == [3, "bla"]

    def test_simple_function_unsub(self):
        """Test simple function (executor) and unsub."""
        calls1 = []
        calls2 = []

        def test_funct1(data):
            """Test function."""
            calls1.append(data)

        def test_funct2(data):
            """Test function."""
            calls2.append(data)

        dispatcher_connect(self.hass, "test1", test_funct1)
        unsub = dispatcher_connect(self.hass, "test2", test_funct2)
        dispatcher_send(self.hass, "test1", 3)
        dispatcher_send(self.hass, "test2", 4)
        self.hass.block_till_done()

        assert calls1 == [3]
        assert calls2 == [4]

        unsub()

        dispatcher_send(self.hass, "test1", 5)
        dispatcher_send(self.hass, "test2", 6)
        self.hass.block_till_done()

        assert calls1 == [3, 5]
        assert calls2 == [4]

        # check don't kill the flow
        unsub()

        dispatcher_send(self.hass, "test1", 7)
        dispatcher_send(self.hass, "test2", 8)
        self.hass.block_till_done()

        assert calls1 == [3, 5, 7]
        assert calls2 == [4]

    def test_simple_callback(self):
        """Test simple callback (async)."""
        calls = []

        @callback
        def test_funct(data):
            """Test function."""
            calls.append(data)

        dispatcher_connect(self.hass, "test", test_funct)
        dispatcher_send(self.hass, "test", 3)
        self.hass.block_till_done()

        assert calls == [3]

        dispatcher_send(self.hass, "test", "bla")
        self.hass.block_till_done()

        assert calls == [3, "bla"]

    def test_simple_coro(self):
        """Test simple coro (async)."""
        calls = []

        @asyncio.coroutine
        def test_funct(data):
            """Test function."""
            calls.append(data)

        dispatcher_connect(self.hass, "test", test_funct)
        dispatcher_send(self.hass, "test", 3)
        self.hass.block_till_done()

        assert calls == [3]

        dispatcher_send(self.hass, "test", "bla")
        self.hass.block_till_done()

        assert calls == [3, "bla"]

    def test_simple_function_multiargs(self):
        """Test simple function (executor)."""
        calls = []

        def test_funct(data1, data2, data3):
            """Test function."""
            calls.append(data1)
            calls.append(data2)
            calls.append(data3)

        dispatcher_connect(self.hass, "test", test_funct)
        dispatcher_send(self.hass, "test", 3, 2, "bla")
        self.hass.block_till_done()

        assert calls == [3, 2, "bla"]


async def test_callback_exception_gets_logged(hass, caplog):
    """Test exception raised by signal handler."""

    @callback
    def bad_handler(*args):
        """Record calls."""
        raise Exception("This is a bad message callback")

    async_dispatcher_connect(hass, "test", bad_handler)
    dispatcher_send(hass, "test", "bad")
    await hass.async_block_till_done()
    await hass.async_block_till_done()

    assert "Exception in bad_handler when dispatching 'test': ('bad',)" in caplog.text