software-engineering/document-service
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Initial commit of document-service

Browse Source
This commit is contained in:
HaimKortovich
2026-04-23 16:20:58 -05:00
commit 51d60f0032
30 changed files with 4357 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

371
tests/fixtures/generate_test_pdfs.py vendored Normal file
View File

@@ -0,0 +1,371 @@
"""
Generate test PDF files for document-service testing.
This script creates various test PDFs with actual AcroForm fields:
- Simple form PDF with basic form fields
- Complex form PDF with multiple field types
- No form PDF without form fields
- Large form PDF for size validation testing
"""
import os
from reportlab.pdfgen import canvas
from reportlab.lib.pagesizes import letter
from reportlab.lib import colors
from pypdf import PdfReader, PdfWriter
from pypdf.generic import (
NameObject,
create_string_object,
NumberObject,
ArrayObject,
DictionaryObject,
BooleanObject,
)
# Output directory
OUTPUT_DIR = os.path.dirname(os.path.abspath(__file__))
def create_simple_form_pdf():
"""Create a simple PDF with basic form fields."""
output_path = os.path.join(OUTPUT_DIR, "simple_form.pdf")
# Create base PDF with reportlab
c = canvas.Canvas(output_path, pagesize=letter)
c.setFont("Helvetica", 16)
c.drawString(100, 750, "Simple Form Test")
c.setFont("Helvetica", 12)
c.drawString(100, 700, "Name:")
c.drawString(100, 650, "Email:")
c.drawString(100, 600, "Phone:")
c.drawString(100, 550, "Country:")
c.drawString(100, 500, "Birth Date:")
c.drawString(100, 450, "Agree to Terms:")
c.save()
# Add actual form fields using pypdf
reader = PdfReader(output_path)
writer = PdfWriter()
# Copy the page
page = reader.pages[0]
writer.add_page(page)
# Create form fields
fields = []
# Name field (text)
name_field = DictionaryObject({
NameObject("/FT"): NameObject("/Tx"),
NameObject("/T"): create_string_object("name"),
NameObject("/V"): create_string_object(""),
NameObject("/Rect"): ArrayObject([NumberObject(200), NumberObject(690), NumberObject(400), NumberObject(710)]),
NameObject("/Ff"): NumberObject(0),
})
fields.append(name_field)
# Email field (text)
email_field = DictionaryObject({
NameObject("/FT"): NameObject("/Tx"),
NameObject("/T"): create_string_object("email"),
NameObject("/V"): create_string_object(""),
NameObject("/Rect"): ArrayObject([NumberObject(200), NumberObject(640), NumberObject(400), NumberObject(660)]),
NameObject("/Ff"): NumberObject(0),
})
fields.append(email_field)
# Phone field (text)
phone_field = DictionaryObject({
NameObject("/FT"): NameObject("/Tx"),
NameObject("/T"): create_string_object("phone"),
NameObject("/V"): create_string_object(""),
NameObject("/Rect"): ArrayObject([NumberObject(200), NumberObject(590), NumberObject(400), NumberObject(610)]),
NameObject("/Ff"): NumberObject(0),
})
fields.append(phone_field)
# Country field (dropdown/choice)
country_field = DictionaryObject({
NameObject("/FT"): NameObject("/Ch"),
NameObject("/T"): create_string_object("country"),
NameObject("/V"): create_string_object(""),
NameObject("/Opt"): ArrayObject([
create_string_object("USA"),
create_string_object("Canada"),
create_string_object("UK"),
create_string_object("Germany"),
create_string_object("France"),
]),
NameObject("/Rect"): ArrayObject([NumberObject(200), NumberObject(540), NumberObject(400), NumberObject(560)]),
NameObject("/Ff"): NumberObject(0),
})
fields.append(country_field)
# Birth date field (text)
birth_date_field = DictionaryObject({
NameObject("/FT"): NameObject("/Tx"),
NameObject("/T"): create_string_object("birth_date"),
NameObject("/V"): create_string_object(""),
NameObject("/Rect"): ArrayObject([NumberObject(200), NumberObject(490), NumberObject(400), NumberObject(510)]),
NameObject("/Ff"): NumberObject(0),
})
fields.append(birth_date_field)
# Agree terms field (checkbox)
agree_field = DictionaryObject({
NameObject("/FT"): NameObject("/Btn"),
NameObject("/T"): create_string_object("agree_terms"),
NameObject("/V"): NameObject("/Off"),
NameObject("/Rect"): ArrayObject([NumberObject(200), NumberObject(440), NumberObject(220), NumberObject(460)]),
NameObject("/Ff"): NumberObject(0),
})
fields.append(agree_field)
# Add fields to the page
page = writer.pages[0]
if "/Annots" not in page:
page[NameObject("/Annots")] = ArrayObject()
for field in fields:
field_obj = DictionaryObject({
NameObject("/Subtype"): NameObject("/Widget"),
NameObject("/FT"): field[NameObject("/FT")],
NameObject("/T"): field[NameObject("/T")],
NameObject("/V"): field.get(NameObject("/V"), NameObject("")),
NameObject("/Rect"): field[NameObject("/Rect")],
NameObject("/Ff"): field.get(NameObject("/Ff"), NumberObject(0)),
})
if NameObject("/Opt") in field:
field_obj[NameObject("/Opt")] = field[NameObject("/Opt")]
page[NameObject("/Annots")].append(field_obj)
# Add AcroForm to the document
acroform = DictionaryObject({
NameObject("/Fields"): ArrayObject(fields),
})
writer._root_object[NameObject("/AcroForm")] = acroform
# Save the PDF
with open(output_path, "wb") as f:
writer.write(f)
print(f"Created: {output_path}")
def create_complex_form_pdf():
"""Create a complex PDF with multiple field types."""
output_path = os.path.join(OUTPUT_DIR, "complex_form.pdf")
# Create base PDF with reportlab
c = canvas.Canvas(output_path, pagesize=letter)
c.setFont("Helvetica", 16)
c.drawString(100, 750, "Complex Form Test")
c.setFont("Helvetica", 12)
c.drawString(100, 700, "First Name:")
c.drawString(100, 650, "Last Name:")
c.drawString(100, 600, "Email:")
c.drawString(100, 550, "Phone:")
c.drawString(100, 500, "Address:")
c.drawString(100, 450, "City:")
c.drawString(100, 400, "State:")
c.drawString(100, 350, "Zip Code:")
c.drawString(100, 300, "Country:")
c.drawString(100, 250, "Gender:")
c.drawString(100, 200, "Interests:")
c.drawString(100, 150, "Agree to Terms:")
c.drawString(100, 100, "Signature:")
c.save()
# Add actual form fields using pypdf
reader = PdfReader(output_path)
writer = PdfWriter()
# Copy the page
page = reader.pages[0]
writer.add_page(page)
# Create form fields
fields = []
# Text fields
text_fields = [
('first_name', 200, 690),
('last_name', 200, 640),
('email', 200, 590),
('phone', 200, 540),
('address', 200, 490),
('city', 200, 440),
('state', 200, 390),
('zip_code', 200, 340),
]
for name, x, y in text_fields:
field = DictionaryObject({
NameObject("/FT"): NameObject("/Tx"),
NameObject("/T"): create_string_object(name),
NameObject("/V"): create_string_object(""),
NameObject("/Rect"): ArrayObject([NumberObject(x), NumberObject(y), NumberObject(x + 200), NumberObject(y + 20)]),
NameObject("/Ff"): NumberObject(0),
})
fields.append(field)
# Country dropdown
country_field = DictionaryObject({
NameObject("/FT"): NameObject("/Ch"),
NameObject("/T"): create_string_object("country"),
NameObject("/V"): create_string_object(""),
NameObject("/Opt"): ArrayObject([
create_string_object("USA"),
create_string_object("Canada"),
create_string_object("UK"),
create_string_object("Germany"),
create_string_object("France"),
]),
NameObject("/Rect"): ArrayObject([NumberObject(200), NumberObject(290), NumberObject(400), NumberObject(310)]),
NameObject("/Ff"): NumberObject(0),
})
fields.append(country_field)
# Radio buttons for gender
male_field = DictionaryObject({
NameObject("/FT"): NameObject("/Btn"),
NameObject("/T"): create_string_object("gender"),
NameObject("/V"): create_string_object("male"),
NameObject("/Rect"): ArrayObject([NumberObject(200), NumberObject(240), NumberObject(220), NumberObject(260)]),
NameObject("/Ff"): NumberObject(0),
})
fields.append(male_field)
female_field = DictionaryObject({
NameObject("/FT"): NameObject("/Btn"),
NameObject("/T"): create_string_object("gender"),
NameObject("/V"): create_string_object("female"),
NameObject("/Rect"): ArrayObject([NumberObject(300), NumberObject(240), NumberObject(320), NumberObject(260)]),
NameObject("/Ff"): NumberObject(0),
})
fields.append(female_field)
# Checkboxes for interests
interests = ['reading', 'sports', 'music', 'travel']
for i, interest in enumerate(interests):
field = DictionaryObject({
NameObject("/FT"): NameObject("/Btn"),
NameObject("/T"): create_string_object(interest),
NameObject("/V"): NameObject("/Off"),
NameObject("/Rect"): ArrayObject([NumberObject(200), NumberObject(190 - (i * 30)), NumberObject(220), NumberObject(210 - (i * 30))]),
NameObject("/Ff"): NumberObject(0),
})
fields.append(field)
# Checkbox for agree terms
agree_field = DictionaryObject({
NameObject("/FT"): NameObject("/Btn"),
NameObject("/T"): create_string_object("agree_terms"),
NameObject("/V"): NameObject("/Off"),
NameObject("/Rect"): ArrayObject([NumberObject(200), NumberObject(140), NumberObject(220), NumberObject(160)]),
NameObject("/Ff"): NumberObject(0),
})
fields.append(agree_field)
# Signature field
signature_field = DictionaryObject({
NameObject("/FT"): NameObject("/Tx"),
NameObject("/T"): create_string_object("signature"),
NameObject("/V"): create_string_object(""),
NameObject("/Rect"): ArrayObject([NumberObject(200), NumberObject(90), NumberObject(400), NumberObject(110)]),
NameObject("/Ff"): NumberObject(0),
})
fields.append(signature_field)
# Add fields to the page
page = writer.pages[0]
if "/Annots" not in page:
page[NameObject("/Annots")] = ArrayObject()
for field in fields:
field_obj = DictionaryObject({
NameObject("/Subtype"): NameObject("/Widget"),
NameObject("/FT"): field[NameObject("/FT")],
NameObject("/T"): field[NameObject("/T")],
NameObject("/V"): field.get(NameObject("/V"), NameObject("")),
NameObject("/Rect"): field[NameObject("/Rect")],
NameObject("/Ff"): field.get(NameObject("/Ff"), NumberObject(0)),
})
if NameObject("/Opt") in field:
field_obj[NameObject("/Opt")] = field[NameObject("/Opt")]
page[NameObject("/Annots")].append(field_obj)
# Add AcroForm to the document
acroform = DictionaryObject({
NameObject("/Fields"): ArrayObject(fields),
})
writer._root_object[NameObject("/AcroForm")] = acroform
# Save the PDF
with open(output_path, "wb") as f:
writer.write(f)
print(f"Created: {output_path}")
def create_no_form_pdf():
"""Create a PDF without form fields."""
output_path = os.path.join(OUTPUT_DIR, "no_form.pdf")
# Create simple PDF without form fields
c = canvas.Canvas(output_path, pagesize=letter)
c.setFont("Helvetica", 16)
c.drawString(100, 750, "No Form Test")
c.setFont("Helvetica", 12)
c.drawString(100, 700, "This PDF has no form fields.")
c.drawString(100, 650, "It is used for testing field discovery")
c.drawString(100, 600, "on documents without AcroForm fields.")
c.save()
print(f"Created: {output_path}")
def create_large_form_pdf():
"""Create a large PDF for size validation testing."""
output_path = os.path.join(OUTPUT_DIR, "large_form.pdf")
# Create a larger PDF with more content
c = canvas.Canvas(output_path, pagesize=letter)
c.setFont("Helvetica", 16)
c.drawString(100, 750, "Large Form Test")
c.setFont("Helvetica", 12)
y = 700
for i in range(50):
c.drawString(100, y, f"Field {i + 1}:")
y -= 50
if y < 50:
c.showPage()
y = 700
c.save()
print(f"Created: {output_path}")
def main():
"""Generate all test PDF files."""
print("Generating test PDF files...")
print(f"Output directory: {OUTPUT_DIR}")
print()
create_simple_form_pdf()
create_complex_form_pdf()
create_no_form_pdf()
create_large_form_pdf()
print()
print("All test PDF files generated successfully!")
if __name__ == "__main__":
main()