From 55a77b2ba26f2e4ebd1a7d4fae904e50c6a943d4 Mon Sep 17 00:00:00 2001 From: Erik Montnemery Date: Fri, 17 Sep 2021 12:51:25 +0200 Subject: [PATCH] Prevent 3rd party lib from opening sockets in ps4 tests (#56330) --- tests/components/ps4/conftest.py | 18 ++++++++++++++++-- 1 file changed, 16 insertions(+), 2 deletions(-) diff --git a/tests/components/ps4/conftest.py b/tests/components/ps4/conftest.py index 155f1c6d5dd..5bb27012b18 100644 --- a/tests/components/ps4/conftest.py +++ b/tests/components/ps4/conftest.py @@ -1,6 +1,7 @@ """Test configuration for PS4.""" -from unittest.mock import patch +from unittest.mock import MagicMock, patch +from pyps4_2ndscreen.ddp import DEFAULT_UDP_PORT, DDPProtocol import pytest @@ -25,6 +26,19 @@ def patch_get_status(): yield mock_get_status +@pytest.fixture +def mock_ddp_endpoint(): + """Mock pyps4_2ndscreen.ddp.async_create_ddp_endpoint.""" + protocol = DDPProtocol() + protocol._local_port = DEFAULT_UDP_PORT + protocol._transport = MagicMock() + with patch( + "homeassistant.components.ps4.async_create_ddp_endpoint", + return_value=(None, protocol), + ): + yield + + @pytest.fixture(autouse=True) -def patch_io(patch_load_json, patch_save_json, patch_get_status): +def patch_io(patch_load_json, patch_save_json, patch_get_status, mock_ddp_endpoint): """Prevent PS4 doing I/O."""