"""Unit tests for Language."""

from product.models import language
from tests.factories import language as lang_factory
from tests.testutils import db


def test_to_dict():
    """Test that the expected properties are returned with correct values."""
    lang = language.Language()
    lang.language_code = 'SUX'
    lang.language = 'Sumerian'
    expected_dict = {
        'code': lang.language_code,
        'iso_code': '',
        'name': lang.language}

    assert lang.to_dict() == expected_dict


@db.test_schema
def test_get_languages_items():
    """Test that the result includes the expected items."""
    lang_count = 3
    languages = lang_factory.LanguageFactory.build_batch(lang_count)
    db.seed_models(languages)
    result = language.get_languages()
    dictified_languages = [lang.to_dict() for lang in languages]

    assert result.message.get('items') == dictified_languages