"""Mocks for ows_product_digital."""

from oto.response import Response

from tests.testutils import constants as test_consts


def ows_product_digital_get_product_by_id(
        mocker, corrections=test_consts.TEST_PRODUCT_CORRECTIONS,
        product_id=test_consts.TEST_PRODUCT_ID, upc=test_consts.TEST_NEW_UPC,
        genre_id=test_consts.TEST_GENRE_ID,
        subgenre_id=test_consts.TEST_SUBGENRE_ID,
        product_name=test_consts.TEST_PRODUCT_NAME ,
        release_status=test_consts.TEST_RELEASE_STATUS,
        is_ec_ar=test_consts.TEST_EC_AR_MODE,
        product_artists=test_consts.TEST_PRODUCT_ARTISTS,
        meta_language='ENG',
        delivered_version='',
        preorder_date='2015-01-01',
        sale_start_date='2016-01-01'):
    """Mock OWS Product get_product_by_product_id."""
    response_data = {
        'corrections': corrections,
        'product_id': product_id, 'upc': upc,
        'genre_id': genre_id, 'subgenre_id': subgenre_id,
        'product_name': product_name,
        'release_status': release_status,
        'is_error_correction_and_action_required': is_ec_ar,
        'product_artists': product_artists,
        'meta_language': meta_language,
        'delivered_version': delivered_version,
        'preorder_date': preorder_date,
        'sale_start_date': sale_start_date,
        'release_date': '2030-01-01'}

    res = Response(message=response_data)
    mocker.patch(
        'backend.models.ows_product_digital.get_product_by_product_id',
        return_value=res)


def ows_product_digital_get_product_spatial(
        mocker, product_id=test_consts.TEST_PRODUCT_ID, found=True):
    """Mock ows_product_digital.get_product_spatial."""
    if found:
        res = Response(message={'product_id': product_id, 'spatial_upc': '123456789012'})
    else:
        from oto import response as oto_response
        res = oto_response.create_not_found_response()
    mocker.patch(
        'backend.models.ows_product_digital.get_product_spatial',
        return_value=res)
    return res


def ows_product_digital_create_product_spatial(
        mocker, success=True):
    """Mock ows_product_digital.create_product_spatial."""
    res = Response() if success else Response(status=500)
    mocker.patch(
        'backend.models.ows_product_digital.create_product_spatial',
        return_value=res)
    return res


def ows_product_digital_get_chinese_product_by_id(
        mocker, corrections=test_consts.TEST_PRODUCT_CORRECTIONS,
        product_id=test_consts.TEST_PRODUCT_ID, upc=test_consts.TEST_NEW_UPC,
        genre_id=test_consts.TEST_GENRE_ID,
        subgenre_id=test_consts.TEST_SUBGENRE_ID,
        release_status=test_consts.TEST_RELEASE_STATUS,
        is_ec_ar=test_consts.TEST_EC_AR_MODE):
    """Mock OWS Product get_product_by_product_id."""
    response_data = {
        'corrections': corrections,
        'product_id': product_id, 'upc': upc,
        'genre_id': genre_id, 'subgenre_id': subgenre_id,
        'release_status': release_status,
        'is_error_correction_and_action_required': is_ec_ar,
        'meta_language': 'CHI'}

    res = Response(message=response_data)
    mocker.patch(
        'backend.models.ows_product_digital.get_product_by_product_id',
        return_value=res)