Improve type hints in google_pubsub tests (#121172)

* Improve type hints in google_pubsub tests

* Remove from .coveragerc
This commit is contained in:
epenet 2024-07-04 11:31:35 +02:00 committed by GitHub
parent aa74ad0061
commit 8aedb1201d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 8 deletions

View File

@ -487,7 +487,6 @@ omit =
homeassistant/components/goodwe/sensor.py
homeassistant/components/google_cloud/tts.py
homeassistant/components/google_maps/device_tracker.py
homeassistant/components/google_pubsub/__init__.py
homeassistant/components/gpsd/__init__.py
homeassistant/components/gpsd/sensor.py
homeassistant/components/greenwave/light.py

View File

@ -1,9 +1,10 @@
"""The tests for the Google Pub/Sub component."""
from collections.abc import Generator
from dataclasses import dataclass
from datetime import datetime
import os
from unittest import mock
from unittest.mock import MagicMock, Mock, patch
import pytest
@ -40,21 +41,21 @@ async def test_nested() -> None:
@pytest.fixture(autouse=True, name="mock_client")
def mock_client_fixture():
def mock_client_fixture() -> Generator[MagicMock]:
"""Mock the pubsub client."""
with mock.patch(f"{GOOGLE_PUBSUB_PATH}.PublisherClient") as client:
with patch(f"{GOOGLE_PUBSUB_PATH}.PublisherClient") as client:
setattr(
client,
"from_service_account_json",
mock.MagicMock(return_value=mock.MagicMock()),
MagicMock(return_value=MagicMock()),
)
yield client
@pytest.fixture(autouse=True, name="mock_is_file")
def mock_is_file_fixture():
def mock_is_file_fixture() -> Generator[MagicMock]:
"""Mock os.path.isfile."""
with mock.patch(f"{GOOGLE_PUBSUB_PATH}.os.path.isfile") as is_file:
with patch(f"{GOOGLE_PUBSUB_PATH}.os.path.isfile") as is_file:
is_file.return_value = True
yield is_file
@ -63,7 +64,7 @@ def mock_is_file_fixture():
def mock_json(hass, monkeypatch):
"""Mock the event bus listener and os component."""
monkeypatch.setattr(
f"{GOOGLE_PUBSUB_PATH}.json.dumps", mock.Mock(return_value=mock.MagicMock())
f"{GOOGLE_PUBSUB_PATH}.json.dumps", Mock(return_value=MagicMock())
)