"""Test release_correction logic."""

import pytest

from src import service
from src.logic import release_correction


def test_get_release_correction_success(request_engine):
    """Test apply correction success."""
    request_engine[service.OWS_PRODUCT_WORKFLOW].add_spec(
        'GET', '/correction/1', response={'foo': 'bar'})
    actual = release_correction.get_release_correction(1)
    assert actual == {'foo': 'bar'}


def test_get_release_correction_throws(request_engine):
    """Test get release correction fail."""
    request_engine[service.OWS_PRODUCT_WORKFLOW].add_spec(
        'GET', '/correction/1',
        response={'foo': 'bar'},  status=400)
    with pytest.raises(Exception) as e:
        release_correction.get_release_correction(1)
    assert e.value.args[0] == 'Failed get release correction for id 1\nStatus code: 400\nError: {"foo": "bar"}\n' # noqa


def test_get_asset_revisions_success(request_engine):
    """Test invalidate assets success."""
    request_engine[service.OWS_PRODUCT_WORKFLOW].add_spec(
        'GET', '/correction/1', response={
            'items': [
                {
                    'table_name': 'track',
                    'field_name': 'track',
                    'key_id': 1,
                    'key_value': 'abc'
                },
                {
                    'table_name': 'releases',
                    'field_name': 'coverart',
                    'key_id': 2,
                    'key_value': 'abc'
                }
            ]
        })
    actual = release_correction.get_asset_revisions(1)
    assert actual == [
        {
            'table_name': 'track',
            'field_name': 'track',
            'key_id': 1,
            'key_value': 'abc'
        },
        {
            'table_name': 'releases',
            'field_name': 'coverart',
            'key_id': 2,
            'key_value': 'abc'
        }
    ]