"""Mocks for fact_conflict."""

from oto.response import Response

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


def get_new_conflicts_for_account(
        mocker, data, total_records=0, offset=0, limit=50):
    """Mock get_new_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_elasticsearch, 'get_new_conflicts_for_account',
        return_value=response)


def get_new_conflicts_count_for_account(mocker, data):
    """Mock get_new_conflicts_for_account function.

    Args:
        data (Response object or list of dicts)
    """
    if isinstance(data, Response):
        response = data
    else:
        response = Response(data, status=200)

    return mocker.patch.object(
        fact_conflict_elasticsearch, 'get_new_conflict_count_for_account',
        return_value=response)