"""Test for the logic of owner."""

from unittest.mock import MagicMock

import pytest
from owsresponse import response

from account.logic import owner


@pytest.fixture
def fixture_owners():
    """Fixture response for list of owners."""
    return response.Response(
        [
            {'owner_abbrivation': '1091 FT', 'owner_name': '1091 Film Transfer'},
            {
                'owner_abbrivation': '1091 FT-H',
                'owner_name': '1091 Film Transfer Hybrid',
            },
            {
                'owner_abbrivation': 'HIFI-AWAL',
                'owner_name': '3rd Part Agmts acquired. HiFI -AWAL US',
            },
        ]
    )


def test_get_owners(monkeypatch, fixture_owners):
    """Test get owners success."""
    expected = fixture_owners.message
    monkeypatch.setattr(owner, 'get_owners', MagicMock(return_value=fixture_owners))
    result = owner.get_owners()

    assert result
    assert result.message == expected