"""Tests for Orchard Pricing Tier logic."""

from unittest.mock import MagicMock

from oto import response

from pricing.logic import orchard_pricing_tier
from pricing.models import orchard_pricing_tier as orchard_pricing_tier_model

import pytest


@pytest.fixture
def fixture_orchard_pricing_tier():
    """Fixture for pricing families list."""
    return response.Response({'name': 'An Orchard Pricing Tier'})


def test_get_orchard_pricing_tier_by_pricing_family_id(
        monkeypatch, fixture_orchard_pricing_tier):
    """Test getting an orchard pricing tier by ID."""
    monkeypatch.setattr(
        orchard_pricing_tier_model,
        'get_orchard_pricing_tier_by_pricing_family_id',
        MagicMock(return_value=fixture_orchard_pricing_tier))

    result = orchard_pricing_tier\
        .get_orchard_pricing_tier_by_pricing_family_id(1)
    assert result
    assert result.message == fixture_orchard_pricing_tier.message


def test_create_orchard_pricing_tier(
        monkeypatch, fixture_orchard_pricing_tier):
    """Test creating an orchard pricing tier."""
    monkeypatch.setattr(
        orchard_pricing_tier_model,
        'create_orchard_pricing_tier',
        MagicMock(return_value=fixture_orchard_pricing_tier))

    data = {'name': 'An Orchard Pricing Tier'}
    result = orchard_pricing_tier.create_orchard_pricing_tier(data)
    assert result
    assert result.message == fixture_orchard_pricing_tier.message


def test_update_orchard_pricing_tier_by_id(
        monkeypatch, fixture_orchard_pricing_tier):
    """Test updating an orchard pricing tier by ID."""
    monkeypatch.setattr(
        orchard_pricing_tier_model,
        'update_orchard_pricing_tier_by_id',
        MagicMock(return_value=fixture_orchard_pricing_tier))

    data = {'name': 'An Orchard Pricing Tier'}
    result = orchard_pricing_tier.update_orchard_pricing_tier_by_id(1, data)
    assert result
    assert result.message == fixture_orchard_pricing_tier.message