product.py 595 Bytes
from floraconcierge.mapping.mixins.model.shop import DEFAULT_PRODUCT_IMAGES_LIST
from floraconcierge.mapping.model.shop import Product
from tests.client import TestClient


class TestProduct(TestClient):
    def test_product_urls(self):
        # roses bouquet
        product_id = 902

        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.iteritems():
            self.assertIn(k, DEFAULT_PRODUCT_IMAGES_LIST)