Speedtestdotnet code cleanup and type hints (#52533)

This commit is contained in:
Rami Mosleh 2021-07-22 13:25:54 +03:00 committed by GitHub
parent 7768f53281
commit 1a450c2084
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
11 changed files with 287 additions and 237 deletions

View file

@ -0,0 +1,16 @@
"""Conftest for speedtestdotnet."""
from unittest.mock import patch
import pytest
from tests.components.speedtestdotnet import MOCK_RESULTS, MOCK_SERVERS
@pytest.fixture(autouse=True)
def mock_api():
"""Mock entry setup."""
with patch("speedtest.Speedtest") as mock_api:
mock_api.return_value.get_servers.return_value = MOCK_SERVERS
mock_api.return_value.get_best_server.return_value = MOCK_SERVERS[1][0]
mock_api.return_value.results.dict.return_value = MOCK_RESULTS
yield mock_api