first commit

modules/product_image_attribute/tests/__init__.py

@@ -0,0 +1,2 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.

modules/product_image_attribute/tests/test_module.py

@@ -0,0 +1,76 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+import unittest
+import urllib.request
+
+from trytond.pool import Pool
+from trytond.tests.test_tryton import (
+    TEST_NETWORK, ModuleTestCase, with_transaction)
+
+
+class ProductImageAttributeTestCase(ModuleTestCase):
+    'Test Product Image Attribute module'
+    module = 'product_image_attribute'
+
+    @unittest.skipUnless(TEST_NETWORK, "requires network")
+    @with_transaction()
+    def test_image_attribute(self):
+        "Test image with attribute"
+        pool = Pool()
+        Attribute = pool.get('product.attribute')
+        AttributeSet = pool.get('product.attribute.set')
+        Image = pool.get('product.image')
+        Product = pool.get('product.product')
+        Template = pool.get('product.template')
+        Uom = pool.get('product.uom')
+
+        attribute1 = Attribute(name='attr1', string="Attribute 1")
+        attribute1.type_ = 'char'
+        attribute1.save()
+        attribute2 = Attribute(name='attr2', string="Attribute 2")
+        attribute2.type_ = 'integer'
+        attribute2.save()
+        attribute_set = AttributeSet(name="Attribute Set")
+        attribute_set.attributes = [attribute1, attribute2]
+        attribute_set.save()
+        template = Template(name="Template")
+        template.default_uom, = Uom.search([], limit=1)
+        template.attribute_set = attribute_set
+        template.save()
+        product = Product(template=template)
+        product.attributes = {
+            'attr1': 'foo',
+            'attr2': 2,
+            }
+        product.save()
+        image1 = Image(template=template)
+        image1.image = urllib.request.urlopen(
+            'https://downloads.tryton.org/tests/product-image/chair.jpg'
+            ).read()
+        image1.save()
+        image2 = Image(template=template, product=product)
+        image2.image = urllib.request.urlopen(
+            'https://downloads.tryton.org/tests/product-image/chair.jpg'
+            ).read()
+        image2.save()
+        image3 = Image(template=template)
+        image3.image = urllib.request.urlopen(
+            'https://downloads.tryton.org/tests/product-image/chair.jpg'
+            ).read()
+        image3.attributes = {
+            'attr1': 'foo',
+            }
+        image3.save()
+        image4 = Image(template=template)
+        image4.image = urllib.request.urlopen(
+            'https://downloads.tryton.org/tests/product-image/chair.jpg'
+            ).read()
+        image4.attributes = {
+            'attr1': 'bar',
+            }
+        image4.save()
+
+        self.assertEqual(list(product.images_used), [image2, image1, image3])
+
+
+del ModuleTestCase