45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
"""Test configuration for PS4."""
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from pyps4_2ndscreen.ddp import DEFAULT_UDP_PORT, DDPProtocol
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def patch_load_json():
|
|
"""Prevent load JSON being used."""
|
|
with patch("homeassistant.components.ps4.load_json", return_value={}) as mock_load:
|
|
yield mock_load
|
|
|
|
|
|
@pytest.fixture
|
|
def patch_save_json():
|
|
"""Prevent save JSON being used."""
|
|
with patch("homeassistant.components.ps4.save_json") as mock_save:
|
|
yield mock_save
|
|
|
|
|
|
@pytest.fixture
|
|
def patch_get_status():
|
|
"""Prevent save JSON being used."""
|
|
with patch("pyps4_2ndscreen.ps4.get_status", return_value=None) as mock_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, mock_ddp_endpoint):
|
|
"""Prevent PS4 doing I/O."""
|