"""Test for ItunesLanguages model.""" from product.models import itunes_languages from tests.factories.itunes_languages import ItunesLanguagesFactory from tests.testutils import db @db.test_schema def test_get_language_by_id(): """Test get_language_by_id.""" expected_response = { 'language': 'Test Language', 'language_code': 'test', 'language_id': 1} db.insert_itunes_languages() response = itunes_languages.get_language_by_id(1) assert response assert response.message == expected_response @db.test_schema def test_get_language_by_id_not_found(): """Test get_language_by_id failure.""" response = itunes_languages.get_language_by_id(12) assert response.status == 404 assert response.errors['code'] == 'not_found_error' @db.test_schema def test_get_all_itunes_languages(): """Test get_all_itunes_languages success.""" db.seed_models(ItunesLanguagesFactory.build_batch(8)) response = itunes_languages.get_all_itunes_languages() assert response assert 'items' in response.message assert 'pagination' in response.message assert len(response.message['items']) == 8 @db.test_schema def test_get_all_itunes_languages_no_data(): """Test get_all_itunes_languages when no data exist.""" response = itunes_languages.get_all_itunes_languages() assert response.status == 200 assert len(response.message['items']) == 0