Fix diagnostics tests

home-assistant · Jan 20, 2022 · bde198e · bde198e
1 parent 048014a
commit bde198e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/components/diagnostics/__init__.py b/tests/components/diagnostics/__init__.py
@@ -18,6 +18,8 @@ async def get_diagnostics_for_config_entry(hass, hass_client, config_entry):
 
 async def get_diagnostics_for_device(hass, hass_client, config_entry, device):
     """Return the diagnostics for the specified device."""
+    assert await async_setup_component(hass, "diagnostics", {})
+
     client = await hass_client()
     response = await client.get(
         f"/api/diagnostics/config_entry/{config_entry.entry_id}/device/{device.id}"