Refactor Apple TV integration (#31952)

Co-authored-by: Franck Nijhof <git@frenck.dev>

tests/components/apple_tv/conftest.py

@@ -0,0 +1,131 @@
+"""Fixtures for component."""
+
+from pyatv import conf, net
+import pytest
+
+from .common import MockPairingHandler, create_conf
+
+from tests.async_mock import patch
+
+
+@pytest.fixture(autouse=True, name="mock_scan")
+def mock_scan_fixture():
+    """Mock pyatv.scan."""
+    with patch("homeassistant.components.apple_tv.config_flow.scan") as mock_scan:
+
+        async def _scan(loop, timeout=5, identifier=None, protocol=None, hosts=None):
+            if not mock_scan.hosts:
+                mock_scan.hosts = hosts
+            return mock_scan.result
+
+        mock_scan.result = []
+        mock_scan.hosts = None
+        mock_scan.side_effect = _scan
+        yield mock_scan
+
+
+@pytest.fixture(name="dmap_pin")
+def dmap_pin_fixture():
+    """Mock pyatv.scan."""
+    with patch("homeassistant.components.apple_tv.config_flow.randrange") as mock_pin:
+        mock_pin.side_effect = lambda start, stop: 1111
+        yield mock_pin
+
+
+@pytest.fixture
+def pairing():
+    """Mock pyatv.scan."""
+    with patch("homeassistant.components.apple_tv.config_flow.pair") as mock_pair:
+
+        async def _pair(config, protocol, loop, session=None, **kwargs):
+            handler = MockPairingHandler(
+                await net.create_session(session), config.get_service(protocol)
+            )
+            handler.always_fail = mock_pair.always_fail
+            return handler
+
+        mock_pair.always_fail = False
+        mock_pair.side_effect = _pair
+        yield mock_pair
+
+
+@pytest.fixture
+def pairing_mock():
+    """Mock pyatv.scan."""
+    with patch("homeassistant.components.apple_tv.config_flow.pair") as mock_pair:
+
+        async def _pair(config, protocol, loop, session=None, **kwargs):
+            return mock_pair
+
+        async def _begin():
+            pass
+
+        async def _close():
+            pass
+
+        mock_pair.close.side_effect = _close
+        mock_pair.begin.side_effect = _begin
+        mock_pair.pin = lambda pin: None
+        mock_pair.side_effect = _pair
+        yield mock_pair
+
+
+@pytest.fixture
+def full_device(mock_scan, dmap_pin):
+    """Mock pyatv.scan."""
+    mock_scan.result.append(
+        create_conf(
+            "127.0.0.1",
+            "MRP Device",
+            conf.MrpService("mrpid", 5555),
+            conf.DmapService("dmapid", None, port=6666),
+            conf.AirPlayService("airplayid", port=7777),
+        )
+    )
+    yield mock_scan
+
+
+@pytest.fixture
+def mrp_device(mock_scan):
+    """Mock pyatv.scan."""
+    mock_scan.result.append(
+        create_conf("127.0.0.1", "MRP Device", conf.MrpService("mrpid", 5555))
+    )
+    yield mock_scan
+
+
+@pytest.fixture
+def dmap_device(mock_scan):
+    """Mock pyatv.scan."""
+    mock_scan.result.append(
+        create_conf(
+            "127.0.0.1",
+            "DMAP Device",
+            conf.DmapService("dmapid", None, port=6666),
+        )
+    )
+    yield mock_scan
+
+
+@pytest.fixture
+def dmap_device_with_credentials(mock_scan):
+    """Mock pyatv.scan."""
+    mock_scan.result.append(
+        create_conf(
+            "127.0.0.1",
+            "DMAP Device",
+            conf.DmapService("dmapid", "dummy_creds", port=6666),
+        )
+    )
+    yield mock_scan
+
+
+@pytest.fixture
+def airplay_device(mock_scan):
+    """Mock pyatv.scan."""
+    mock_scan.result.append(
+        create_conf(
+            "127.0.0.1", "AirPlay Device", conf.AirPlayService("airplayid", port=7777)
+        )
+    )
+    yield mock_scan