product.py 739 Bytes

Edit Raw Blame History

from floraconcierge.mapping.mixins.model.shop import DEFAULT_PRODUCT_IMAGES_LIST
from floraconcierge.mapping.model.shop.calculate import Product
from tests.client import TestClient

# roses bouquet
product_id = 902


class TestProduct(TestClient):
    def test_product_urls(self):
        product = self.client.call('shop/product', {'id': product_id})
        self.assertIsInstance(product, Product)

        self.assertTrue(product.urls, 'Not found urls in product #%s' % product_id)
        for k, v in product.urls.items():
            self.assertIn(k, DEFAULT_PRODUCT_IMAGES_LIST)

    def test_product_reviews(self):
        reviews = self.client.call('shop/reviews', {'product_id': product_id})
        self.assertIsNotNone(reviews)