python-kasa · rytilahti · Mar 6, 2024 · Mar 5, 2024 · Mar 5, 2024 · Mar 5, 2024
diff --git a/devtools/dump_devinfo.py b/devtools/dump_devinfo.py
@@ -39,6 +39,7 @@
 FixtureResult = namedtuple("FixtureResult", "filename, folder, data")
 
 SMART_FOLDER = "kasa/tests/fixtures/smart/"
+SMART_CHILD_FOLDER = "kasa/tests/fixtures/smart/child/"
 IOT_FOLDER = "kasa/tests/fixtures/"
 
 _LOGGER = logging.getLogger(__name__)
@@ -531,7 +532,9 @@ def get_smart_child_fixture(response):
         model += f"({region})"
 
     save_filename = f"{model}_{hw_version}_{sw_version}.json"
-    return FixtureResult(filename=save_filename, folder=SMART_FOLDER, data=response)
+    return FixtureResult(
+        filename=save_filename, folder=SMART_CHILD_FOLDER, data=response
+    )
 
 
 async def get_smart_fixtures(device: SmartDevice, batch_size: int):

diff --git a/kasa/tests/device_fixtures.py b/kasa/tests/device_fixtures.py
@@ -277,7 +277,7 @@ def check_categories():
 
 
 def device_for_fixture_name(model, protocol):
-    if protocol == "SMART":
+    if "SMART" in protocol:
         for d in PLUGS_SMART:
             if d in model:
                 return SmartDevice
@@ -345,17 +345,21 @@ async def get_device_for_fixture(fixture_data: FixtureInfo):
     d = device_for_fixture_name(fixture_data.name, fixture_data.protocol)(
         host="127.0.0.123"
     )
-    if fixture_data.protocol == "SMART":
+    if "SMART" in fixture_data.protocol:
         d.protocol = FakeSmartProtocol(fixture_data.data, fixture_data.name)
     else:
         d.protocol = FakeIotProtocol(fixture_data.data)
+
+    discovery_data = None
     if "discovery_result" in fixture_data.data:
         discovery_data = {"result": fixture_data.data["discovery_result"]}
-    else:
+    elif "system" in fixture_data.data:
         discovery_data = {
             "system": {"get_sysinfo": fixture_data.data["system"]["get_sysinfo"]}
         }
-    d.update_from_discover_info(discovery_data)
+    if discovery_data:  # Child devices do not have discovery info
+        d.update_from_discover_info(discovery_data)
+
     await _update_and_close(d)
     return d
 

diff --git a/kasa/tests/discovery_fixtures.py b/kasa/tests/discovery_fixtures.py
@@ -8,7 +8,7 @@
 
 from .fakeprotocol_iot import FakeIotProtocol
 from .fakeprotocol_smart import FakeSmartProtocol
-from .fixtureinfo import FIXTURE_DATA, FixtureInfo, filter_fixtures, idgenerator
+from .fixtureinfo import FixtureInfo, filter_fixtures, idgenerator
 
 
 def _make_unsupported(device_family, encrypt_type):
@@ -42,8 +42,10 @@ def _make_unsupported(device_family, encrypt_type):
 }
 
 
-def parametrize_discovery(desc, root_key):
-    filtered_fixtures = filter_fixtures(desc, data_root_filter=root_key)
+def parametrize_discovery(desc, *, data_root_filter, protocol_filter=None):
+    filtered_fixtures = filter_fixtures(
+        desc, data_root_filter=data_root_filter, protocol_filter=protocol_filter
+    )
     return pytest.mark.parametrize(
         "discovery_mock",
         filtered_fixtures,
@@ -52,10 +54,15 @@ def parametrize_discovery(desc, root_key):
     )
 
 
-new_discovery = parametrize_discovery("new discovery", "discovery_result")
+new_discovery = parametrize_discovery(
+    "new discovery", data_root_filter="discovery_result"
+)
 
 
-@pytest.fixture(params=FIXTURE_DATA, ids=idgenerator)
+@pytest.fixture(
+    params=filter_fixtures("discoverable", protocol_filter={"SMART", "IOT"}),
+    ids=idgenerator,
+)
 def discovery_mock(request, mocker):
     fixture_info: FixtureInfo = request.param
     fixture_data = fixture_info.data
@@ -128,7 +135,7 @@ async def mock_discover(self):
         side_effect=lambda *_, **__: [(None, None, None, None, (dm.ip, 0))],
     )
 
-    if fixture_info.protocol == "SMART":
+    if "SMART" in fixture_info.protocol:
         proto = FakeSmartProtocol(fixture_data, fixture_info.name)
     else:
         proto = FakeIotProtocol(fixture_data)
@@ -142,7 +149,10 @@ async def _query(request, retry_count: int = 3):
     yield dm
 
 
-@pytest.fixture(params=FIXTURE_DATA, ids=idgenerator)
+@pytest.fixture(
+    params=filter_fixtures("discoverable", protocol_filter={"SMART", "IOT"}),
+    ids=idgenerator,
+)
 def discovery_data(request, mocker):
     """Return raw discovery file contents as JSON. Used for discovery tests."""
     fixture_info = request.param

diff --git a/kasa/tests/fixtureinfo.py b/kasa/tests/fixtureinfo.py
@@ -29,8 +29,17 @@ class FixtureInfo(NamedTuple):
     )
 ]
 
+SUPPORTED_SMART_CHILD_DEVICES = [
+    (device, "SMART.CHILD")
+    for device in glob.glob(
+        os.path.dirname(os.path.abspath(__file__)) + "/fixtures/smart/child/*.json"
+    )
+]
+
 
-SUPPORTED_DEVICES = SUPPORTED_IOT_DEVICES + SUPPORTED_SMART_DEVICES
+SUPPORTED_DEVICES = (
+    SUPPORTED_IOT_DEVICES + SUPPORTED_SMART_DEVICES + SUPPORTED_SMART_CHILD_DEVICES
+)
 
 
 def idgenerator(paramtuple: FixtureInfo):
@@ -50,6 +59,8 @@ def get_fixture_info() -> List[FixtureInfo]:
         folder = Path(__file__).parent / "fixtures"
         if protocol == "SMART":
             folder = folder / "smart"
+        if protocol == "SMART.CHILD":
+            folder = folder / "smart/child"
         p = folder / file
 
         with open(p) as f:

diff --git a/kasa/tests/fixtures/smart/child/.gitkeep b/kasa/tests/fixtures/smart/child/.gitkeep
@@ -0,0 +1 @@
+Can be deleted when first fixture is added