backend improvements

- move_page
- merge_all_single_pages
- constructor with optional parameter
This commit is contained in:
Niklas Müller 2024-01-22 16:47:35 +01:00
parent f4b8029593
commit 6b74b89029
6 changed files with 300 additions and 38 deletions

View file

@ -1,26 +1,116 @@
import pytest
import os
import shutil
from pdf_util.pdf_project_manager import pdf_project_manager
def test_basic_object_creation():
test_pdf_project_manager = pdf_project_manager()
print(test_pdf_project_manager.uuid)
assert len(test_pdf_project_manager.uuid) == 36
shutil.rmtree('/app/projects/' + test_pdf_project_manager.uuid)
def test_folder_creation():
test_pdf_project_manager = pdf_project_manager()
print(test_pdf_project_manager.uuid)
assert os.path.isdir('/app/projects/' + test_pdf_project_manager.uuid)
shutil.rmtree('/app/projects/' + test_pdf_project_manager.uuid)
def test_Projecttest_1():
"""
test_project = pdf_project_manager()
def test_merge_all_single_pages():
test_pdf_project_manager = pdf_project_manager()
print(test_pdf_project_manager.uuid)
assert os.path.isdir('/app/projects/' + test_pdf_project_manager.uuid)
"""
assert True
test_pdf_project_manager.add_pdf("/app/tests/sample_pdfs/sample_10_page.pdf")
print(os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf'))
assert os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf')
print(os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size)
assert os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size == os.stat("/app/tests/sample_pdfs/sample_10_page.pdf").st_size
os.remove('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf')
assert not os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf')
test_pdf_project_manager.merge_all_single_pages()
print(os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf'))
assert os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf')
print(os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size)
assert os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size == 81291
shutil.rmtree('/app/projects/' + test_pdf_project_manager.uuid)
def test_add_multiple_pdfs():
test_pdf_project_manager = pdf_project_manager()
print(test_pdf_project_manager.uuid)
assert os.path.isdir('/app/projects/' + test_pdf_project_manager.uuid)
test_pdf_project_manager.add_pdf("/app/tests/sample_pdfs/sample_10_page.pdf")
print(os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf'))
assert os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf')
print(os.path.isdir('/app/projects/' + test_pdf_project_manager.uuid + '/splitted'))
assert os.path.isdir('/app/projects/' + test_pdf_project_manager.uuid + '/splitted')
print(os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/splitted/0001.pdf'))
assert os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/splitted/0001.pdf')
print(os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/splitted/0001.pdf').st_size)
assert os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/splitted/0001.pdf').st_size == 3167
test_pdf_project_manager.add_pdf("/app/tests/sample_pdfs/sample_2_page.pdf")
print(os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf'))
assert os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf')
print(os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/splitted/0001.pdf').st_size)
assert os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/splitted/0001.pdf').st_size == 3167
print(os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/splitted/0011.pdf').st_size)
assert os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/splitted/0011.pdf').st_size == 1804
print(os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size)
assert os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size == 48408
shutil.rmtree('/app/projects/' + test_pdf_project_manager.uuid)
def test_move_pages():
test_pdf_project_manager = pdf_project_manager()
print(test_pdf_project_manager.uuid)
assert os.path.isdir('/app/projects/' + test_pdf_project_manager.uuid)
test_pdf_project_manager.add_pdf("/app/tests/sample_pdfs/sample_10_page.pdf")
test_pdf_project_manager.add_pdf("/app/tests/sample_pdfs/sample_2_page.pdf")
print(os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size)
assert os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size == 48408
os.remove('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf')
assert not os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf')
test_pdf_project_manager.move_page(1, 4)
print(os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/splitted/0004.pdf').st_size)
assert os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/splitted/0004.pdf').st_size == 3167
print(os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size)
assert os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size == 83909
os.remove('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf')
assert not os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf')
test_pdf_project_manager.move_page(4, 1)
print(os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/splitted/0001.pdf').st_size)
assert os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/splitted/0001.pdf').st_size == 3167
print(os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size)
assert os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size == 83908
os.remove('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf')
assert not os.path.isfile('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf')
test_pdf_project_manager.move_page(1, 12)
test_pdf_project_manager.move_page(12, 2)
assert os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/splitted/0002.pdf').st_size == 3167
print(os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size)
assert os.stat('/app/projects/' + test_pdf_project_manager.uuid + '/complete.pdf').st_size == 83909
shutil.rmtree('/app/projects/' + test_pdf_project_manager.uuid)