Use Device instead of SmartDevice where feasible (#1585)

tests/fakeprotocol_iot.py

@@ -190,7 +190,6 @@ AMBIENT_MODULE = {
     },
 }
 
-
 MOTION_MODULE = {
     "get_adc_value": {"value": 50, "err_code": 0},
     "get_config": {

tests/smart/modules/test_childsetup.py

@@ -5,7 +5,7 @@ import logging
 import pytest
 from pytest_mock import MockerFixture
 
-from kasa import Feature, Module, SmartDevice
+from kasa import Device, Feature, Module
 
 from ...device_fixtures import parametrize
 
@@ -15,7 +15,7 @@ childsetup = parametrize(
 
 
 @childsetup
-async def test_childsetup_features(dev: SmartDevice):
+async def test_childsetup_features(dev: Device):
     """Test the exposed features."""
     cs = dev.modules.get(Module.ChildSetup)
     assert cs
@@ -27,7 +27,7 @@ async def test_childsetup_features(dev: SmartDevice):
 
 @childsetup
 async def test_childsetup_pair(
-    dev: SmartDevice, mocker: MockerFixture, caplog: pytest.LogCaptureFixture
+    dev: Device, mocker: MockerFixture, caplog: pytest.LogCaptureFixture
 ):
     """Test device pairing."""
     caplog.set_level(logging.INFO)
@@ -51,7 +51,7 @@ async def test_childsetup_pair(
 
 @childsetup
 async def test_childsetup_unpair(
-    dev: SmartDevice, mocker: MockerFixture, caplog: pytest.LogCaptureFixture
+    dev: Device, mocker: MockerFixture, caplog: pytest.LogCaptureFixture
 ):
     """Test unpair."""
     mock_query_helper = mocker.spy(dev, "_query_helper")

tests/smart/modules/test_powerprotection.py

@@ -1,7 +1,7 @@
 import pytest
 from pytest_mock import MockerFixture
 
-from kasa import Module, SmartDevice
+from kasa import Device, Module
 
 from ...device_fixtures import get_parent_and_child_modules, parametrize
 
@@ -35,7 +35,7 @@ async def test_features(dev, feature, prop_name, type):
 
 
 @powerprotection
-async def test_set_enable(dev: SmartDevice, mocker: MockerFixture):
+async def test_set_enable(dev: Device, mocker: MockerFixture):
     """Test enable."""
     powerprot = next(get_parent_and_child_modules(dev, Module.PowerProtection))
     assert powerprot
@@ -88,7 +88,7 @@ async def test_set_enable(dev: SmartDevice, mocker: MockerFixture):
 
 
 @powerprotection
-async def test_set_threshold(dev: SmartDevice, mocker: MockerFixture):
+async def test_set_threshold(dev: Device, mocker: MockerFixture):
     """Test enable."""
     powerprot = next(get_parent_and_child_modules(dev, Module.PowerProtection))
     assert powerprot

tests/smartcam/modules/test_childsetup.py

@@ -5,7 +5,7 @@ import logging
 import pytest
 from pytest_mock import MockerFixture
 
-from kasa import Feature, Module, SmartDevice
+from kasa import Device, Feature, Module
 
 from ...device_fixtures import parametrize
 
@@ -15,7 +15,7 @@ childsetup = parametrize(
 
 
 @childsetup
-async def test_childsetup_features(dev: SmartDevice):
+async def test_childsetup_features(dev: Device):
     """Test the exposed features."""
     cs = dev.modules[Module.ChildSetup]
 
@@ -26,7 +26,7 @@ async def test_childsetup_features(dev: SmartDevice):
 
 @childsetup
 async def test_childsetup_pair(
-    dev: SmartDevice, mocker: MockerFixture, caplog: pytest.LogCaptureFixture
+    dev: Device, mocker: MockerFixture, caplog: pytest.LogCaptureFixture
 ):
     """Test device pairing."""
     caplog.set_level(logging.INFO)
@@ -69,7 +69,7 @@ async def test_childsetup_pair(
 
 @childsetup
 async def test_childsetup_unpair(
-    dev: SmartDevice, mocker: MockerFixture, caplog: pytest.LogCaptureFixture
+    dev: Device, mocker: MockerFixture, caplog: pytest.LogCaptureFixture
 ):
     """Test unpair."""
     mock_query_helper = mocker.spy(dev, "_query_helper")