"""Test cases for pass through logic."""

from unittest import mock

import pytest

from ows_accounting import response
from ows_accounting.logic import passthrough


@pytest.mark.parametrize('period_id, expected_data', [
    ('232', {}),
    ('4242', {'test': 'eq_bonus'})
])
@mock.patch('ows_accounting.logic.passthrough.passthrough')
def test_get_eq_bonus(passthrough_models, period_id, expected_data):
    """Test get_eq_bonus function."""
    account_id = '4321'
    expected_result = response.Response(expected_data)
    passthrough_models.get_account_eq_bonus.return_value = expected_data
    result = passthrough.get_eq_bonus(account_id, period_id)
    assert result.status == expected_result.status
    assert result.message == expected_result.message


@pytest.mark.parametrize('period_id, expected_data', [
    ('232', {}),
    ('4242', {'test': 'eq_details'})
])
@mock.patch('ows_accounting.logic.passthrough.passthrough')
def test_get_eq_payment_details(passthrough_models, period_id, expected_data):
    """Test get_eq_payment_details function."""
    account_id = '4321'
    expected_result = response.Response(expected_data)
    passthrough_models.get_account_eq_payment_details.return_value = (
        expected_data)
    result = passthrough.get_eq_payment_details(account_id, period_id)
    assert result.status == expected_result.status
    assert result.message == expected_result.message