Example #1
0
def mock_schedule_object() -> Generator[None, None, SwitcherV2Schedule]:
    """Fixture for the aioswitcher.schedules.SwitcherV2Schedule object.

    Returns:
      Mocked ``SwitcherV2Schedule`` object.

    """
    mock_object = MagicMock(SwitcherV2Schedule)
    mock_object.schedule_id = consts.DUMMY_SCHEDULE_ID
    mock_object.enabled = True
    mock_object.recurring = True
    mock_object.days = [WEEKDAY_TUP[get_next_weekday()]]
    mock_object.start_time = consts.DUMMY_START_TIME
    mock_object.end_time = consts.DUMMY_END_TIME
    mock_object.duration = consts.DUMMY_DURATION

    mock_initial_response = MagicMock(SwitcherV2Schedule)
    mock_initial_response.init_future = Future()
    mock_initial_response.init_future.set_result(mock_object)

    return mock_initial_response