"""Tests for country model."""

import pytest

from account.models import country
from tests.unit import db_operations


@pytest.fixture
def db_fixture():
    """Set up the country table."""
    db_operations.create_tables()
    db_operations.seed_country_table()


def test_get_countries(db_fixture):
    expected = [
        {'id': 200, 'name': 'Aruba'},
        {'id': 28, 'name': 'Bangladesh'},
        {'id': 6, 'name': 'France'},
    ]
    result = country.get_countries()
    assert result
    assert result.message == expected


def test_get_country_id_by_code_found(db_fixture):
    result = country.get_country_id_by_code('FRA')
    assert result == 6


def test_get_country_id_by_code_case_insensitive(db_fixture):
    result = country.get_country_id_by_code('fra')
    assert result == 6


def test_get_country_id_by_code_not_found(db_fixture):
    result = country.get_country_id_by_code('ZZZ')
    assert result is None