"""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."""