108 lines
2.1 KiB
Go
108 lines
2.1 KiB
Go
package renderer
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/hex"
|
|
"strings"
|
|
"testing"
|
|
"unicode/utf16"
|
|
)
|
|
|
|
func TestRenderer_RenderContainsText(t *testing.T) {
|
|
blocks := []Block{
|
|
{Tag: TagTitle, Lines: []string{"ACT"}},
|
|
{Tag: TagText, Lines: []string{"Executor: Jane Doe", "Amount: 100 USD"}},
|
|
}
|
|
|
|
r := Renderer{
|
|
Issuer: Issuer{
|
|
LegalName: "Sendico Ltd",
|
|
LegalAddress: "12 Market Street, London, UK",
|
|
},
|
|
OwnerPassword: "",
|
|
}
|
|
|
|
pdfBytes, err := r.Render(blocks, "deadbeef")
|
|
if err != nil {
|
|
t.Fatalf("Render: %v", err)
|
|
}
|
|
|
|
if len(pdfBytes) == 0 {
|
|
t.Fatalf("expected PDF bytes")
|
|
}
|
|
|
|
checks := []string{"Sendico Ltd", "Jane Doe", "100 USD", "Document integrity hash"}
|
|
|
|
for _, token := range checks {
|
|
if !containsPDFText(pdfBytes, token) {
|
|
t.Fatalf("expected PDF to contain %q", token)
|
|
}
|
|
}
|
|
}
|
|
|
|
func containsPDFText(pdfBytes []byte, text string) bool {
|
|
if bytes.Contains(pdfBytes, []byte(text)) {
|
|
return true
|
|
}
|
|
|
|
hexText := hex.EncodeToString([]byte(text))
|
|
|
|
if bytes.Contains(pdfBytes, []byte(strings.ToUpper(hexText))) {
|
|
return true
|
|
}
|
|
|
|
if bytes.Contains(pdfBytes, []byte(strings.ToLower(hexText))) {
|
|
return true
|
|
}
|
|
|
|
utf16Bytes := encodeUTF16BE(text, false)
|
|
|
|
if bytes.Contains(pdfBytes, utf16Bytes) {
|
|
return true
|
|
}
|
|
|
|
utf16Hex := hex.EncodeToString(utf16Bytes)
|
|
|
|
if bytes.Contains(pdfBytes, []byte(strings.ToUpper(utf16Hex))) {
|
|
return true
|
|
}
|
|
|
|
if bytes.Contains(pdfBytes, []byte(strings.ToLower(utf16Hex))) {
|
|
return true
|
|
}
|
|
|
|
utf16BytesBOM := encodeUTF16BE(text, true)
|
|
if bytes.Contains(pdfBytes, utf16BytesBOM) {
|
|
return true
|
|
}
|
|
|
|
utf16HexBOM := hex.EncodeToString(utf16BytesBOM)
|
|
|
|
if bytes.Contains(pdfBytes, []byte(strings.ToUpper(utf16HexBOM))) {
|
|
return true
|
|
}
|
|
|
|
return bytes.Contains(pdfBytes, []byte(strings.ToLower(utf16HexBOM)))
|
|
}
|
|
|
|
func encodeUTF16BE(text string, withBOM bool) []byte {
|
|
encoded := utf16.Encode([]rune(text))
|
|
length := len(encoded) * 2
|
|
|
|
if withBOM {
|
|
length += 2
|
|
}
|
|
|
|
out := make([]byte, 0, length)
|
|
|
|
if withBOM {
|
|
out = append(out, 0xFE, 0xFF)
|
|
}
|
|
|
|
for _, v := range encoded {
|
|
out = append(out, byte(v>>8), byte(v))
|
|
}
|
|
|
|
return out
|
|
}
|