test_data_feeds.py

import csv
from io import StringIO
from unittest.mock import Mock

from django.utils.encoding import smart_text

from saleor.data_feeds.google_merchant import (
    get_feed_items,
    item_attributes,
    item_availability,
    item_google_product_category,
    write_feed,
)
from saleor.product.models import AttributeValue, Category


def test_saleor_feed_items(product, site_settings):
    valid_variant = product.variants.first()
    items = get_feed_items()
    assert len(items) == 1
    categories = Category.objects.all()
    discounts = []
    category_paths = {}
    attributes_dict = {}
    current_site = site_settings.site
    attribute_values_dict = {
        smart_text(a.pk): smart_text(a) for a in AttributeValue.objects.all()
    }
    attributes = item_attributes(
        items[0],
        categories,
        category_paths,
        current_site,
        discounts,
        attributes_dict,
        attribute_values_dict,
    )
    assert attributes.get("mpn") == valid_variant.sku
    assert attributes.get("availability") == "in stock"


def test_saleor_get_feed_items_having_no_stock_info(variant, site_settings):
    variant.stocks.all().delete()
    assert item_availability(variant) == "out of stock"


def test_category_formatter(db):
    main_category = Category(name="Main", slug="main")
    main_category.save()
    main_category_item = Mock(product=Mock(category=main_category))
    sub_category = Category(name="Sub", slug="sub", parent=main_category)
    sub_category.save()
    sub_category_item = Mock(product=Mock(category=sub_category))
    assert item_google_product_category(main_category_item, {}) == "Main"
    assert item_google_product_category(sub_category_item, {}) == "Main > Sub"


def test_write_feed(product, monkeypatch):
    buffer = StringIO()
    write_feed(buffer)
    buffer.seek(0)
    dialect = csv.Sniffer().sniff(buffer.getvalue())
    assert dialect.delimiter == csv.excel_tab.delimiter
    assert dialect.quotechar == csv.excel_tab.quotechar
    assert dialect.escapechar == csv.excel_tab.escapechar
    assert csv.Sniffer().has_header(buffer.getvalue())
    lines = [line for line in csv.reader(buffer, dialect=csv.excel_tab)]
    assert len(lines) == 2
    header = lines[0]
    google_required_fields = [
        "id",
        "title",
        "image_link",
        "availability",
        "price",
        "condition",
    ]
    for field in google_required_fields:
        assert field in header