"""Tests for Pricing Family logic."""

from unittest.mock import MagicMock

from oto import response

from pricing.logic import pricing_family
from pricing.models import pricing_family as pricing_family_model

import pytest


@pytest.fixture
def fixture_families():
    """Fixture for pricing families list."""
    return response.Response({'items': ['a pricing family']})


@pytest.fixture
def fixture_pricing_family():
    """Fixture for pricing families list."""
    return response.Response({'name': 'A Pricing Family'})


def test_get_pricing_families(monkeypatch, fixture_families):
    """Test getting all pricing families."""
    monkeypatch.setattr(
        pricing_family_model, 'get_pricing_families', MagicMock(
            return_value=fixture_families))

    result = pricing_family.get_pricing_families()
    assert result
    assert result.message == fixture_families.message


def test_create_pricing_family(
        monkeypatch, fixture_pricing_family):
    """Test creating a Pricing Family."""
    monkeypatch.setattr(
        pricing_family_model,
        'create_pricing_family',
        MagicMock(return_value=fixture_pricing_family))

    data = {'name': 'A Pricing Family'}
    result = pricing_family.create_pricing_family(data)
    assert result
    assert result.message == fixture_pricing_family.message