test_translation.py

import pytest

from saleor.product.models import (
    AttributeTranslation,
    AttributeValueTranslation,
    CategoryTranslation,
    CollectionTranslation,
    ProductTranslation,
    ProductVariantTranslation,
)
from saleor.shipping.models import ShippingMethodTranslation


@pytest.fixture
def product_translation_pl(product):
    return ProductTranslation.objects.create(
        language_code="pl",
        product=product,
        name="Polish name",
        description="Polish description",
    )


@pytest.fixture
def attribute_value_translation_fr(translated_attribute):
    value = translated_attribute.attribute.values.first()
    return AttributeValueTranslation.objects.create(
        language_code="fr", attribute_value=value, name="French name"
    )


@pytest.fixture
def shipping_method_translation_fr(shipping_method):
    return ShippingMethodTranslation.objects.create(
        language_code="fr", shipping_method=shipping_method, name="French name"
    )


def test_translation(product, settings, product_translation_fr):
    assert product.translated.name == "Test product"
    assert not product.translated.description

    settings.LANGUAGE_CODE = "fr"
    assert product.translated.name == "French name"
    assert product.translated.description == "French description"


def test_translation_str_returns_str_of_instance(
    product, product_translation_fr, settings
):
    assert str(product.translated) == str(product)
    settings.LANGUAGE_CODE = "fr"
    assert str(product.translated.translation) == str(product_translation_fr)


def test_wrapper_gets_proper_wrapper(
    product, product_translation_fr, settings, product_translation_pl
):
    assert product.translated.translation is None

    settings.LANGUAGE_CODE = "fr"
    assert product.translated.translation == product_translation_fr

    settings.LANGUAGE_CODE = "pl"
    assert product.translated.translation == product_translation_pl


def test_getattr(product, settings, product_translation_fr, product_type):
    settings.LANGUAGE_CODE = "fr"
    assert product.translated.product_type == product_type


def test_translation_not_override_id(settings, product, product_translation_fr):
    settings.LANGUAGE_CODE = "fr"
    translated_product = product.translated
    assert translated_product.id == product.id
    assert not translated_product.id == product_translation_fr


def test_collection_translation(settings, collection):
    settings.LANGUAGE_CODE = "fr"
    french_name = "French name"
    CollectionTranslation.objects.create(
        language_code="fr", name=french_name, collection=collection
    )
    assert collection.translated.name == french_name


def test_category_translation(settings, category):
    settings.LANGUAGE_CODE = "fr"
    french_name = "French name"
    french_description = "French description"
    CategoryTranslation.objects.create(
        language_code="fr",
        name=french_name,
        description=french_description,
        category=category,
    )
    assert category.translated.name == french_name
    assert category.translated.description == french_description


def test_product_variant_translation(settings, variant):
    settings.LANGUAGE_CODE = "fr"
    french_name = "French name"
    ProductVariantTranslation.objects.create(
        language_code="fr", name=french_name, product_variant=variant
    )
    assert variant.translated.name == french_name


def test_attribute_translation(settings, color_attribute):
    AttributeTranslation.objects.create(
        language_code="fr", attribute=color_attribute, name="French name"
    )
    assert not color_attribute.translated.name == "French name"
    settings.LANGUAGE_CODE = "fr"
    assert color_attribute.translated.name == "French name"


def test_attribute_value_translation(settings, product, attribute_value_translation_fr):
    attribute = product.product_type.product_attributes.first().values.first()
    assert not attribute.translated.name == "French name"
    settings.LANGUAGE_CODE = "fr"
    assert attribute.translated.name == "French name"


def test_voucher_translation(settings, voucher, voucher_translation_fr):
    assert not voucher.translated.name == "French name"
    settings.LANGUAGE_CODE = "fr"
    assert voucher.translated.name == "French name"


def shipping_method_translation(
    settings, shipping_method, shipping_method_translation_fr
):
    assert not shipping_method.translated.name == "French name"
    settings.LANGUAGE_CODE = "fr"
    assert shipping_method.translated.name == "French name"