evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Move imports in dsmr component (#27974)

Browse source 
* Move imports in dsmr component

* Review

* Fix tests
This commit is contained in:
Diefferson Koderer Môro 2019-10-22 16:50:49 +00:00 • committed by cgtobi
parent 0226b76e0a
commit 04dbe5bc84
2 changed files with 12 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/components/dsmr/test_sensor.py
View file

@ -11,9 +11,11 @@ from decimal import Decimal
from unittest.mock import Mock
import asynctest
import pytest
from homeassistant.bootstrap import async_setup_component
from homeassistant.components.dsmr.sensor import DerivativeDSMREntity
import pytest
from tests.common import assert_setup_component
@ -34,10 +36,11 @@ def mock_connection_factory(monkeypatch):
# apply the mock to both connection factories
monkeypatch.setattr(
"dsmr_parser.clients.protocol.create_dsmr_reader", connection_factory
"homeassistant.components.dsmr.sensor.create_dsmr_reader", connection_factory
)
monkeypatch.setattr(
"dsmr_parser.clients.protocol.create_tcp_dsmr_reader", connection_factory
"homeassistant.components.dsmr.sensor.create_tcp_dsmr_reader",
connection_factory,
)
return connection_factory, transport, protocol
@ -158,7 +161,8 @@ def test_connection_errors_retry(hass, monkeypatch, mock_connection_factory):
)
monkeypatch.setattr(
"dsmr_parser.clients.protocol.create_dsmr_reader", first_fail_connection_factory
"homeassistant.components.dsmr.sensor.create_dsmr_reader",
first_fail_connection_factory,
)
yield from async_setup_component(hass, "sensor", {"sensor": config})