90 lines
2.7 KiB
Go
90 lines
2.7 KiB
Go
package sign
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/digitorus/pdf"
|
|
)
|
|
|
|
var testFiles = []struct {
|
|
file string
|
|
expectedCatalogs map[CertType]string
|
|
}{
|
|
{
|
|
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 _, 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 := inputFile.Stat()
|
|
if err != nil {
|
|
st.Errorf("Failed to load test PDF")
|
|
return
|
|
}
|
|
size := finfo.Size()
|
|
|
|
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: 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 {
|
|
st.Errorf("%s", err.Error())
|
|
return
|
|
}
|
|
|
|
if catalog != expectedCatalog {
|
|
st.Errorf("Catalog mismatch, expected %s, but got %s", expectedCatalog, catalog)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
}
|