Fixes and support for ApprovalSignature and TimeStampSignature
This commit is contained in:
@@ -1,70 +1,88 @@
|
||||
package sign
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"testing"
|
||||
|
||||
"github.com/digitorus/pdf"
|
||||
)
|
||||
|
||||
var test_files = []struct {
|
||||
var testFiles = []struct {
|
||||
file string
|
||||
expected_catalog string
|
||||
expectedCatalogs map[CertType]string
|
||||
}{
|
||||
{"../testfiles/testfile20.pdf", "11 0 obj\n<< /Type /Catalog /Version /2.0 /Pages 3 0 R /AcroForm << /Fields [10 0 R] /NeedAppearances false /SigFlags 1 >> /Perms << /UR3 0 0 R >> >>\nendobj\n"},
|
||||
{"../testfiles/testfile21.pdf", "17 0 obj\n<< /Type /Catalog /Version /1.0 /Names 6 0 R /Pages 9 0 R /AcroForm << /Fields [16 0 R] /NeedAppearances false /SigFlags 1 >> /Perms << /UR3 0 0 R >> >>\nendobj\n"},
|
||||
{
|
||||
file: "../testfiles/testfile20.pdf",
|
||||
expectedCatalogs: map[CertType]string{
|
||||
CertificationSignature: "11 0 obj\n<< /Type /Catalog /Pages 3 0 R /AcroForm << /Fields [10 0 R] /NeedAppearances false /SigFlags 3 >> >>\nendobj\n",
|
||||
UsageRightsSignature: "11 0 obj\n<< /Type /Catalog /Pages 3 0 R /AcroForm << /Fields [10 0 R] /NeedAppearances false /SigFlags 1 >> >>\nendobj\n",
|
||||
ApprovalSignature: "11 0 obj\n<< /Type /Catalog /Pages 3 0 R /AcroForm << /Fields [10 0 R] /NeedAppearances false /SigFlags 3 >> >>\nendobj\n",
|
||||
},
|
||||
},
|
||||
{
|
||||
file: "../testfiles/testfile21.pdf",
|
||||
expectedCatalogs: map[CertType]string{
|
||||
CertificationSignature: "17 0 obj\n<< /Type /Catalog /Pages 9 0 R /Names 6 0 R /AcroForm << /Fields [16 0 R] /NeedAppearances false /SigFlags 3 >> >>\nendobj\n",
|
||||
UsageRightsSignature: "17 0 obj\n<< /Type /Catalog /Pages 9 0 R /Names 6 0 R /AcroForm << /Fields [16 0 R] /NeedAppearances false /SigFlags 1 >> >>\nendobj\n",
|
||||
ApprovalSignature: "17 0 obj\n<< /Type /Catalog /Pages 9 0 R /Names 6 0 R /AcroForm << /Fields [16 0 R] /NeedAppearances false /SigFlags 3 >> >>\nendobj\n"},
|
||||
},
|
||||
}
|
||||
|
||||
func TestCreateCatalog(t *testing.T) {
|
||||
for _, test_file := range test_files {
|
||||
input_file, err := os.Open(test_file.file)
|
||||
if err != nil {
|
||||
t.Errorf("Failed to load test PDF")
|
||||
return
|
||||
}
|
||||
for _, testFile := range testFiles {
|
||||
for certType, expectedCatalog := range testFile.expectedCatalogs {
|
||||
t.Run(fmt.Sprintf("%s_certType-%d", testFile.file, certType), func(st *testing.T) {
|
||||
inputFile, err := os.Open(testFile.file)
|
||||
if err != nil {
|
||||
st.Errorf("Failed to load test PDF")
|
||||
return
|
||||
}
|
||||
|
||||
finfo, err := input_file.Stat()
|
||||
if err != nil {
|
||||
t.Errorf("Failed to load test PDF")
|
||||
return
|
||||
}
|
||||
size := finfo.Size()
|
||||
finfo, err := inputFile.Stat()
|
||||
if err != nil {
|
||||
st.Errorf("Failed to load test PDF")
|
||||
return
|
||||
}
|
||||
size := finfo.Size()
|
||||
|
||||
rdr, err := pdf.NewReader(input_file, size)
|
||||
if err != nil {
|
||||
t.Errorf("Failed to load test PDF")
|
||||
return
|
||||
}
|
||||
rdr, err := pdf.NewReader(inputFile, size)
|
||||
if err != nil {
|
||||
st.Errorf("Failed to load test PDF")
|
||||
return
|
||||
}
|
||||
|
||||
context := SignContext{
|
||||
Filesize: size + 1,
|
||||
PDFReader: rdr,
|
||||
InputFile: input_file,
|
||||
VisualSignData: VisualSignData{
|
||||
ObjectId: uint32(rdr.XrefInformation.ItemCount),
|
||||
},
|
||||
CatalogData: CatalogData{
|
||||
ObjectId: uint32(rdr.XrefInformation.ItemCount) + 1,
|
||||
},
|
||||
InfoData: InfoData{
|
||||
ObjectId: uint32(rdr.XrefInformation.ItemCount) + 2,
|
||||
},
|
||||
SignData: SignData{
|
||||
Signature: SignDataSignature{
|
||||
CertType: UsageRightsSignature,
|
||||
DocMDPPerm: AllowFillingExistingFormFieldsAndSignaturesPerms,
|
||||
},
|
||||
},
|
||||
}
|
||||
context := SignContext{
|
||||
Filesize: size + 1,
|
||||
PDFReader: rdr,
|
||||
InputFile: inputFile,
|
||||
VisualSignData: VisualSignData{
|
||||
ObjectId: uint32(rdr.XrefInformation.ItemCount),
|
||||
},
|
||||
CatalogData: CatalogData{
|
||||
ObjectId: uint32(rdr.XrefInformation.ItemCount) + 1,
|
||||
},
|
||||
InfoData: InfoData{
|
||||
ObjectId: uint32(rdr.XrefInformation.ItemCount) + 2,
|
||||
},
|
||||
SignData: SignData{
|
||||
Signature: SignDataSignature{
|
||||
CertType: certType,
|
||||
DocMDPPerm: AllowFillingExistingFormFieldsAndSignaturesPerms,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
catalog, err := context.createCatalog()
|
||||
if err != nil {
|
||||
t.Errorf("%s", err.Error())
|
||||
return
|
||||
}
|
||||
catalog, err := context.createCatalog()
|
||||
if err != nil {
|
||||
st.Errorf("%s", err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
if catalog != test_file.expected_catalog {
|
||||
t.Errorf("Catalog mismatch, expected %s, but got %s", test_file.expected_catalog, catalog)
|
||||
if catalog != expectedCatalog {
|
||||
st.Errorf("Catalog mismatch, expected %s, but got %s", expectedCatalog, catalog)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user