media ok + test
This commit is contained in:
Frederik Jaeckel
parent
0100653643
commit
2933a217e2
13 changed files with 565 additions and 33 deletions
24
tests/__init__.py
Normal file
24
tests/__init__.py
Normal file
|
|
@ -0,0 +1,24 @@
|
|||
# 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 trytond.tests.test_tryton
|
||||
import unittest
|
||||
|
||||
from trytond.modules.cashbook_media.tests.test_line import LineTestCase
|
||||
|
||||
|
||||
__all__ = ['suite']
|
||||
|
||||
|
||||
class CashbookTestCase(\
|
||||
LineTestCase,
|
||||
):
|
||||
'Test cashbook module'
|
||||
module = 'cashbook_media'
|
||||
|
||||
# end CashbookTestCase
|
||||
|
||||
def suite():
|
||||
suite = trytond.tests.test_tryton.suite()
|
||||
suite.addTests(unittest.TestLoader().loadTestsFromTestCase(CashbookTestCase))
|
||||
return suite
|
||||
5
tests/img_data.py
Normal file
5
tests/img_data.py
Normal file
File diff suppressed because one or more lines are too long
228
tests/test_line.py
Normal file
228
tests/test_line.py
Normal file
|
|
@ -0,0 +1,228 @@
|
|||
# -*- 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
|
||||
Loading…
Add table
Add a link
Reference in a new issue