From 510a3ae9154144a120c24d58ea0f7822f8927c0d Mon Sep 17 00:00:00 2001 From: Marc Mueller <30130371+cdce8p@users.noreply.github.com> Date: Sun, 25 Apr 2021 20:16:38 +0200 Subject: [PATCH] Improve zeroconf test fixture (#49657) --- tests/components/conftest.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/tests/components/conftest.py b/tests/components/conftest.py index 3b1781ba510..9e029e159a1 100644 --- a/tests/components/conftest.py +++ b/tests/components/conftest.py @@ -3,12 +3,15 @@ from unittest.mock import patch import pytest -from homeassistant.components import zeroconf -zeroconf.orig_install_multiple_zeroconf_catcher = ( - zeroconf.install_multiple_zeroconf_catcher -) -zeroconf.install_multiple_zeroconf_catcher = lambda zc: None +@pytest.fixture(scope="session", autouse=True) +def patch_zeroconf_multiple_catcher(): + """Patch zeroconf wrapper that detects if multiple instances are used.""" + with patch( + "homeassistant.components.zeroconf.install_multiple_zeroconf_catcher", + side_effect=lambda zc: None, + ): + yield @pytest.fixture(autouse=True)