core/tests/helpers/test_singleton.py

"""Test singleton helper."""
from unittest.mock import Mock

import pytest

from homeassistant.helpers import singleton


@pytest.fixture
def mock_hass():
    """Mock hass fixture."""
    return Mock(data={})


@pytest.mark.parametrize("result", (object(), {}, []))
async def test_singleton_async(mock_hass, result):
    """Test singleton with async function."""

    @singleton.singleton("test_key")
    async def something(hass):
        return result

    result1 = await something(mock_hass)
    result2 = await something(mock_hass)
    assert result1 is result
    assert result1 is result2
    assert "test_key" in mock_hass.data
    assert mock_hass.data["test_key"] is result1


@pytest.mark.parametrize("result", (object(), {}, []))
def test_singleton(mock_hass, result):
    """Test singleton with function."""

    @singleton.singleton("test_key")
    def something(hass):
        return result

    result1 = something(mock_hass)
    result2 = something(mock_hass)
    assert result1 is result
    assert result1 is result2
    assert "test_key" in mock_hass.data
    assert mock_hass.data["test_key"] is result1