"""Mocks for ows_assets."""

import time

from oto.response import Response

from backend.models import ows_assets
from tests.testutils.functions import wrap_api_results


def ows_assets_copy_assets(mocker, sleep_seconds=0, response=None):
    """Mock ows_assets copy_assets response."""
    if response is None:
        response = Response({'status': 'ok'})

    def _side_effect(*args):
        time.sleep(sleep_seconds)
        return response

    mocker.patch(
        'backend.models.ows_assets.copy_assets', side_effect=_side_effect)


def ows_assets_get_product_assets_v1(mocker, data_list=None):
    """Mock ows_assets get_product_assets response."""
    data_list = data_list if data_list else []
    mocker.patch.object(
        ows_assets,
        'get_product_assets_v1',
        return_value=Response(wrap_api_results(data_list)))


def ows_assets_get_product_assets_v2(mocker, data_list=None):
    """Mock ows_assets get_product_assets response."""
    data_list = data_list if data_list else []
    mocker.patch.object(
        ows_assets,
        'get_product_assets_v2',
        return_value=Response(wrap_api_results(data_list)))


def ows_assets_get_match_audio_results(mocker, data=None):
    """Mock ows_assets get_match_audio_results response."""
    mocker.patch.object(
        ows_assets,
        'get_match_audio_results',
        return_value=data)