229 lines
7.8 KiB
Python
229 lines
7.8 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# This file is part of the cashbook-module from m-ds for Tryton.
|
||
|
|
# The COPYRIGHT file at the top level of this repository contains the
|
||
|
|
# full copyright notices and license terms.
|
||
|
|
|
||
|
|
from io import BytesIO
|
||
|
|
from PIL import Image
|
||
|
|
from trytond.tests.test_tryton import ModuleTestCase, with_transaction
|
||
|
|
from trytond.pool import Pool
|
||
|
|
from trytond.transaction import Transaction
|
||
|
|
from trytond.exceptions import UserError
|
||
|
|
from trytond.modules.cashbook.tests import CashbookTestCase
|
||
|
|
from datetime import date
|
||
|
|
from decimal import Decimal
|
||
|
|
from .img_data import img_data_png, dok_data_pdf, text_data
|
||
|
|
|
||
|
|
|
||
|
|
class LineTestCase(CashbookTestCase):
|
||
|
|
'Test cashbook line module'
|
||
|
|
module = 'cashbook_media'
|
||
|
|
|
||
|
|
@with_transaction()
|
||
|
|
def test_media_add_image(self):
|
||
|
|
""" create cook/line, add png-file
|
||
|
|
"""
|
||
|
|
pool = Pool()
|
||
|
|
Book = pool.get('cashbook.book')
|
||
|
|
Lines = pool.get('cashbook.line')
|
||
|
|
|
||
|
|
types = self.prep_type()
|
||
|
|
category = self.prep_category(cattype='in')
|
||
|
|
company = self.prep_company()
|
||
|
|
party = self.prep_party()
|
||
|
|
book, = Book.create([{
|
||
|
|
'name': 'Book 1',
|
||
|
|
'btype': types.id,
|
||
|
|
'company': company.id,
|
||
|
|
'currency': company.currency.id,
|
||
|
|
'number_sequ': self.prep_sequence().id,
|
||
|
|
'start_date': date(2022, 5, 1),
|
||
|
|
'lines': [('create', [{
|
||
|
|
'date': date(2022, 5, 1),
|
||
|
|
'description': 'Text 1',
|
||
|
|
'category': category.id,
|
||
|
|
'bookingtype': 'in',
|
||
|
|
'amount': Decimal('1.0'),
|
||
|
|
'party': party.id,
|
||
|
|
},])],
|
||
|
|
}])
|
||
|
|
self.assertEqual(book.name, 'Book 1')
|
||
|
|
self.assertEqual(len(book.lines), 1)
|
||
|
|
self.assertEqual(book.state, 'open')
|
||
|
|
|
||
|
|
# add image to line-1
|
||
|
|
Lines.write(*[
|
||
|
|
[book.lines[0]],
|
||
|
|
{
|
||
|
|
'media': img_data_png,
|
||
|
|
'media_name': 'image.png',
|
||
|
|
}])
|
||
|
|
self.assertEqual(book.lines[0].media_size, 18428)
|
||
|
|
self.assertEqual(book.lines[0].media_mime, 'image/png')
|
||
|
|
self.assertEqual(book.lines[0].media_name, 'image.png')
|
||
|
|
|
||
|
|
# replace image at line-1 by pdf
|
||
|
|
Lines.write(*[
|
||
|
|
[book.lines[0]],
|
||
|
|
{
|
||
|
|
'media': dok_data_pdf,
|
||
|
|
}])
|
||
|
|
self.assertEqual(book.lines[0].media_size, 8724)
|
||
|
|
self.assertEqual(book.lines[0].media_mime, 'application/pdf')
|
||
|
|
self.assertEqual(book.lines[0].media_name, 'image.png')
|
||
|
|
|
||
|
|
# create line with pdf
|
||
|
|
Book.write(*[
|
||
|
|
[book],
|
||
|
|
{
|
||
|
|
'lines': [('create', [{
|
||
|
|
'date': date(2022, 5, 2),
|
||
|
|
'description': 'Text 2',
|
||
|
|
'category': category.id,
|
||
|
|
'bookingtype': 'in',
|
||
|
|
'amount': Decimal('1.0'),
|
||
|
|
'party': party.id,
|
||
|
|
'media': dok_data_pdf,
|
||
|
|
'media_name': 'data.pdf',
|
||
|
|
}])],
|
||
|
|
}
|
||
|
|
])
|
||
|
|
self.assertEqual(len(book.lines), 2)
|
||
|
|
self.assertEqual(book.lines[1].media_size, 8724)
|
||
|
|
self.assertEqual(book.lines[1].media_mime, 'application/pdf')
|
||
|
|
self.assertEqual(book.lines[1].media_name, 'data.pdf')
|
||
|
|
|
||
|
|
@with_transaction()
|
||
|
|
def test_media_add_invalid_file(self):
|
||
|
|
""" create cook/line, add txt-file
|
||
|
|
"""
|
||
|
|
pool = Pool()
|
||
|
|
Book = pool.get('cashbook.book')
|
||
|
|
Lines = pool.get('cashbook.line')
|
||
|
|
|
||
|
|
types = self.prep_type()
|
||
|
|
category = self.prep_category(cattype='in')
|
||
|
|
company = self.prep_company()
|
||
|
|
party = self.prep_party()
|
||
|
|
book, = Book.create([{
|
||
|
|
'name': 'Book 1',
|
||
|
|
'btype': types.id,
|
||
|
|
'company': company.id,
|
||
|
|
'currency': company.currency.id,
|
||
|
|
'number_sequ': self.prep_sequence().id,
|
||
|
|
'start_date': date(2022, 5, 1),
|
||
|
|
'lines': [('create', [{
|
||
|
|
'date': date(2022, 5, 1),
|
||
|
|
'description': 'Text 1',
|
||
|
|
'category': category.id,
|
||
|
|
'bookingtype': 'in',
|
||
|
|
'amount': Decimal('1.0'),
|
||
|
|
'party': party.id,
|
||
|
|
},])],
|
||
|
|
}])
|
||
|
|
self.assertEqual(book.name, 'Book 1')
|
||
|
|
self.assertEqual(len(book.lines), 1)
|
||
|
|
self.assertEqual(book.state, 'open')
|
||
|
|
|
||
|
|
# add invalid file
|
||
|
|
self.assertRaisesRegex(UserError,
|
||
|
|
"The file type 'text/plain' of the record '05/02/2022|Rev|1.00 usd|Text 2 [Cat1]' is not allowed. (allowed: PNG, JPG, PDF)",
|
||
|
|
Book.write,
|
||
|
|
*[
|
||
|
|
[book],
|
||
|
|
{
|
||
|
|
'lines': [('create', [{
|
||
|
|
'date': date(2022, 5, 2),
|
||
|
|
'description': 'Text 2',
|
||
|
|
'category': category.id,
|
||
|
|
'bookingtype': 'in',
|
||
|
|
'amount': Decimal('1.0'),
|
||
|
|
'party': party.id,
|
||
|
|
'media': text_data,
|
||
|
|
'media_name': 'text.txt',
|
||
|
|
}])],
|
||
|
|
}
|
||
|
|
])
|
||
|
|
|
||
|
|
# replace image at line-1 by invalid file
|
||
|
|
self.assertRaisesRegex(UserError,
|
||
|
|
"The file type 'text/plain' of the record '05/02/2022|Rev|1.00 usd|Text 2 [Cat1]' is not allowed. (allowed: PNG, JPG, PDF)",
|
||
|
|
Lines.write,
|
||
|
|
*[
|
||
|
|
[book.lines[0]],
|
||
|
|
{
|
||
|
|
'media': text_data,
|
||
|
|
'media_name': 'text.txt',
|
||
|
|
},
|
||
|
|
])
|
||
|
|
|
||
|
|
@with_transaction()
|
||
|
|
def test_media_add_big_file(self):
|
||
|
|
""" create cook/line, add big png-file
|
||
|
|
"""
|
||
|
|
pool = Pool()
|
||
|
|
Book = pool.get('cashbook.book')
|
||
|
|
Lines = pool.get('cashbook.line')
|
||
|
|
|
||
|
|
types = self.prep_type()
|
||
|
|
category = self.prep_category(cattype='in')
|
||
|
|
company = self.prep_company()
|
||
|
|
party = self.prep_party()
|
||
|
|
|
||
|
|
book, = Book.create([{
|
||
|
|
'name': 'Book 1',
|
||
|
|
'btype': types.id,
|
||
|
|
'company': company.id,
|
||
|
|
'currency': company.currency.id,
|
||
|
|
'number_sequ': self.prep_sequence().id,
|
||
|
|
'start_date': date(2022, 5, 1),
|
||
|
|
'lines': [('create', [{
|
||
|
|
'date': date(2022, 5, 1),
|
||
|
|
'description': 'Text 1',
|
||
|
|
'category': category.id,
|
||
|
|
'bookingtype': 'in',
|
||
|
|
'amount': Decimal('1.0'),
|
||
|
|
'party': party.id,
|
||
|
|
},])],
|
||
|
|
}])
|
||
|
|
self.assertEqual(book.name, 'Book 1')
|
||
|
|
self.assertEqual(len(book.lines), 1)
|
||
|
|
self.assertEqual(book.state, 'open')
|
||
|
|
|
||
|
|
# construct image
|
||
|
|
with BytesIO() as fhdl:
|
||
|
|
img1 = Image.new('RGB', (3200, 1340))
|
||
|
|
img1.save(fhdl, 'PNG', optimize=True)
|
||
|
|
del img1
|
||
|
|
|
||
|
|
fhdl.seek(0)
|
||
|
|
img_big_data = fhdl.read()
|
||
|
|
|
||
|
|
# create line with png, should be resized
|
||
|
|
Book.write(*[
|
||
|
|
[book],
|
||
|
|
{
|
||
|
|
'lines': [('create', [{
|
||
|
|
'date': date(2022, 5, 2),
|
||
|
|
'description': 'Text 2',
|
||
|
|
'category': category.id,
|
||
|
|
'bookingtype': 'in',
|
||
|
|
'amount': Decimal('1.0'),
|
||
|
|
'party': party.id,
|
||
|
|
'media': img_big_data,
|
||
|
|
'media_name': 'big.png',
|
||
|
|
}])],
|
||
|
|
}
|
||
|
|
])
|
||
|
|
self.assertEqual(len(book.lines), 2)
|
||
|
|
self.assertEqual(book.lines[1].media_mime, 'image/jpeg')
|
||
|
|
self.assertEqual(book.lines[1].media_size, 10221)
|
||
|
|
self.assertEqual(book.lines[1].media_name, 'big.jpg')
|
||
|
|
|
||
|
|
# check image size
|
||
|
|
with BytesIO(book.lines[1].media) as fhdl:
|
||
|
|
img2 = Image.open(fhdl, 'r')
|
||
|
|
self.assertEqual(img2.size, (2000, 837))
|
||
|
|
|
||
|
|
# end LineTestCase
|