"""Tests for budget_caps logic tier."""


from unittest.mock import MagicMock


from ows_accounting import response
from ows_accounting.logic import budget_caps
from ows_accounting.models import fact_sales as fact_sales_model


def test_get_budget_caps(monkeypatch):
    """Test getting the average monthly net revenue."""
    payload = {
        'items': [
            {
                'account_type': 'vendor',
                'account_id': 123,
                'monthly_total_budget': 2500,
                'budget_type': 'white_list'
            },
            {
                'account_type': 'vendor',
                'account_id': 124,
                'monthly_total_budget': None,
                'budget_type': 'general'}]}

    monkeypatch.setattr(
        fact_sales_model, 'get_budget_caps', MagicMock(
            return_value=response.Response(payload)))

    result = budget_caps.get_budget_caps()

    fact_sales_model.get_budget_caps.assert_called_once()
    assert result
    assert result.message == payload