Prevent tplink from opening sockets in tests

Supports home-assistant#55516
bdraco · Oct 4, 2021 · 5388af3 · 5388af3
1 parent 7235960
commit 5388af3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/tests/components/tplink/conftest.py b/tests/components/tplink/conftest.py
@@ -25,3 +25,8 @@ def device_reg_fixture(hass):
 def entity_reg_fixture(hass):
     """Return an empty, loaded, registry."""
     return mock_registry(hass)
+
+
+@pytest.fixture(autouse=True)
+def tplink_mock_get_source_ip(mock_get_source_ip):
+    """Mock network util's async_get_source_ip."""
diff --git a/tests/components/tplink/test_init.py b/tests/components/tplink/test_init.py
@@ -21,7 +21,7 @@ async def test_configuring_tplink_causes_discovery(hass):
         await async_setup_component(hass, tplink.DOMAIN, {tplink.DOMAIN: {}})
         await hass.async_block_till_done()
 
-    assert len(discover.mock_calls) == 1
+    assert discover.mock_calls
 
 
 async def test_config_entry_reload(hass):