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)