nemo
/
pystitcher
mirror of https://github.com/captn3m0/pystitcher.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[tests] Validate metadata

This commit is contained in:
Nemo 2021-07-04 02:12:39 +05:30
parent 08cdc2e39b
commit df75e04d1d
3 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/book-flatten.md
View File

@ -1,5 +1,4 @@
existing_bookmarks: flatten existing_bookmarks: flatten
title: Super Jelly Book
# Super Potato Book # Super Potato Book

3
tests/book-rotate.md
View File

@ -1,7 +1,4 @@
existing_bookmarks: remove existing_bookmarks: remove
author: Wiki, the Cat
subject: A book about adventures of Wiki, the cat.
keywords: wiki,potato,jelly
# Super Potato Book # Super Potato Book
# Volume 1 # Volume 1

20
tests/test_integration.py
View File

@ -7,11 +7,11 @@ from itertools import chain
ROOT_DIR = os.path.dirname(os.path.abspath(__file__)) + "/../" ROOT_DIR = os.path.dirname(os.path.abspath(__file__)) + "/../"
TEST_DATA = [ TEST_DATA = [
("clean",6, [('Super Potato Book', 0, 0), ('Volume 1', 0, 0), ('Part 1', 0, 1), ('Volume 2', 3, 0), ('Part 2', 3, 1)]), ("clean",6, {'Author': 'Wiki, the Cat', 'Title': 'Super Potato Book', 'Subject': 'A book about adventures of Wiki, the cat.', 'Keywords': 'wiki,potato,jelly'}, [('Super Potato Book', 0, 0), ('Volume 1', 0, 0), ('Part 1', 0, 1), ('Volume 2', 3, 0), ('Part 2', 3, 1)]),
("keep",6, [('Super Potato Book', 0, 0), ('Volume 1', 0, 0), ('Part 1', 0, 1), ('Chapter 1', 0, 2), ('Chapter 2', 1, 2), ('Scene 1', 1, 3), ('Scene 2', 2, 3), ('Volume 2', 3, 0), ('Part 3', 3, 1), ('Chapter 3', 3, 2), ('Chapter 4', 4, 2), ('Scene 3', 4, 3), ('Scene 4', 5, 3)]), ("keep",6, {'Title': 'Super Potato Book'}, [('Super Potato Book', 0, 0), ('Volume 1', 0, 0), ('Part 1', 0, 1), ('Chapter 1', 0, 2), ('Chapter 2', 1, 2), ('Scene 1', 1, 3), ('Scene 2', 2, 3), ('Volume 2', 3, 0), ('Part 3', 3, 1), ('Chapter 3', 3, 2), ('Chapter 4', 4, 2), ('Scene 3', 4, 3), ('Scene 4', 5, 3)]),
("flatten", 6, [('Super Potato Book', 0, 0), ('Volume 1', 0, 0), ('Part 1', 0, 1), ('Chapter 1', 0, 2), ('Chapter 2', 1, 2), ('Scene 1', 1, 2), ('Scene 2', 2, 2), ('Volume 2', 3, 0), ('Part 3', 3, 1), ('Chapter 3', 3, 2), ('Chapter 4', 4, 2), ('Scene 3', 4, 2), ('Scene 4', 5, 2)]), ("flatten", 6, {}, [('Super Potato Book', 0, 0), ('Volume 1', 0, 0), ('Part 1', 0, 1), ('Chapter 1', 0, 2), ('Chapter 2', 1, 2), ('Scene 1', 1, 2), ('Scene 2', 2, 2), ('Volume 2', 3, 0), ('Part 3', 3, 1), ('Chapter 3', 3, 2), ('Chapter 4', 4, 2), ('Scene 3', 4, 2), ('Scene 4', 5, 2)]),
("rotate", 9, [('Super Potato Book', 0, 0), ('Volume 1', 0, 0), ('Part 1', 0, 1), ('Volume 2', 3, 0), ('Part 2', 3, 1), ('Volume 3', 6, 0), ('Part 3', 6, 1)]), ("rotate", 9, {}, [('Super Potato Book', 0, 0), ('Volume 1', 0, 0), ('Part 1', 0, 1), ('Volume 2', 3, 0), ('Part 2', 3, 1), ('Volume 3', 6, 0), ('Part 3', 6, 1)]),
("min",3, [('Part 1', 0, 0), ('Chapter 1', 0, 1), ('Chapter 2', 1, 1), ('Scene 1', 1, 2), ('Scene 2', 2, 2)]) ("min",3, {}, [('Part 1', 0, 0), ('Chapter 1', 0, 1), ('Chapter 2', 1, 1), ('Scene 1', 1, 2), ('Scene 2', 2, 2)])
] ]
def pdf_name(name): def pdf_name(name):
@ -41,10 +41,14 @@ def get_all_bookmarks(pdf):
bookmarks = flatten_bookmarks(pdf.getOutlines()) bookmarks = flatten_bookmarks(pdf.getOutlines())
return [(d[0]['/Title'], pdf.getDestinationPageNumber(d[0]), d[1]) for d in bookmarks] return [(d[0]['/Title'], pdf.getDestinationPageNumber(d[0]), d[1]) for d in bookmarks]
@pytest.mark.parametrize("name,pages,bookmarks", TEST_DATA) @pytest.mark.parametrize("name,pages,metadata, bookmarks", TEST_DATA)
def test_book(name, pages, bookmarks): def test_book(name, pages, metadata, bookmarks):
output_file = render(name) output_file = render(name)
pdf = PyPDF3.PdfFileReader(output_file) pdf = PyPDF3.PdfFileReader(output_file)
assert pages == pdf.getNumPages() assert pages == pdf.getNumPages()
assert bookmarks == get_all_bookmarks(pdf) assert bookmarks == get_all_bookmarks(pdf)
info = pdf.getDocumentInfo()
assert 'pystitcher/1.0.1' == info['/Producer']
assert 'pystitcher/1.0.1' == info['/Creator']
for key in metadata:
assert info["/%s" % key] == metadata[key]