"""Mocks for fact_conflict."""

from oto.response import Response

from conflict_manager.models import fact_conflict
from tests.testutils import wrap_api_results


def get_new_youtube_conflicts_for_account(
        mocker, data, total_records=0, offset=0, limit=50):
    """Mock get_new_youtube_conflicts_for_account function.

    Args:
        data (Response object or list of dicts)
    """
    if isinstance(data, Response):
        response = data
    else:
        response = Response(wrap_api_results(
            data, total_records=total_records, offset=offset,
            limit=limit))

    return mocker.patch.object(
        fact_conflict, 'get_new_youtube_conflicts_for_account',
        return_value=response)


def get_actioned_youtube_conflicts_for_account(
        mocker, data, total_records=0, offset=0, limit=50):
    """Mock get_actioned_youtube_conflicts_for_account function.

    Args:
        data (Response object or list of dicts)
    """
    if isinstance(data, Response):
        response = data
    else:
        response = Response(wrap_api_results(
            data, total_records=total_records, offset=offset, limit=limit))

    return mocker.patch.object(
        fact_conflict, 'get_actioned_youtube_conflicts_for_account',
        return_value=response)


def get_resolved_youtube_conflicts_for_account(
        mocker, data, total_records=0, offset=0, limit=50):
    """Mock get_resolved_youtube_conflicts_for_account function.

    Args:
        data (Response object or list of dicts)
    """
    if isinstance(data, Response):
        response = data
    else:
        response = Response(wrap_api_results(
            data, total_records=total_records, offset=offset, limit=limit))

    return mocker.patch.object(
        fact_conflict, 'get_resolved_youtube_conflicts_for_account',
        return_value=response)