"""Tests for subaccount_royalty_collection model."""

from contracts.models import subaccount_royalty_collection


def test_get_active_subaccount_royalty_collection(db_fixture, monkeypatch):
    """Test active subaccount royalty collection."""
    subaccount_id = 1
    expected_response = {
        'subaccount_royalty_collection_id': 2,
        'subaccount_id': subaccount_id,
        'active': True,
    }
    response = subaccount_royalty_collection.get_active_subaccount_royalty_collection(
        subaccount_id
    )
    assert response.message == expected_response


def test_get_subaccount_royalty_collection_territories(db_fixture, monkeypatch):
    """Test active subaccount royalty collection."""
    subaccount_royalty_collection_id = 1
    expected_response = [1]

    response = (
        subaccount_royalty_collection.get_subaccount_royalty_collection_territories(
            subaccount_royalty_collection_id
        )
    )
    assert response.message == expected_response


def test_get_subaccount_id_by_royalty_collection_id(db_fixture, monkeypatch):
    """Test get subaccount id by royalty collection id."""
    subaccount_royalty_collection_id = 1
    expected_response = 1

    response = subaccount_royalty_collection.get_subaccount_id_by_royalty_collection_id(
        subaccount_royalty_collection_id
    )
    assert response.message == expected_response


def test_get_royalty_collection(db_fixture, monkeypatch):
    """Test get subaccount id by royalty collection id."""
    subaccount_royalty_collection_id = 1
    expected_response = {
        'subaccount_royalty_collection_id': 1,
        'subaccount_id': 1,
        'active': False,
        'territories': [1],
    }

    response = subaccount_royalty_collection.get_royalty_collection(
        subaccount_royalty_collection_id
    )
    assert response.message == expected_response