1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
# 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 io
import urllib.request
import PIL.Image
from trytond.pool import Pool
from trytond.tests.test_tryton import ModuleTestCase, with_transaction
class ProductImageTestCase(ModuleTestCase):
'Test Product Image module'
module = 'product_image'
@with_transaction()
def test_image(self):
"Test image"
pool = Pool()
Image = pool.get('product.image')
Template = pool.get('product.template')
Uom = pool.get('product.uom')
template = Template(name="Template")
template.default_uom, = Uom.search([], limit=1)
template.save()
image = Image(template=template)
image.image = urllib.request.urlopen(
'https://picsum.photos/200').read()
image.save()
self.assertEqual(template.image_url, None)
template.code = "CODE"
template.save()
self.assertRegex(
template.image_url,
r'/product/image/CODE/.*/Template\?s=64')
img = PIL.Image.open(io.BytesIO(image.get(size=100)))
self.assertEqual(img.size, (100, 100))
@with_transaction()
def test_get_image_url(self):
"Test get_image_url"
pool = Pool()
Image = pool.get('product.image')
Template = pool.get('product.template')
Uom = pool.get('product.uom')
template = Template(name="Template", code="CODE")
template.default_uom, = Uom.search([], limit=1)
template.save()
for _ in range(3):
image = Image(template=template)
image.image = urllib.request.urlopen(
'https://picsum.photos/200').read()
image.save()
self.assertRegex(
template.get_image_url(i=2),
r'/product/image/CODE/.*/Template\?i=2')
self.assertRegex(
template.get_image_url(s=400),
r'/product/image/CODE/.*/Template\?s=400')
del ModuleTestCase
|
