Python AsyncMock.async_get_event_summary示例

示例#1

def create_mock_frigate_client() -> AsyncMock:
    """Create mock frigate client."""
    mock_client = AsyncMock()
    mock_client.async_get_stats = AsyncMock(return_value=TEST_STATS)
    mock_client.async_get_config = AsyncMock(return_value=TEST_CONFIG)
    mock_client.async_get_event_summary = AsyncMock(
        return_value=TEST_EVENT_SUMMARY)
    mock_client.async_get_version = AsyncMock(return_value=TEST_SERVER_VERSION)
    return mock_client

示例#2

async def test_async_browse_media_async_get_event_summary_error(
        caplog: Any, frigate_client: AsyncMock, hass: HomeAssistant) -> None:
    """Test API error behavior."""
    frigate_client.async_get_event_summary = AsyncMock(
        side_effect=FrigateApiClientError)

    await setup_mock_frigate_config_entry(hass, client=frigate_client)

    with pytest.raises(MediaSourceError):
        await media_source.async_browse_media(
            hass,
            f"{const.URI_SCHEME}{DOMAIN}/{TEST_FRIGATE_INSTANCE_ID}"
            "/event-search/clips",
        )