From 62c55e6698a7b5e51a6acdd6ba303ddb4789a75b Mon Sep 17 00:00:00 2001
From: Shay Levy <levyshay1@gmail.com>
Date: Thu, 20 Jan 2022 22:23:08 +0200
Subject: [PATCH] Fix atag slow tests (#64567)

* Fix atag slow tests

* Fix sleep fixture
---
 tests/components/atag/conftest.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)
 create mode 100644 tests/components/atag/conftest.py

diff --git a/tests/components/atag/conftest.py b/tests/components/atag/conftest.py
new file mode 100644
index 00000000000..7df3bfecf11
--- /dev/null
+++ b/tests/components/atag/conftest.py
@@ -0,0 +1,17 @@
+"""Provide common Atag fixtures."""
+import asyncio
+from unittest.mock import patch
+
+import pytest
+
+
+@pytest.fixture(autouse=True)
+async def mock_pyatag_sleep():
+    """Mock out pyatag sleeps."""
+    asyncio_sleep = asyncio.sleep
+
+    async def sleep(duration, loop=None):
+        await asyncio_sleep(0)
+
+    with patch("pyatag.gateway.asyncio.sleep", new=sleep):
+        yield