"""Test for the logic of country."""

from unittest.mock import MagicMock

from owsresponse import response

from account.models import country


def test_get_countries(monkeypatch):
    """Test get countries."""
    expected = [{'id': 200, 'name': 'Aruba'}]
    monkeypatch.setattr(
        country, 'get_countries', MagicMock(return_value=response.Response(expected))
    )
    result = country.get_countries()
    assert result
    assert result.status == 200
    assert result.message == expected