Fix AsyncMock imports (#129192)

This commit is contained in:
Marc Mueller 2024-10-26 01:47:27 +02:00 committed by GitHub
parent 10300cc478
commit bdfb47e999
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
8 changed files with 7 additions and 25 deletions

View File

@ -1,7 +1,7 @@
"""AirGradient tests configuration.""" """AirGradient tests configuration."""
from collections.abc import Generator from collections.abc import Generator
from unittest.mock import patch from unittest.mock import AsyncMock, patch
from airgradient import Config, Measures from airgradient import Config, Measures
import pytest import pytest
@ -10,7 +10,6 @@ from homeassistant.components.airgradient.const import DOMAIN
from homeassistant.const import CONF_HOST from homeassistant.const import CONF_HOST
from tests.common import MockConfigEntry, load_fixture from tests.common import MockConfigEntry, load_fixture
from tests.components.smhi.common import AsyncMock
@pytest.fixture @pytest.fixture

View File

@ -1,7 +1,7 @@
"""Cambridge Audio tests configuration.""" """Cambridge Audio tests configuration."""
from collections.abc import Generator from collections.abc import Generator
from unittest.mock import Mock, patch from unittest.mock import AsyncMock, Mock, patch
from aiostreammagic.models import ( from aiostreammagic.models import (
Display, Display,
@ -19,7 +19,6 @@ from homeassistant.components.cambridge_audio.const import DOMAIN
from homeassistant.const import CONF_HOST from homeassistant.const import CONF_HOST
from tests.common import MockConfigEntry, load_fixture, load_json_array_fixture from tests.common import MockConfigEntry, load_fixture, load_json_array_fixture
from tests.components.smhi.common import AsyncMock
@pytest.fixture @pytest.fixture

View File

@ -2,7 +2,7 @@
from collections.abc import Generator from collections.abc import Generator
from typing import Any from typing import Any
from unittest.mock import MagicMock, patch from unittest.mock import AsyncMock, MagicMock, patch
from geniushubclient import GeniusDevice, GeniusZone from geniushubclient import GeniusDevice, GeniusZone
import pytest import pytest
@ -11,7 +11,6 @@ from homeassistant.components.geniushub.const import DOMAIN
from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_TOKEN, CONF_USERNAME from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_TOKEN, CONF_USERNAME
from tests.common import MockConfigEntry, load_json_array_fixture from tests.common import MockConfigEntry, load_json_array_fixture
from tests.components.smhi.common import AsyncMock
@pytest.fixture @pytest.fixture

View File

@ -1,7 +1,7 @@
"""Mastodon tests configuration.""" """Mastodon tests configuration."""
from collections.abc import Generator from collections.abc import Generator
from unittest.mock import patch from unittest.mock import AsyncMock, patch
import pytest import pytest
@ -9,7 +9,6 @@ from homeassistant.components.mastodon.const import CONF_BASE_URL, DOMAIN
from homeassistant.const import CONF_ACCESS_TOKEN, CONF_CLIENT_ID, CONF_CLIENT_SECRET from homeassistant.const import CONF_ACCESS_TOKEN, CONF_CLIENT_ID, CONF_CLIENT_SECRET
from tests.common import MockConfigEntry, load_json_object_fixture from tests.common import MockConfigEntry, load_json_object_fixture
from tests.components.smhi.common import AsyncMock
@pytest.fixture @pytest.fixture

View File

@ -1,7 +1,7 @@
"""Mealie tests configuration.""" """Mealie tests configuration."""
from collections.abc import Generator from collections.abc import Generator
from unittest.mock import patch from unittest.mock import AsyncMock, patch
from aiomealie import ( from aiomealie import (
About, About,
@ -20,7 +20,6 @@ from homeassistant.components.mealie.const import DOMAIN
from homeassistant.const import CONF_API_TOKEN, CONF_HOST from homeassistant.const import CONF_API_TOKEN, CONF_HOST
from tests.common import MockConfigEntry, load_fixture from tests.common import MockConfigEntry, load_fixture
from tests.components.smhi.common import AsyncMock
SHOPPING_LIST_ID = "list-id-1" SHOPPING_LIST_ID = "list-id-1"
SHOPPING_ITEM_NOTE = "Shopping Item 1" SHOPPING_ITEM_NOTE = "Shopping Item 1"

View File

@ -1,7 +1,7 @@
"""NYTGames tests configuration.""" """NYTGames tests configuration."""
from collections.abc import Generator from collections.abc import Generator
from unittest.mock import patch from unittest.mock import AsyncMock, patch
from nyt_games.models import ConnectionsStats, WordleStats from nyt_games.models import ConnectionsStats, WordleStats
import pytest import pytest
@ -10,7 +10,6 @@ from homeassistant.components.nyt_games.const import DOMAIN
from homeassistant.const import CONF_TOKEN from homeassistant.const import CONF_TOKEN
from tests.common import MockConfigEntry, load_fixture from tests.common import MockConfigEntry, load_fixture
from tests.components.smhi.common import AsyncMock
@pytest.fixture @pytest.fixture

View File

@ -1,7 +1,7 @@
"""Smarty tests configuration.""" """Smarty tests configuration."""
from collections.abc import Generator from collections.abc import Generator
from unittest.mock import patch from unittest.mock import AsyncMock, patch
import pytest import pytest
@ -9,7 +9,6 @@ from homeassistant.components.smarty import DOMAIN
from homeassistant.const import CONF_HOST from homeassistant.const import CONF_HOST
from tests.common import MockConfigEntry from tests.common import MockConfigEntry
from tests.components.smhi.common import AsyncMock
@pytest.fixture @pytest.fixture

View File

@ -1,11 +0,0 @@
"""Common test utilities."""
from unittest.mock import Mock
class AsyncMock(Mock):
"""Implements Mock async."""
async def __call__(self, *args, **kwargs):
"""Hack for async support for Mock."""
return super().__call__(*args, **kwargs)