# 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 datetime as dt
import io
import unittest
import urllib.request

import PIL.Image

from trytond.pool import Pool
from trytond.tests.test_tryton import (
    TEST_NETWORK, ModuleTestCase, with_transaction)


class ProductImageTestCase(ModuleTestCase):
    'Test Product Image module'
    module = 'product_image'

    @unittest.skipUnless(TEST_NETWORK, "requires network")
    @with_transaction()
    def test_image_square(self):
        "Test image square"
        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://downloads.tryton.org/tests/product-image/chair.jpg'
            ).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))

        img = PIL.Image.open(io.BytesIO(image.get(size=(150, 200))))
        self.assertEqual(img.size, (150, 150))

    @unittest.skipUnless(TEST_NETWORK, "requires network")
    @with_transaction()
    def test_image_non_square(self):
        "Test image non square"
        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://downloads.tryton.org/tests/product-image/chair-800.jpg'
            ).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, (78, 100))

        img = PIL.Image.open(io.BytesIO(image.get(size=(100, 200))))
        self.assertEqual(img.size, (100, 128))

    @with_transaction()
    def test_round_size(self):
        "Test round size"
        pool = Pool()
        Image = pool.get('product.image')
        for size, result in [
                (1, 1),
                (2, 2),
                (19, 20),
                (45, 50),
                (101, 128),
                (129, 150),
                ]:
            with self.subTest(size=size):
                self.assertEqual(Image._round_size(size), result)

    @unittest.skipUnless(TEST_NETWORK, "requires network")
    @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://downloads.tryton.org/tests/product-image/chair.jpg'
                ).read()
            image.save()

        template.write_date = dt.datetime(2025, 1, 1)  # fix timestamp

        self.assertRegex(
            template.get_image_url(i=2),
            r'^/product/image/CODE/.*/Template\?i=2&t=AAAAAGd0hYA$')
        self.assertRegex(
            template.get_image_url(s=400),
            r'^/product/image/CODE/.*/Template\?s=400&t=AAAAAGd0hYA$')


del ModuleTestCase